Alex Mikitik
014204cfb7
As recommended, an `internal/assert` package was added with a reduced set of assertions. All tests were then refactored to use the internal assertions. When more complex assertions were used, they have been rewritten using logic and the simplified assertions. Fancy formatting for failures was omitted. The `internal/assert/assertions.diff` function could be overwritten for better formatting. That is where diff libraries are used in other test suites. Refs: #872 Co-authored-by: Alex Mikitik <alex.mikitik@oracle.com>
81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
package benchmark_test
|
|
|
|
import (
|
|
"compress/gzip"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/pelletier/go-toml/v2"
|
|
"github.com/pelletier/go-toml/v2/internal/assert"
|
|
)
|
|
|
|
var bench_inputs = []struct {
|
|
name string
|
|
jsonLen int
|
|
}{
|
|
// from https://gist.githubusercontent.com/feeeper/2197d6d734729625a037af1df14cf2aa/raw/2f22b120e476d897179be3c1e2483d18067aa7df/config.toml
|
|
{"config", 806507},
|
|
|
|
// converted from https://github.com/miloyip/nativejson-benchmark
|
|
{"canada", 2090234},
|
|
{"citm_catalog", 479897},
|
|
{"twitter", 428778},
|
|
{"code", 1940472},
|
|
|
|
// converted from https://raw.githubusercontent.com/mailru/easyjson/master/benchmark/example.json
|
|
{"example", 7779},
|
|
}
|
|
|
|
func TestUnmarshalDatasetCode(t *testing.T) {
|
|
for _, tc := range bench_inputs {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
buf := fixture(t, tc.name)
|
|
|
|
var v interface{}
|
|
assert.NoError(t, toml.Unmarshal(buf, &v))
|
|
|
|
b, err := json.Marshal(v)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, len(b), tc.jsonLen)
|
|
})
|
|
}
|
|
}
|
|
|
|
func BenchmarkUnmarshalDataset(b *testing.B) {
|
|
for _, tc := range bench_inputs {
|
|
b.Run(tc.name, func(b *testing.B) {
|
|
buf := fixture(b, tc.name)
|
|
b.SetBytes(int64(len(buf)))
|
|
b.ReportAllocs()
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
var v interface{}
|
|
assert.NoError(b, toml.Unmarshal(buf, &v))
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// fixture returns the uncompressed contents of path.
|
|
func fixture(tb testing.TB, path string) []byte {
|
|
tb.Helper()
|
|
|
|
file := path + ".toml.gz"
|
|
f, err := os.Open(filepath.Join("testdata", file))
|
|
if os.IsNotExist(err) {
|
|
tb.Skip("benchmark fixture not found:", file)
|
|
}
|
|
assert.NoError(tb, err)
|
|
defer f.Close()
|
|
|
|
gz, err := gzip.NewReader(f)
|
|
assert.NoError(tb, err)
|
|
|
|
buf, err := ioutil.ReadAll(gz)
|
|
assert.NoError(tb, err)
|
|
return buf
|
|
}
|