Implement TestMain

Author: tsarn

CMakeLists.txt

@@ -1,5 +1,10 @@
 cmake_minimum_required(VERSION 3.12)
 project(fuzz VERSION 0.1)
 
-add_executable(fuzz src/test.cpp src/test.h src/main.cpp)
-set_target_properties(fuzz PROPERTIES CXX_STANDARD 17)
+add_library(fuzz src/test.cpp src/test.h src/test_main.h src/test_main.cpp)
+add_library(fuzz::fuzz ALIAS fuzz)
+target_include_directories(fuzz PUBLIC src/)
+
+target_compile_features(fuzz PUBLIC cxx_std_17)
+
+add_subdirectory(examples)

examples/CMakeLists.txt

@@ -0,0 +1,2 @@
+add_executable(fib_generator fib_generator.cpp)
+target_link_libraries(fib_generator PRIVATE fuzz::fuzz)

examples/fib.cpp

@@ -0,0 +1,4 @@
+int fib(int n) {
+    if (n <= 1) return 1;
+    return fib(n-1) + fib(n-2);
+}

examples/fib_generator.cpp

@@ -0,0 +1,12 @@
+#include "test_main.h"
+
+int main(int argc, char **argv) {
+    auto n = primitiveType(PrimitiveType::INT, 0, 10);
+    auto tInt = primitiveType(PrimitiveType::INT);
+    TestSignature fib{ "fib", { n }, tInt };
+
+    TestMain(argc, argv)
+        .fuzz(fib, 5)
+        .run();
+    return 0;
+}

src/main.cpp

@@ -64,7 +64,7 @@ void Test::printPreludeGenerator(std::ostream &ss) const {
     printFunctionCall(ss);
     ss << ";\n";
 
-    ss << "std::cout << \"#define " << macro << " \" << " << var << " << \"\\n\"";
+    ss << "std::cout << \"#define " << macro << " \" << " << var << " << \"\\n\"\n";
 }
 
 void Test::printFunctionCall(std::ostream &ss) const {

src/test.cpp

@@ -60,7 +60,8 @@ struct Value {
 
     template <class Generator>
     static Value generate(Generator& gen, const Type::ptr& type) {
-        std::uniform_int_distribution<PrimitiveInteger> distribution;
+        auto [a, b] = type->getRange();
+        std::uniform_int_distribution<PrimitiveInteger> distribution{a, b};
         return Value { type, distribution(gen) };
     }
 };
@@ -80,7 +81,7 @@ struct Test
     void printFunctionCall(std::ostream &) const;
 
     template <class Generator>
-    static Test generate(Generator gen, std::string name, TestSignature signature) {
+    static Test generate(Generator &gen, std::string name, TestSignature signature) {
         Test test;
         test.name = std::move(name);
         test.signature = std::move(signature);

src/test.h

@@ -0,0 +1,46 @@
+#include "test_main.h"
+
+#include <iostream>
+#include <cstdlib>
+
+TestMain::TestMain(int argc, char **argv) : gen { std::random_device{}() } {
+    for (int i = 0; i < argc; ++i) {
+        args.emplace_back(argv[i]);
+    }
+}
+
+TestMain &TestMain::add(Test test) {
+    tests.push_back(std::move(test));
+    return *this;
+}
+
+void TestMain::usage() {
+    std::cerr << "Usage: " << args[0] << " <-g|-t>";
+    std::exit(1);
+}
+
+void TestMain::run() {
+    if (args.size() != 2) {
+        usage();
+    }
+
+    if (args[1] == "-g") {
+        for (const auto& test : tests) {
+            test.printPreludeGenerator(std::cout);
+        }
+    } else if (args[1] == "-t") {
+        for (const auto& test : tests) {
+            test.print(std::cout);
+        }
+    } else {
+        usage();
+    }
+}
+
+TestMain &TestMain::fuzz(const TestSignature& testSignature, int n) {
+    for (int i = 0; i < n; ++i) {
+        add(Test::generate(gen, "test_" + std::to_string(i), testSignature));
+    }
+
+    return *this;
+}

src/test_main.cpp

@@ -0,0 +1,22 @@
+#pragma once
+
+#include "test.h"
+#include <vector>
+#include <string>
+
+class TestMain {
+private:
+    std::vector<std::string> args;
+    std::vector<Test> tests;
+    std::mt19937 gen;
+
+private:
+    void usage();
+
+public:
+    TestMain(int argc, char **argv);
+
+    TestMain &add(Test test);
+    TestMain &fuzz(const TestSignature& testSignature, int n);
+    void run();
+};