tofino_gtest_utils.h

 
#ifndef BACKENDS_TOFINO_BF_P4C_TEST_GTEST_TOFINO_GTEST_UTILS_H_
#define BACKENDS_TOFINO_BF_P4C_TEST_GTEST_TOFINO_GTEST_UTILS_H_
 
#include <optional>
#include <string>
 
#include "bf-p4c/arch/arch.h"
#include "bf-p4c/bf-p4c-options.h"
#include "bf-p4c/device.h"
#include "gtest/gtest.h"
#include "lib/compile_context.h"
 
namespace P4 {
namespace IR {
namespace BFN {
class Pipe;
}  // namespace BFN
class P4Program;
}  // namespace IR
}  // namespace P4
 
namespace P4::Test {
 
const char *tna_header();
const char *t2na_header();
 
struct MidendTestCase {
    static std::optional<MidendTestCase> create(const std::string &source);
 
    const IR::P4Program *program;
 
    const IR::P4Program *frontendProgram;
};
 
struct TofinoPipeTestCase {
    static std::optional<TofinoPipeTestCase> create(const std::string &source);
 
    static std::optional<TofinoPipeTestCase> createWithThreadLocalInstances(
        const std::string &source);
 
    const IR::BFN::Pipe *pipe;
 
    const IR::P4Program *frontendProgram;
};
 
class BackendTest : public ::testing::Test {
 protected:
    BackendTest() : autoBFNContext(new BFNContext()) {}
 
    AutoCompileContext autoBFNContext;
};
 
class TofinoBackendTest : public BackendTest {
 public:
    TofinoBackendTest() { Device::init("Tofino"_cs); }
};
 
class JBayBackendTest : public BackendTest {
 public:
    JBayBackendTest() { Device::init("Tofino2"_cs); }
};
 
}  // namespace P4::Test
 
#endif /* BACKENDS_TOFINO_BF_P4C_TEST_GTEST_TOFINO_GTEST_UTILS_H_ */