test/unit/utils.py

import os
import unittest
from scripts.test.shared import WASM_OPT, run_process, options


class BinaryenTestCase(unittest.TestCase):
    def input_path(self, filename):
        return os.path.join(options.binaryen_test, 'unit', 'input', filename)

    def roundtrip(self, filename, opts=[]):
        path = self.input_path(filename)
        p = run_process(WASM_OPT + ['-g', '-o', 'a.wasm', path] + opts)
        self.assertEqual(p.returncode, 0)
        with open(path, 'rb') as f:
            with open('a.wasm', 'rb') as g:
                self.assertEqual(g.read(), f.read())

    def disassemble(self, filename):
        path = self.input_path(filename)
        p = run_process(WASM_OPT + ['--print', '-o', os.devnull, path], check=False,
                        capture_output=True)
        self.assertEqual(p.returncode, 0)
        self.assertEqual(p.stderr, '')
        return p.stdout

    def check_features(self, filename, features, opts=[]):
        path = self.input_path(filename)
        cmd = WASM_OPT + ['--print-features', '-o', os.devnull, path] + opts
        p = run_process(cmd, check=False, capture_output=True)
        self.assertEqual(p.returncode, 0)
        self.assertEqual(p.stderr, '')
        self.assertEqual(p.stdout.split('\n')[:-1],
                         ['--enable-' + f for f in features])