1// Copyright 2012 The Go Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style
3// license that can be found in the LICENSE file.
4
5package strconv_test
6
7import (
8	"runtime"
9	. "strconv"
10	"strings"
11	"testing"
12)
13
14var (
15	globalBuf [64]byte
16	nextToOne = "1.00000000000000011102230246251565404236316680908203125" + strings.Repeat("0", 10000) + "1"
17
18	mallocTest = []struct {
19		count int
20		desc  string
21		fn    func()
22	}{
23		{0, `AppendInt(localBuf[:0], 123, 10)`, func() {
24			var localBuf [64]byte
25			AppendInt(localBuf[:0], 123, 10)
26		}},
27		{0, `AppendInt(globalBuf[:0], 123, 10)`, func() { AppendInt(globalBuf[:0], 123, 10) }},
28		{0, `AppendFloat(localBuf[:0], 1.23, 'g', 5, 64)`, func() {
29			var localBuf [64]byte
30			AppendFloat(localBuf[:0], 1.23, 'g', 5, 64)
31		}},
32		{0, `AppendFloat(globalBuf[:0], 1.23, 'g', 5, 64)`, func() { AppendFloat(globalBuf[:0], 1.23, 'g', 5, 64) }},
33		// In practice we see 7 for the next one, but allow some slop.
34		// Before pre-allocation in appendQuotedWith, we saw 39.
35		{10, `AppendQuoteToASCII(nil, oneMB)`, func() { AppendQuoteToASCII(nil, string(oneMB)) }},
36		{0, `ParseFloat("123.45", 64)`, func() { ParseFloat("123.45", 64) }},
37		{0, `ParseFloat("123.456789123456789", 64)`, func() { ParseFloat("123.456789123456789", 64) }},
38		{0, `ParseFloat("1.000000000000000111022302462515654042363166809082031251", 64)`, func() {
39			ParseFloat("1.000000000000000111022302462515654042363166809082031251", 64)
40		}},
41		{0, `ParseFloat("1.0000000000000001110223024625156540423631668090820312500...001", 64)`, func() {
42			ParseFloat(nextToOne, 64)
43		}},
44	}
45)
46
47var oneMB []byte // Will be allocated to 1MB of random data by TestCountMallocs.
48
49func TestCountMallocs(t *testing.T) {
50	if testing.Short() {
51		t.Skip("skipping malloc count in short mode")
52	}
53	if runtime.GOMAXPROCS(0) > 1 {
54		t.Skip("skipping; GOMAXPROCS>1")
55	}
56	// Allocate a big messy buffer for AppendQuoteToASCII's test.
57	oneMB = make([]byte, 1e6)
58	for i := range oneMB {
59		oneMB[i] = byte(i)
60	}
61	for _, mt := range mallocTest {
62		allocs := testing.AllocsPerRun(100, mt.fn)
63		if max := float64(mt.count); allocs > max {
64			t.Errorf("%s: %v allocs, want <=%v", mt.desc, allocs, max)
65		}
66	}
67}
68
69// Sink makes sure the compiler cannot optimize away the benchmarks.
70var Sink struct {
71	Bool       bool
72	Int        int
73	Int64      int64
74	Uint64     uint64
75	Float64    float64
76	Complex128 complex128
77	Error      error
78	Bytes      []byte
79}
80
81func TestAllocationsFromBytes(t *testing.T) {
82	const runsPerTest = 100
83	bytes := struct{ Bool, Number, String, Buffer []byte }{
84		Bool:   []byte("false"),
85		Number: []byte("123456789"),
86		String: []byte("hello, world!"),
87		Buffer: make([]byte, 1024),
88	}
89
90	checkNoAllocs := func(f func()) func(t *testing.T) {
91		return func(t *testing.T) {
92			t.Helper()
93			if allocs := testing.AllocsPerRun(runsPerTest, f); allocs != 0 {
94				t.Errorf("got %v allocs, want 0 allocs", allocs)
95			}
96		}
97	}
98
99	t.Run("Atoi", checkNoAllocs(func() {
100		Sink.Int, Sink.Error = Atoi(string(bytes.Number))
101	}))
102	t.Run("ParseBool", checkNoAllocs(func() {
103		Sink.Bool, Sink.Error = ParseBool(string(bytes.Bool))
104	}))
105	t.Run("ParseInt", checkNoAllocs(func() {
106		Sink.Int64, Sink.Error = ParseInt(string(bytes.Number), 10, 64)
107	}))
108	t.Run("ParseUint", checkNoAllocs(func() {
109		Sink.Uint64, Sink.Error = ParseUint(string(bytes.Number), 10, 64)
110	}))
111	t.Run("ParseFloat", checkNoAllocs(func() {
112		Sink.Float64, Sink.Error = ParseFloat(string(bytes.Number), 64)
113	}))
114	t.Run("ParseComplex", checkNoAllocs(func() {
115		Sink.Complex128, Sink.Error = ParseComplex(string(bytes.Number), 128)
116	}))
117	t.Run("CanBackquote", checkNoAllocs(func() {
118		Sink.Bool = CanBackquote(string(bytes.String))
119	}))
120	t.Run("AppendQuote", checkNoAllocs(func() {
121		Sink.Bytes = AppendQuote(bytes.Buffer[:0], string(bytes.String))
122	}))
123	t.Run("AppendQuoteToASCII", checkNoAllocs(func() {
124		Sink.Bytes = AppendQuoteToASCII(bytes.Buffer[:0], string(bytes.String))
125	}))
126	t.Run("AppendQuoteToGraphic", checkNoAllocs(func() {
127		Sink.Bytes = AppendQuoteToGraphic(bytes.Buffer[:0], string(bytes.String))
128	}))
129}
130
131func TestErrorPrefixes(t *testing.T) {
132	_, errInt := Atoi("INVALID")
133	_, errBool := ParseBool("INVALID")
134	_, errFloat := ParseFloat("INVALID", 64)
135	_, errInt64 := ParseInt("INVALID", 10, 64)
136	_, errUint64 := ParseUint("INVALID", 10, 64)
137
138	vectors := []struct {
139		err  error  // Input error
140		want string // Function name wanted
141	}{
142		{errInt, "Atoi"},
143		{errBool, "ParseBool"},
144		{errFloat, "ParseFloat"},
145		{errInt64, "ParseInt"},
146		{errUint64, "ParseUint"},
147	}
148
149	for _, v := range vectors {
150		nerr, ok := v.err.(*NumError)
151		if !ok {
152			t.Errorf("test %s, error was not a *NumError", v.want)
153			continue
154		}
155		if got := nerr.Func; got != v.want {
156			t.Errorf("mismatching Func: got %s, want %s", got, v.want)
157		}
158	}
159
160}
161