UTBot don't decorate struct field [BUG] #465 (#470)

* UTBot don't decorate struct field [BUG] #465
UnitTestBot · Sep 23, 2022 · 1dd46ac · 1dd46ac
1 parent 80106dd
commit 1dd46ac
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 6 deletions.
diff --git a/server/src/NameDecorator.cpp b/server/src/NameDecorator.cpp
@@ -75,7 +75,7 @@ const std::unordered_set<std::string> NameDecorator::CPP_KEYWORDS = {
     "explicit",
     "export",
     "extern",
-    "false",
+//    "false",
     "float",
     "for",
     "friend",
@@ -115,7 +115,7 @@ const std::unordered_set<std::string> NameDecorator::CPP_KEYWORDS = {
     "this",
     "thread_local",
     "throw",
-    "true",
+//    "true",
     "try",
     "typedef",
     "typeid",

diff --git a/server/src/Tests.h b/server/src/Tests.h
@@ -294,7 +294,7 @@ namespace tests {
             if (structInfo.fields[i].name.empty())
                 return "";
 
-            std::string prefix = "." + structInfo.fields[i].name + " = ";
+            std::string prefix = "." + NameDecorator::decorate(structInfo.fields[i].name) + " = ";
             if (structInfo.isCLike) {
                 return prefix;
             }

diff --git a/server/src/visitors/AssertsVisitor.cpp b/server/src/visitors/AssertsVisitor.cpp
@@ -39,6 +39,9 @@ namespace visitor {
                 macroName = PrinterUtils::EXPECT_DOUBLE_EQ;
             }
         }
+        std::for_each(args.begin(), args.end(), [&type](std::string &arg) {
+            arg = NameDecorator::decorate(arg);
+        });
         return VerboseAssertsVisitor::FunctionSignature{ macroName, std::move(args) };
     }
 

diff --git a/server/src/visitors/ParametrizedAssertsVisitor.cpp b/server/src/visitors/ParametrizedAssertsVisitor.cpp
@@ -78,9 +78,7 @@ namespace visitor {
             printer->strDeclareVar(
                 type.typeName(), PrinterUtils::fillVarName(access, PrinterUtils::EXPECTED), value);
         }
-        else {
-            printer->ss << value << NL;
-        }
+        AbstractValueViewVisitor::visitStruct(type, name, view, access, depth);
     }
 
     void ParametrizedAssertsVisitor::visitPrimitive(const types::Type &type,