common.h

/*
 * SPDX-FileCopyrightText: 2023 Google LLC
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_TEST_BACKEND_COMMON_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_TEST_BACKEND_COMMON_H_
 
#include <filesystem>
#include <optional>
#include <vector>
 
#include <inja/inja.hpp>
 
#include "backends/p4tools/modules/testgen/lib/test_framework.h"
#include "backends/p4tools/modules/testgen/lib/test_object.h"
#include "backends/p4tools/modules/testgen/lib/test_spec.h"
 
namespace P4::P4Tools::P4Testgen::Bmv2 {

class Bmv2TestFramework : public TestFramework {
 public:
    explicit Bmv2TestFramework(const TestBackendConfiguration &testBackendConfiguration);
 
 protected:
    static std::string formatHexExpressionWithSeparators(const IR::Expression &expr);

    virtual inja::json getControlPlane(const TestSpec *testSpec) const;

    [[nodiscard]] virtual inja::json getClone(const TestObjectMap &cloneSpecs) const;

    [[nodiscard]] virtual inja::json::array_t getMeter(const TestObjectMap &meterValues) const;

    [[nodiscard]] virtual inja::json getControlPlaneTable(const TableConfig &tblConfig) const;

    [[nodiscard]] virtual inja::json getControlPlaneForTable(
        const TableMatchMap &matches, const std::vector<ActionArg> &args) const;

    virtual inja::json getSend(const TestSpec *testSpec) const;

    virtual inja::json getExpectedPacket(const TestSpec *testSpec) const;
};
}  // namespace P4::P4Tools::P4Testgen::Bmv2
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_TARGETS_BMV2_TEST_BACKEND_COMMON_H_ */