Unit test of replace_symbolt

Author: tautschnig

unit/Makefile

@@ -35,6 +35,7 @@ SRC += unit_tests.cpp \
        util/irep_sharing.cpp \
        util/message.cpp \
        util/optional.cpp \
+       util/replace_symbol.cpp \
        util/sharing_node.cpp \
        util/sharing_map.cpp \
        util/small_map.cpp \

unit/util/replace_symbol.cpp

@@ -0,0 +1,54 @@
+/*******************************************************************\
+
+ Module: replace_symbolt unit tests
+
+ Author: Michael Tautschnig
+
+\*******************************************************************/
+
+#include <testing-utils/catch.hpp>
+
+#include <util/replace_symbol.h>
+#include <util/std_expr.h>
+
+TEST_CASE("Replace all symbols in expression", "[core][util][replace_symbol]")
+{
+  symbol_exprt s1("a", typet("some_type"));
+  symbol_exprt s2("b", typet("some_type"));
+
+  exprt binary("binary", typet("some_type"));
+  binary.copy_to_operands(s1);
+  binary.copy_to_operands(s2);
+
+  exprt other_expr("other");
+
+  replace_symbolt r;
+  r.insert("a", other_expr);
+
+  REQUIRE(r.replace(binary) == false);
+  REQUIRE(binary.op0() == other_expr);
+
+  REQUIRE(r.replace(s1) == false);
+  REQUIRE(s1 == other_expr);
+
+  REQUIRE(r.replace(s2) == true);
+  REQUIRE(s2 == symbol_exprt("b", typet("some_type")));
+}
+
+TEST_CASE("Lvalue only", "[core][util][optional]")
+{
+  symbol_exprt s1("a", typet("some_type"));
+  symbol_exprt s2("b", typet("some_type"));
+
+  exprt binary("binary", typet("some_type"));
+  binary.copy_to_operands(address_of_exprt(s1));
+  binary.copy_to_operands(address_of_exprt(s2));
+
+  constant_exprt c("some_value", typet("some_type"));
+
+  replace_symbolt r;
+  r.insert("a", c);
+
+  REQUIRE(r.replace(binary) == true);
+  REQUIRE(binary.op0() == address_of_exprt(s1));
+}