Generate correct answers using #defines

Author: tsarn

src/main.cpp

@@ -6,6 +6,10 @@ int main() {
     TestSignature max{ "std::max", { tInt, tInt }, tInt };
 
     std::mt19937 gen { std::random_device{}() };
-    std::cout << Test::generate(gen, "ExampleTest", max).print() << std::endl;
+    auto test = Test::generate(gen, "ExampleTest", max);
+    test.print(std::cout);
+    std::cout << std::endl << std::endl;
+    test.printPreludeGenerator(std::cout);
+    std::cout << std::endl << std::endl;
     return 0;
 }

src/test.cpp

@@ -23,10 +23,30 @@ std::string printPrimitiveType(PrimitiveType type) {
     return "<unknown>";
 }
 
-std::string Test::print() const {
-    std::stringstream ss;
+void Test::print(std::ostream &ss) const {
     ss << "TEST(" << signature.name << "Test, " << name << ") {\n";
-    ss << "    " << signature.name << "(";
+    ss << "    ASSERT_EQ(";
+    printFunctionCall(ss);
+    ss << ", " << resultMacroName() << ");\n";
+    ss << "}\n";
+}
+
+std::string Test::resultMacroName() const {
+    return "FUZZ_TEST_" + name;
+}
+
+void Test::printPreludeGenerator(std::ostream &ss) const {
+    auto macro = resultMacroName();
+    auto var = macro + "_result";
+    ss << "auto " << var << " = ";
+    printFunctionCall(ss);
+    ss << ";\n";
+
+    ss << "std::cout << \"#define " << macro << " \" << " << var << " << \"\\n\"";
+}
+
+void Test::printFunctionCall(std::ostream &ss) const {
+    ss << signature.name << "(";
 
     bool first = true;
 
@@ -39,7 +59,6 @@ std::string Test::print() const {
 
         first = false;
     }
-    ss << ");\n";
-    ss << "}\n";
-    return ss.str();
+
+    ss << ")";
 }

src/test.h

@@ -54,7 +54,10 @@ struct Test
     // after test launch
     std::optional<Value> returnValue;
 
-    [[nodiscard]] std::string print() const;
+    [[nodiscard]] std::string resultMacroName() const;
+    void print(std::ostream &) const;
+    void printPreludeGenerator(std::ostream &) const;
+    void printFunctionCall(std::ostream &) const;
 
     template <class Generator>
     static Test generate(Generator gen, std::string name, TestSignature signature) {