testgen.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TESTGEN_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TESTGEN_H_
 
#include "backends/p4tools/common/p4ctool.h"
 
#include "backends/p4tools/modules/testgen/core/target.h"
#include "backends/p4tools/modules/testgen/lib/test_framework.h"
#include "backends/p4tools/modules/testgen/options.h"
 
namespace P4::P4Tools::P4Testgen {

class Testgen : public AbstractP4cTool<TestgenOptions> {
 protected:
    void registerTarget() override;
 
    int mainImpl(const CompilerResult &compilerResult) override;
 
 public:
    static std::optional<AbstractTestList> generateTests(const TestgenOptions &testgenOptions);
    static std::optional<AbstractTestList> generateTests(std::string_view program,
                                                         const TestgenOptions &testgenOptions);

    static int writeTests(const TestgenOptions &testgenOptions);

    static int writeTests(std::string_view program, const TestgenOptions &testgenOptions);
 
    virtual ~Testgen() = default;
};
 
}  // namespace P4::P4Tools::P4Testgen
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TESTGEN_H_ */