Basic unit test generation

Author: tsarn

.gitignore

@@ -30,3 +30,6 @@
 *.exe
 *.out
 *.app
+
+.idea
+cmake-build-*

CMakeLists.txt

@@ -0,0 +1,5 @@
+cmake_minimum_required(VERSION 3.12)
+project(fuzz VERSION 0.1)
+
+add_executable(fuzz src/test.cpp src/test.h)
+set_target_properties(fuzz PROPERTIES CXX_STANDARD 17)

src/test.cpp

@@ -0,0 +1,42 @@
+#include "test.h"
+
+#include <sstream>
+
+std::string Value::print() const {
+    return "static_cast<" + type.print() + ">(" + std::to_string(value) + ")";
+}
+
+std::string Type::print() const {
+    switch (kind) {
+        case Kind::PRIMITIVE_TYPE: return printPrimitiveType(primitiveType);
+        case Kind::POINTER: return pointerTo->print() + "*";
+    }
+}
+
+std::string printPrimitiveType(PrimitiveType type) {
+    switch (type) {
+        case PrimitiveType::CHAR: return "char";
+        case PrimitiveType::INT: return "int";
+    }
+}
+
+std::string Test::print() const {
+    std::stringstream ss;
+    ss << "TEST(" << signature.name << "Test, " << name << ") {\n";
+    ss << "    EXPECT_EQ(" << signature.name << "(";
+
+    bool first = true;
+
+    for (const auto &arg : arguments) {
+        if (!first) {
+            ss << ", ";
+        }
+
+        ss << arg.print();
+
+        first = false;
+    }
+    ss << "), " << returnValue.print() << ");\n";
+    ss << "}\n";
+    return ss.str();
+}

src/test.h

@@ -0,0 +1,56 @@
+#pragma once
+
+#include <vector>
+#include <string>
+#include <any>
+#include <optional>
+#include <memory>
+
+enum class Kind {
+    PRIMITIVE_TYPE,
+    POINTER
+};
+
+enum class PrimitiveType {
+    CHAR,
+    INT
+};
+
+std::string printPrimitiveType(PrimitiveType type);
+
+struct Type {
+    union {
+        PrimitiveType primitiveType;
+        std::unique_ptr<Type> pointerTo;
+    };
+
+    Kind kind;
+
+    [[nodiscard]] std::string print() const;
+};
+
+std::any fillValue(Type x);
+
+struct TestSignature {
+    std::string name;
+    std::vector<Type> parameterTypes;
+    Type returnType;
+};
+
+struct Value {
+    Type type;
+    int value;
+    [[nodiscard]] std::string print() const;
+};
+
+struct Test
+{
+    std::string name;
+    TestSignature signature;
+    std::vector<Value> arguments;
+
+    // after test launch
+    Value returnValue;
+
+    [[nodiscard]] std::string print() const;
+};