metering_test.go

package metering

import (
	"bytes"
	"encoding/json"
	"fmt"
	"github.com/stretchr/testify/assert"
	"github.com/yyh1102/go-wasm-metering/test"
	"github.com/yyh1102/go-wasm-metering/toolkit"
	"io/ioutil"
	"path"
	"testing"
)

func readCostTable(path string) (toolkit.JSON, error) {
	data, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, err
	}
	jsonObj := make(toolkit.JSON)
	err = json.Unmarshal(data, &jsonObj)
	if err != nil {
		return nil, err
	}
	return jsonObj, nil
}

func readWasmModule(path string) ([]toolkit.JSON, error) {
	var jsonArr []toolkit.JSON
	jsonData, err := ioutil.ReadFile(path)
	if err != nil {
		return nil, err
	}
	if err = json.Unmarshal(jsonData, &jsonArr); err != nil {
		return nil, err
	}
	return jsonArr, nil
}

func TestBasic(t *testing.T) {
	wasm, err := ioutil.ReadFile(path.Join("test", "in", "wasm", "basic.wasm"))
	assert.Nil(t, err)

	meteredWasm, _, err := MeterWASM(wasm, &Options{
		CostTable: test.DefaultCostTable,
	})
	assert.Nil(t, err)
	meteredJson := toolkit.Wasm2Json(meteredWasm)

	expectedWasm, err := ioutil.ReadFile(path.Join("test", "expected-out", "wasm", "basic.wasm"))
	assert.Nil(t, err)
	expectedJson := toolkit.Wasm2Json(expectedWasm)
	//fmt.Printf("%#v\n%#v\n", meteredJson, expectedJson)
	assert.Equal(t, true, assert.ObjectsAreEqual(meteredJson, expectedJson))
	assert.Equal(t, 0, bytes.Compare(meteredWasm, expectedWasm))
	//entries1 := meteredJson[1]["entries"].([]toolkit.TypeEntry)
	//entries2 := meteredJson[2]["entries"].([]toolkit.TypeEntry)
	//assert.Equal(t, "metering", entries2[0]["module_str"])
	//assert.Equal(t, "usegas", entries2[0]["field_str"].(string))
	//assert.Equal(t, "i32", entries1[1]["params"].([]interface{})[0].(string))
}

func TestBasicMeteringTests(t *testing.T) {
	dirName := path.Join("test", "in")
	dir, err := ioutil.ReadDir(path.Join(dirName, "wasm"))
	assert.Nil(t, err)
	for _, file := range dir {
		// read wasm json.
		wasm, err := ioutil.ReadFile(path.Join(dirName, "wasm", file.Name()))
		assert.Nil(t, err)

		module := toolkit.Wasm2Json(wasm)

		// read cost table json.
		metering := Metering{
			opts: Options{
				CostTable: test.DefaultCostTable,
				ModuleStr: defaultModuleStr,
				FieldStr:  defaultFieldStr,
				MeterType: defaultMeterType,
			},
		}
		//fmt.Printf("%s %#v\n", file.Name(), module)
		meteredModule, _, err := metering.meterJSON(module)
		if err != nil {
			assert.Equal(t, "basic+import.wasm", file.Name())
			continue
		}
		//fmt.Printf("%s old %#v\n", file.Name(), meteredModule)
		//fmt.Printf("Gas: %v\n", gasCost)

		expectedWasm, err := ioutil.ReadFile(path.Join("test", "expected-out", "wasm", file.Name()))
		assert.Nil(t, err)
		expectedJson := toolkit.Wasm2Json(expectedWasm)
		//fmt.Printf("%s exp %#v\n", file.Name(), expectedJson)

		if !assert.Equal(t, true, assert.ObjectsAreEqual(meteredModule, expectedJson)) {
			fmt.Printf("file name %s\n", file.Name())
			fmt.Printf("%#v\n%#v\n", meteredModule, expectedJson)
		}
	}

	//fmt.Printf("Basic metering tests failed cases %d", failed)
}