Unit test for generic_type_index

Author: romainbrenguier

unit/java_bytecode/java_types/generic_type_index.cpp

@@ -0,0 +1,92 @@
+/*******************************************************************\
+
+ Module: Unit tests for java_types
+
+ Author: Diffblue Ltd.
+
+\*******************************************************************/
+
+#include <testing-utils/catch.hpp>
+#include <java_types.h>
+
+SCENARIO("generic_type_index", "[core][java_types]")
+{
+  GIVEN("Generic type LGenericClass<TX;TY;>; and parameters X, Y, Z")
+  {
+    const auto symbol_type = symbol_typet("MyType");
+    const auto generic_symbol_type = java_generic_symbol_typet(
+      symbol_type, "LGenericClass<TX;TY;>;", "PrefixClassName");
+    java_generic_parametert paramX("PrefixClassName::X", symbol_typet());
+    java_generic_parametert paramY("PrefixClassName::Y", symbol_typet());
+    java_generic_parametert paramZ("PrefixClassName::Z", symbol_typet());
+
+    WHEN("Looking for parameter indexes")
+    {
+      const auto indexX = generic_symbol_type.generic_type_index(paramX);
+      const auto indexY = generic_symbol_type.generic_type_index(paramY);
+      const auto indexZ = generic_symbol_type.generic_type_index(paramZ);
+
+      THEN("X has index 0, Y index 1 and Z is not found")
+      {
+        REQUIRE(indexX.has_value());
+        REQUIRE(indexX.value() == 0);
+        REQUIRE(indexY.has_value());
+        REQUIRE(indexY.value() == 1);
+        REQUIRE_FALSE(indexZ.has_value());
+      }
+    }
+  }
+
+  GIVEN("Generic type LGenericClass<Tkey;Tvalue;>; and"
+          " parameters key, value, x")
+  {
+    const auto symbol_type = symbol_typet("MyType");
+    const auto generic_symbol_type = java_generic_symbol_typet(
+      symbol_type, "LGenericClass<Tkey;Tvalue;>;", "PrefixClassName");
+    java_generic_parametert param0("PrefixClassName::key", symbol_typet());
+    java_generic_parametert param1("PrefixClassName::value", symbol_typet());
+    java_generic_parametert param2("PrefixClassName::x", symbol_typet());
+
+    WHEN("Looking for parameter indexes")
+    {
+      const auto index_param0 = generic_symbol_type.generic_type_index(param0);
+      const auto index_param1 = generic_symbol_type.generic_type_index(param1);
+      const auto index_param2 = generic_symbol_type.generic_type_index(param2);
+
+      THEN("key has index 0, value index 1 and x is not found")
+      {
+        REQUIRE(index_param0.has_value());
+        REQUIRE(index_param0.value() == 0);
+        REQUIRE(index_param1.has_value());
+        REQUIRE(index_param1.value() == 1);
+        REQUIRE_FALSE(index_param2.has_value());
+      }
+    }
+  }
+
+  GIVEN("Generic type Ljava/util/HashMap<TK;TV;>; and parameters K, V, T")
+  {
+    const auto symbol_type = symbol_typet("MyType");
+    const auto generic_symbol_type = java_generic_symbol_typet(
+      symbol_type, "Ljava/util/HashMap<TK;TV;>;", "java.util.HashMap");
+    java_generic_parametert param0("java.util.HashMap::K", symbol_typet());
+    java_generic_parametert param1("java.util.HashMap::V", symbol_typet());
+    java_generic_parametert param2("java.util.HashMap::T", symbol_typet());
+
+    WHEN("Looking for parameter indexes")
+    {
+      const auto index_param0 = generic_symbol_type.generic_type_index(param0);
+      const auto index_param1 = generic_symbol_type.generic_type_index(param1);
+      const auto index_param2 = generic_symbol_type.generic_type_index(param2);
+
+      THEN("K has index 0, V index 1 and T is not found")
+      {
+        REQUIRE(index_param0.has_value());
+        REQUIRE(index_param0.value() == 0);
+        REQUIRE(index_param1.has_value());
+        REQUIRE(index_param1.value() == 1);
+        REQUIRE_FALSE(index_param2.has_value());
+      }
+    }
+  }
+}