path: root/tests/init.go

package tests

import (
    "encoding/json"
    "fmt"
    "io"
    "io/ioutil"
    "net/http"
    "os"
    "path/filepath"
)

var (
    baseDir            = filepath.Join(".", "files")
    blockTestDir       = filepath.Join(baseDir, "BlockTests")
    stateTestDir       = filepath.Join(baseDir, "StateTests")
    transactionTestDir = filepath.Join(baseDir, "TransactionTests")
    vmTestDir          = filepath.Join(baseDir, "VMTests")

    blockSkipTests = []string{"SimpleTx3"}
    transSkipTests = []string{"TransactionWithHihghNonce256"}
    stateSkipTests = []string{"mload32bitBound_return", "mload32bitBound_return2"}
    vmSkipTests    = []string{}
)

func readJSON(reader io.Reader, value interface{}) error {
    data, err := ioutil.ReadAll(reader)
    if err != nil {
        return fmt.Errorf("Error reading JSON file", err.Error())
    }

    if err = json.Unmarshal(data, &value); err != nil {
        if syntaxerr, ok := err.(*json.SyntaxError); ok {
            line := findLine(data, syntaxerr.Offset)
            return fmt.Errorf("JSON syntax error at line %v: %v", line, err)
        }
        return fmt.Errorf("JSON unmarshal error: %v", err)
    }
    return nil
}

// findLine returns the line number for the given offset into data.
func findLine(data []byte, offset int64) (line int) {
    line = 1
    for i, r := range string(data) {
        if int64(i) >= offset {
            return
        }
        if r == '\n' {
            line++
        }
    }
    return
}

func readHttpFile(uri string, value interface{}) error {
    resp, err := http.Get(uri)
    if err != nil {
        return err
    }
    defer resp.Body.Close()

    err = readJSON(resp.Body, value)
    if err != nil {
        return err
    }
    return nil
}

func readTestFile(fn string, value interface{}) error {
    file, err := os.Open(fn)
    if err != nil {
        return err
    }
    defer file.Close()

    err = readJSON(file, value)
    if err != nil {
        return fmt.Errorf("%s in file %s", err.Error(), fn)
    }
    return nil
}