1 files changed, 64 insertions, 0 deletions
diff --git a/compiler/mir-onnx-importer/ONNXNodeConverterRegistry.test.cpp b/compiler/mir-onnx-importer/ONNXNodeConverterRegistry.test.cpp
new file mode 100644
index 000000000..dfc3e4216
--- /dev/null
+++ b/compiler/mir-onnx-importer/ONNXNodeConverterRegistry.test.cpp
@@ -0,0 +1,64 @@
+/*
+ * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "ONNXNodeConverterRegistry.h"
+#include "ONNXHelpers.h"
+
+#include "gtest/gtest.h"
+
+using namespace mir_onnx;
+
+void converterV1(const onnx::NodeProto &node, ConverterContext *ctx) {}
+void converterV3(const onnx::NodeProto &node, ConverterContext *ctx) {}
+void converterV7(const onnx::NodeProto &node, ConverterContext *ctx) {}
+
+class NodeConverterRegsitryTest : public ::testing::Test
+{
+protected:
+  void SetUp() override
+  {
+    registry.registerConverter("dummy", 1, converterV1);
+    registry.registerConverter("dummy", 3, converterV3);
+    registry.registerConverter("dummy", 7, converterV7);
+    registry.registerConverter("dummy", firstUnknownOpset, nullptr);
+  }
+
+  NodeConverterRegistry registry;
+};
+
+TEST_F(NodeConverterRegsitryTest, existing_lookup_works)
+{
+  auto res = registry.lookup("dummy", 1);
+  ASSERT_EQ(res, &converterV1);
+}
+
+TEST_F(NodeConverterRegsitryTest, skipped_lookup_works)
+{
+  auto res = registry.lookup("dummy", 2);
+  ASSERT_EQ(res, &converterV1);
+}
+
+TEST_F(NodeConverterRegsitryTest, first_unknown_version_works)
+{
+  auto res = registry.lookup("dummy", 14);
+  ASSERT_EQ(res, nullptr);
+}
+
+TEST_F(NodeConverterRegsitryTest, lower_than_first_version)
+{
+  auto res = registry.lookup("dummy", 0);
+  ASSERT_EQ(res, nullptr);
+}