Add unit test for array_poolt

unit/Makefile

@@ -14,6 +14,7 @@ SRC += unit_tests.cpp \
        analyses/does_remove_const/is_type_at_least_as_const_as.cpp \
        goto-programs/goto_trace_output.cpp \
        path_strategies.cpp \
+       solvers/refinement/array_pool/array_pool.cpp \
        solvers/refinement/string_constraint_generator_valueof/calculate_max_string_length.cpp \
        solvers/refinement/string_constraint_generator_valueof/get_numeric_value_from_character.cpp \
        solvers/refinement/string_constraint_generator_valueof/is_digit_with_radix.cpp \

unit/solvers/refinement/array_pool/array_pool.cpp

@@ -0,0 +1,148 @@
+/*******************************************************************\
+
+ Module: Unit tests for array pool
+   solvers/refinement/string_constraint_generator.h
+
+ Author: Diffblue Ltd.
+
+\*******************************************************************/
+
+#include <testing-utils/catch.hpp>
+
+#include <solvers/refinement/string_constraint_generator.h>
+
+SCENARIO(
+  "array_pool", "[core][solvers][refinement][string_constraint_generator]")
+{
+  const std::size_t pointer_width = 16;
+  const auto char_type = unsignedbv_typet(8);
+  const auto length_type = signedbv_typet(32);
+  const auto pointer_type = pointer_typet(char_type, pointer_width);
+
+  GIVEN("An array pool")
+  {
+    symbol_generatort symbol_generator;
+    array_poolt pool(symbol_generator);
+
+    WHEN("Looking for a pointer symbol")
+    {
+      const symbol_exprt pointer("my_pointer", pointer_type);
+      const symbol_exprt pointer_length("my_pointer_length", length_type);
+      const array_string_exprt associated_array =
+        pool.find(pointer, pointer_length);
+
+      THEN("A second find give the same array")
+      {
+        const array_string_exprt second_lookup =
+          pool.find(pointer, pointer_length);
+
+        REQUIRE(second_lookup == associated_array);
+      }
+    }
+
+    WHEN("Looking for the address of the first element of a constant array")
+    {
+      const array_typet array_type(char_type, from_integer(3, length_type));
+      const exprt array = [array_type, char_type]{
+        array_exprt a(array_type);
+        a.operands().push_back(from_integer('f', char_type));
+        a.operands().push_back(from_integer('o', char_type));
+        a.operands().push_back(from_integer('o', char_type));
+        return a;
+      }();
+        const exprt first_element =
+          index_exprt(array, from_integer(0, length_type));
+      const exprt pointer = address_of_exprt(first_element, pointer_type);
+      const array_string_exprt associated_array =
+        pool.find(pointer, array_type.size());
+
+      THEN("The associated array should be the pointed array")
+      {
+        REQUIRE(associated_array == array);
+      }
+    }
+
+    WHEN("Looking for a null pointer")
+    {
+      const null_pointer_exprt null_pointer(pointer_type);
+      const symbol_exprt pointer_length("null_pointer_length", length_type);
+      const array_string_exprt associated_array =
+        pool.find(null_pointer, pointer_length);
+
+      THEN("`find` always gives the same array")
+      {
+        const symbol_exprt pointer_length2("null_pointer_length2", length_type);
+        const array_string_exprt second_lookup =
+          pool.find(null_pointer, pointer_length2);
+
+        REQUIRE(second_lookup == associated_array);
+      }
+    }
+
+    WHEN("Looking for an if-expression")
+    {
+      const exprt boolean_symbol = symbol_exprt("boolean", bool_typet());
+      const exprt true_case = symbol_exprt("pointer1", pointer_type);
+      const exprt false_case = symbol_exprt("pointer2", pointer_type);
+      const exprt if_expr = if_exprt(boolean_symbol, true_case, false_case);
+      const symbol_exprt pointer_length("pointer_length", length_type);
+
+      const array_string_exprt associated_array =
+        pool.find(if_expr, pointer_length);
+
+      THEN("Arrays associated to the subexpressions are the subexpressions of "
+           "the associated array")
+      {
+        const symbol_exprt pointer_length1("pointer_length1", length_type);
+        const array_string_exprt associated_to_true =
+          pool.find(true_case, pointer_length1);
+        const symbol_exprt pointer_length2("pointer_length2", length_type);
+        const array_string_exprt associated_to_false =
+          pool.find(false_case, pointer_length2);
+
+        const typet recomposed_type = array_typet(
+          char_type, if_exprt(boolean_symbol, pointer_length, pointer_length));
+        const exprt recomposed_array = if_exprt(
+          boolean_symbol,
+          associated_to_true,
+          associated_to_false,
+          recomposed_type);
+
+        REQUIRE(associated_array == recomposed_array);
+      }
+    }
+
+    WHEN("Looking for two pointer symbols")
+    {
+      const exprt true_case = symbol_exprt("pointer1", pointer_type);
+      const symbol_exprt pointer_length1("pointer_length1", length_type);
+      const exprt false_case = symbol_exprt("pointer2", pointer_type);
+      const symbol_exprt pointer_length2("pointer_length2", length_type);
+
+      const array_string_exprt associated_to_true =
+        pool.find(true_case, pointer_length1);
+      const array_string_exprt associated_to_false =
+        pool.find(false_case, pointer_length2);
+
+      THEN("Looking for an if-expression containing these two symbols")
+      {
+        const exprt boolean_symbol = symbol_exprt("boolean", bool_typet());
+        const exprt if_expr = if_exprt(boolean_symbol, true_case, false_case);
+        const symbol_exprt pointer_length("pointer_length", length_type);
+        const array_string_exprt associated_array =
+          pool.find(if_expr, pointer_length);
+
+        const typet recomposed_type = array_typet(
+          char_type,
+          if_exprt(boolean_symbol, pointer_length1, pointer_length2));
+        const exprt recomposed_array = if_exprt(
+          boolean_symbol,
+          associated_to_true,
+          associated_to_false,
+          recomposed_type);
+
+        REQUIRE(associated_array == recomposed_array);
+      }
+    }
+  }
+}