test/gtest/print-test.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

#include <iostream>

#include "parser/wat-parser.h"
#include "support/colors.h"
#include "wasm.h"
#include "gtest/gtest.h"

#ifndef wasm_test_gtest_print_test_h
#define wasm_test_gtest_print_test_h

// Helper test fixture for parsing wast and checking some printed output.
class PrintTest : public ::testing::Test {
  bool colors = Colors::isEnabled();

public:
  PrintTest() { Colors::setEnabled(false); }

protected:
  void TearDown() override { Colors::setEnabled(colors); }

  void parseWast(wasm::Module& wasm, const std::string& wast) {
    auto parsed = wasm::WATParser::parseModule(wasm, wast);
    if (auto* err = parsed.getErr()) {
      wasm::Fatal() << err->msg << "\n";
    }
  }
};

#endif