| 1 |
747 |
jeremybenn |
// Copyright 2009 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 |
|
|
|
| 5 |
|
|
package fmt_test
|
| 6 |
|
|
|
| 7 |
|
|
import (
|
| 8 |
|
|
"bufio"
|
| 9 |
|
|
"bytes"
|
| 10 |
|
|
"errors"
|
| 11 |
|
|
. "fmt"
|
| 12 |
|
|
"io"
|
| 13 |
|
|
"math"
|
| 14 |
|
|
"reflect"
|
| 15 |
|
|
"regexp"
|
| 16 |
|
|
"strings"
|
| 17 |
|
|
"testing"
|
| 18 |
|
|
"unicode/utf8"
|
| 19 |
|
|
)
|
| 20 |
|
|
|
| 21 |
|
|
type ScanTest struct {
|
| 22 |
|
|
text string
|
| 23 |
|
|
in interface{}
|
| 24 |
|
|
out interface{}
|
| 25 |
|
|
}
|
| 26 |
|
|
|
| 27 |
|
|
type ScanfTest struct {
|
| 28 |
|
|
format string
|
| 29 |
|
|
text string
|
| 30 |
|
|
in interface{}
|
| 31 |
|
|
out interface{}
|
| 32 |
|
|
}
|
| 33 |
|
|
|
| 34 |
|
|
type ScanfMultiTest struct {
|
| 35 |
|
|
format string
|
| 36 |
|
|
text string
|
| 37 |
|
|
in []interface{}
|
| 38 |
|
|
out []interface{}
|
| 39 |
|
|
err string
|
| 40 |
|
|
}
|
| 41 |
|
|
|
| 42 |
|
|
var (
|
| 43 |
|
|
boolVal bool
|
| 44 |
|
|
intVal int
|
| 45 |
|
|
int8Val int8
|
| 46 |
|
|
int16Val int16
|
| 47 |
|
|
int32Val int32
|
| 48 |
|
|
int64Val int64
|
| 49 |
|
|
uintVal uint
|
| 50 |
|
|
uint8Val uint8
|
| 51 |
|
|
uint16Val uint16
|
| 52 |
|
|
uint32Val uint32
|
| 53 |
|
|
uint64Val uint64
|
| 54 |
|
|
float32Val float32
|
| 55 |
|
|
float64Val float64
|
| 56 |
|
|
stringVal string
|
| 57 |
|
|
stringVal1 string
|
| 58 |
|
|
bytesVal []byte
|
| 59 |
|
|
runeVal rune
|
| 60 |
|
|
complex64Val complex64
|
| 61 |
|
|
complex128Val complex128
|
| 62 |
|
|
renamedBoolVal renamedBool
|
| 63 |
|
|
renamedIntVal renamedInt
|
| 64 |
|
|
renamedInt8Val renamedInt8
|
| 65 |
|
|
renamedInt16Val renamedInt16
|
| 66 |
|
|
renamedInt32Val renamedInt32
|
| 67 |
|
|
renamedInt64Val renamedInt64
|
| 68 |
|
|
renamedUintVal renamedUint
|
| 69 |
|
|
renamedUint8Val renamedUint8
|
| 70 |
|
|
renamedUint16Val renamedUint16
|
| 71 |
|
|
renamedUint32Val renamedUint32
|
| 72 |
|
|
renamedUint64Val renamedUint64
|
| 73 |
|
|
renamedUintptrVal renamedUintptr
|
| 74 |
|
|
renamedStringVal renamedString
|
| 75 |
|
|
renamedBytesVal renamedBytes
|
| 76 |
|
|
renamedFloat32Val renamedFloat32
|
| 77 |
|
|
renamedFloat64Val renamedFloat64
|
| 78 |
|
|
renamedComplex64Val renamedComplex64
|
| 79 |
|
|
renamedComplex128Val renamedComplex128
|
| 80 |
|
|
)
|
| 81 |
|
|
|
| 82 |
|
|
type FloatTest struct {
|
| 83 |
|
|
text string
|
| 84 |
|
|
in float64
|
| 85 |
|
|
out float64
|
| 86 |
|
|
}
|
| 87 |
|
|
|
| 88 |
|
|
// Xs accepts any non-empty run of the verb character
|
| 89 |
|
|
type Xs string
|
| 90 |
|
|
|
| 91 |
|
|
func (x *Xs) Scan(state ScanState, verb rune) error {
|
| 92 |
|
|
tok, err := state.Token(true, func(r rune) bool { return r == verb })
|
| 93 |
|
|
if err != nil {
|
| 94 |
|
|
return err
|
| 95 |
|
|
}
|
| 96 |
|
|
s := string(tok)
|
| 97 |
|
|
if !regexp.MustCompile("^" + string(verb) + "+$").MatchString(s) {
|
| 98 |
|
|
return errors.New("syntax error for xs")
|
| 99 |
|
|
}
|
| 100 |
|
|
*x = Xs(s)
|
| 101 |
|
|
return nil
|
| 102 |
|
|
}
|
| 103 |
|
|
|
| 104 |
|
|
var xVal Xs
|
| 105 |
|
|
|
| 106 |
|
|
// IntString accepts an integer followed immediately by a string.
|
| 107 |
|
|
// It tests the embedding of a scan within a scan.
|
| 108 |
|
|
type IntString struct {
|
| 109 |
|
|
i int
|
| 110 |
|
|
s string
|
| 111 |
|
|
}
|
| 112 |
|
|
|
| 113 |
|
|
func (s *IntString) Scan(state ScanState, verb rune) error {
|
| 114 |
|
|
if _, err := Fscan(state, &s.i); err != nil {
|
| 115 |
|
|
return err
|
| 116 |
|
|
}
|
| 117 |
|
|
|
| 118 |
|
|
tok, err := state.Token(true, nil)
|
| 119 |
|
|
if err != nil {
|
| 120 |
|
|
return err
|
| 121 |
|
|
}
|
| 122 |
|
|
s.s = string(tok)
|
| 123 |
|
|
return nil
|
| 124 |
|
|
}
|
| 125 |
|
|
|
| 126 |
|
|
var intStringVal IntString
|
| 127 |
|
|
|
| 128 |
|
|
// myStringReader implements Read but not ReadRune, allowing us to test our readRune wrapper
|
| 129 |
|
|
// type that creates something that can read runes given only Read().
|
| 130 |
|
|
type myStringReader struct {
|
| 131 |
|
|
r *strings.Reader
|
| 132 |
|
|
}
|
| 133 |
|
|
|
| 134 |
|
|
func (s *myStringReader) Read(p []byte) (n int, err error) {
|
| 135 |
|
|
return s.r.Read(p)
|
| 136 |
|
|
}
|
| 137 |
|
|
|
| 138 |
|
|
func newReader(s string) *myStringReader {
|
| 139 |
|
|
return &myStringReader{strings.NewReader(s)}
|
| 140 |
|
|
}
|
| 141 |
|
|
|
| 142 |
|
|
var scanTests = []ScanTest{
|
| 143 |
|
|
// Basic types
|
| 144 |
|
|
{"T\n", &boolVal, true}, // boolean test vals toggle to be sure they are written
|
| 145 |
|
|
{"F\n", &boolVal, false}, // restored to zero value
|
| 146 |
|
|
{"21\n", &intVal, 21},
|
| 147 |
|
|
{"0\n", &intVal, 0},
|
| 148 |
|
|
{"000\n", &intVal, 0},
|
| 149 |
|
|
{"0x10\n", &intVal, 0x10},
|
| 150 |
|
|
{"-0x10\n", &intVal, -0x10},
|
| 151 |
|
|
{"0377\n", &intVal, 0377},
|
| 152 |
|
|
{"-0377\n", &intVal, -0377},
|
| 153 |
|
|
{"0\n", &uintVal, uint(0)},
|
| 154 |
|
|
{"000\n", &uintVal, uint(0)},
|
| 155 |
|
|
{"0x10\n", &uintVal, uint(0x10)},
|
| 156 |
|
|
{"0377\n", &uintVal, uint(0377)},
|
| 157 |
|
|
{"22\n", &int8Val, int8(22)},
|
| 158 |
|
|
{"23\n", &int16Val, int16(23)},
|
| 159 |
|
|
{"24\n", &int32Val, int32(24)},
|
| 160 |
|
|
{"25\n", &int64Val, int64(25)},
|
| 161 |
|
|
{"127\n", &int8Val, int8(127)},
|
| 162 |
|
|
{"-21\n", &intVal, -21},
|
| 163 |
|
|
{"-22\n", &int8Val, int8(-22)},
|
| 164 |
|
|
{"-23\n", &int16Val, int16(-23)},
|
| 165 |
|
|
{"-24\n", &int32Val, int32(-24)},
|
| 166 |
|
|
{"-25\n", &int64Val, int64(-25)},
|
| 167 |
|
|
{"-128\n", &int8Val, int8(-128)},
|
| 168 |
|
|
{"+21\n", &intVal, +21},
|
| 169 |
|
|
{"+22\n", &int8Val, int8(+22)},
|
| 170 |
|
|
{"+23\n", &int16Val, int16(+23)},
|
| 171 |
|
|
{"+24\n", &int32Val, int32(+24)},
|
| 172 |
|
|
{"+25\n", &int64Val, int64(+25)},
|
| 173 |
|
|
{"+127\n", &int8Val, int8(+127)},
|
| 174 |
|
|
{"26\n", &uintVal, uint(26)},
|
| 175 |
|
|
{"27\n", &uint8Val, uint8(27)},
|
| 176 |
|
|
{"28\n", &uint16Val, uint16(28)},
|
| 177 |
|
|
{"29\n", &uint32Val, uint32(29)},
|
| 178 |
|
|
{"30\n", &uint64Val, uint64(30)},
|
| 179 |
|
|
{"255\n", &uint8Val, uint8(255)},
|
| 180 |
|
|
{"32767\n", &int16Val, int16(32767)},
|
| 181 |
|
|
{"2.3\n", &float64Val, 2.3},
|
| 182 |
|
|
{"2.3e1\n", &float32Val, float32(2.3e1)},
|
| 183 |
|
|
{"2.3e2\n", &float64Val, 2.3e2},
|
| 184 |
|
|
{"2.3p2\n", &float64Val, 2.3 * 4},
|
| 185 |
|
|
{"2.3p+2\n", &float64Val, 2.3 * 4},
|
| 186 |
|
|
{"2.3p+66\n", &float64Val, 2.3 * (1 << 32) * (1 << 32) * 4},
|
| 187 |
|
|
{"2.3p-66\n", &float64Val, 2.3 / ((1 << 32) * (1 << 32) * 4)},
|
| 188 |
|
|
{"2.35\n", &stringVal, "2.35"},
|
| 189 |
|
|
{"2345678\n", &bytesVal, []byte("2345678")},
|
| 190 |
|
|
{"(3.4e1-2i)\n", &complex128Val, 3.4e1 - 2i},
|
| 191 |
|
|
{"-3.45e1-3i\n", &complex64Val, complex64(-3.45e1 - 3i)},
|
| 192 |
|
|
{"-.45e1-1e2i\n", &complex128Val, complex128(-.45e1 - 100i)},
|
| 193 |
|
|
{"hello\n", &stringVal, "hello"},
|
| 194 |
|
|
|
| 195 |
|
|
// Renamed types
|
| 196 |
|
|
{"true\n", &renamedBoolVal, renamedBool(true)},
|
| 197 |
|
|
{"F\n", &renamedBoolVal, renamedBool(false)},
|
| 198 |
|
|
{"101\n", &renamedIntVal, renamedInt(101)},
|
| 199 |
|
|
{"102\n", &renamedIntVal, renamedInt(102)},
|
| 200 |
|
|
{"103\n", &renamedUintVal, renamedUint(103)},
|
| 201 |
|
|
{"104\n", &renamedUintVal, renamedUint(104)},
|
| 202 |
|
|
{"105\n", &renamedInt8Val, renamedInt8(105)},
|
| 203 |
|
|
{"106\n", &renamedInt16Val, renamedInt16(106)},
|
| 204 |
|
|
{"107\n", &renamedInt32Val, renamedInt32(107)},
|
| 205 |
|
|
{"108\n", &renamedInt64Val, renamedInt64(108)},
|
| 206 |
|
|
{"109\n", &renamedUint8Val, renamedUint8(109)},
|
| 207 |
|
|
{"110\n", &renamedUint16Val, renamedUint16(110)},
|
| 208 |
|
|
{"111\n", &renamedUint32Val, renamedUint32(111)},
|
| 209 |
|
|
{"112\n", &renamedUint64Val, renamedUint64(112)},
|
| 210 |
|
|
{"113\n", &renamedUintptrVal, renamedUintptr(113)},
|
| 211 |
|
|
{"114\n", &renamedStringVal, renamedString("114")},
|
| 212 |
|
|
{"115\n", &renamedBytesVal, renamedBytes([]byte("115"))},
|
| 213 |
|
|
|
| 214 |
|
|
// Custom scanners.
|
| 215 |
|
|
{" vvv ", &xVal, Xs("vvv")},
|
| 216 |
|
|
{" 1234hello", &intStringVal, IntString{1234, "hello"}},
|
| 217 |
|
|
|
| 218 |
|
|
// Fixed bugs
|
| 219 |
|
|
{"2147483648\n", &int64Val, int64(2147483648)}, // was: integer overflow
|
| 220 |
|
|
}
|
| 221 |
|
|
|
| 222 |
|
|
var scanfTests = []ScanfTest{
|
| 223 |
|
|
{"%v", "TRUE\n", &boolVal, true},
|
| 224 |
|
|
{"%t", "false\n", &boolVal, false},
|
| 225 |
|
|
{"%v", "-71\n", &intVal, -71},
|
| 226 |
|
|
{"%v", "0377\n", &intVal, 0377},
|
| 227 |
|
|
{"%v", "0x44\n", &intVal, 0x44},
|
| 228 |
|
|
{"%d", "72\n", &intVal, 72},
|
| 229 |
|
|
{"%c", "a\n", &runeVal, 'a'},
|
| 230 |
|
|
{"%c", "\u5072\n", &runeVal, '\u5072'},
|
| 231 |
|
|
{"%c", "\u1234\n", &runeVal, '\u1234'},
|
| 232 |
|
|
{"%d", "73\n", &int8Val, int8(73)},
|
| 233 |
|
|
{"%d", "+74\n", &int16Val, int16(74)},
|
| 234 |
|
|
{"%d", "75\n", &int32Val, int32(75)},
|
| 235 |
|
|
{"%d", "76\n", &int64Val, int64(76)},
|
| 236 |
|
|
{"%b", "1001001\n", &intVal, 73},
|
| 237 |
|
|
{"%o", "075\n", &intVal, 075},
|
| 238 |
|
|
{"%x", "a75\n", &intVal, 0xa75},
|
| 239 |
|
|
{"%v", "71\n", &uintVal, uint(71)},
|
| 240 |
|
|
{"%d", "72\n", &uintVal, uint(72)},
|
| 241 |
|
|
{"%d", "73\n", &uint8Val, uint8(73)},
|
| 242 |
|
|
{"%d", "74\n", &uint16Val, uint16(74)},
|
| 243 |
|
|
{"%d", "75\n", &uint32Val, uint32(75)},
|
| 244 |
|
|
{"%d", "76\n", &uint64Val, uint64(76)},
|
| 245 |
|
|
{"%b", "1001001\n", &uintVal, uint(73)},
|
| 246 |
|
|
{"%o", "075\n", &uintVal, uint(075)},
|
| 247 |
|
|
{"%x", "a75\n", &uintVal, uint(0xa75)},
|
| 248 |
|
|
{"%x", "A75\n", &uintVal, uint(0xa75)},
|
| 249 |
|
|
{"%U", "U+1234\n", &intVal, int(0x1234)},
|
| 250 |
|
|
{"%U", "U+4567\n", &uintVal, uint(0x4567)},
|
| 251 |
|
|
|
| 252 |
|
|
// Strings
|
| 253 |
|
|
{"%s", "using-%s\n", &stringVal, "using-%s"},
|
| 254 |
|
|
{"%x", "7573696e672d2578\n", &stringVal, "using-%x"},
|
| 255 |
|
|
{"%q", `"quoted\twith\\do\u0075bl\x65s"` + "\n", &stringVal, "quoted\twith\\doubles"},
|
| 256 |
|
|
{"%q", "`quoted with backs`\n", &stringVal, "quoted with backs"},
|
| 257 |
|
|
|
| 258 |
|
|
// Byte slices
|
| 259 |
|
|
{"%s", "bytes-%s\n", &bytesVal, []byte("bytes-%s")},
|
| 260 |
|
|
{"%x", "62797465732d2578\n", &bytesVal, []byte("bytes-%x")},
|
| 261 |
|
|
{"%q", `"bytes\rwith\vdo\u0075bl\x65s"` + "\n", &bytesVal, []byte("bytes\rwith\vdoubles")},
|
| 262 |
|
|
{"%q", "`bytes with backs`\n", &bytesVal, []byte("bytes with backs")},
|
| 263 |
|
|
|
| 264 |
|
|
// Renamed types
|
| 265 |
|
|
{"%v\n", "true\n", &renamedBoolVal, renamedBool(true)},
|
| 266 |
|
|
{"%t\n", "F\n", &renamedBoolVal, renamedBool(false)},
|
| 267 |
|
|
{"%v", "101\n", &renamedIntVal, renamedInt(101)},
|
| 268 |
|
|
{"%c", "\u0101\n", &renamedIntVal, renamedInt('\u0101')},
|
| 269 |
|
|
{"%o", "0146\n", &renamedIntVal, renamedInt(102)},
|
| 270 |
|
|
{"%v", "103\n", &renamedUintVal, renamedUint(103)},
|
| 271 |
|
|
{"%d", "104\n", &renamedUintVal, renamedUint(104)},
|
| 272 |
|
|
{"%d", "105\n", &renamedInt8Val, renamedInt8(105)},
|
| 273 |
|
|
{"%d", "106\n", &renamedInt16Val, renamedInt16(106)},
|
| 274 |
|
|
{"%d", "107\n", &renamedInt32Val, renamedInt32(107)},
|
| 275 |
|
|
{"%d", "108\n", &renamedInt64Val, renamedInt64(108)},
|
| 276 |
|
|
{"%x", "6D\n", &renamedUint8Val, renamedUint8(109)},
|
| 277 |
|
|
{"%o", "0156\n", &renamedUint16Val, renamedUint16(110)},
|
| 278 |
|
|
{"%d", "111\n", &renamedUint32Val, renamedUint32(111)},
|
| 279 |
|
|
{"%d", "112\n", &renamedUint64Val, renamedUint64(112)},
|
| 280 |
|
|
{"%d", "113\n", &renamedUintptrVal, renamedUintptr(113)},
|
| 281 |
|
|
{"%s", "114\n", &renamedStringVal, renamedString("114")},
|
| 282 |
|
|
{"%q", "\"1155\"\n", &renamedBytesVal, renamedBytes([]byte("1155"))},
|
| 283 |
|
|
{"%g", "116e1\n", &renamedFloat32Val, renamedFloat32(116e1)},
|
| 284 |
|
|
{"%g", "-11.7e+1", &renamedFloat64Val, renamedFloat64(-11.7e+1)},
|
| 285 |
|
|
{"%g", "11+6e1i\n", &renamedComplex64Val, renamedComplex64(11 + 6e1i)},
|
| 286 |
|
|
{"%g", "-11.+7e+1i", &renamedComplex128Val, renamedComplex128(-11. + 7e+1i)},
|
| 287 |
|
|
|
| 288 |
|
|
// Interesting formats
|
| 289 |
|
|
{"here is\tthe value:%d", "here is the\tvalue:118\n", &intVal, 118},
|
| 290 |
|
|
{"%% %%:%d", "% %:119\n", &intVal, 119},
|
| 291 |
|
|
|
| 292 |
|
|
// Corner cases
|
| 293 |
|
|
{"%x", "FFFFFFFF\n", &uint32Val, uint32(0xFFFFFFFF)},
|
| 294 |
|
|
|
| 295 |
|
|
// Custom scanner.
|
| 296 |
|
|
{"%s", " sss ", &xVal, Xs("sss")},
|
| 297 |
|
|
{"%2s", "sssss", &xVal, Xs("ss")},
|
| 298 |
|
|
|
| 299 |
|
|
// Fixed bugs
|
| 300 |
|
|
{"%d\n", "27\n", &intVal, 27}, // ok
|
| 301 |
|
|
{"%d\n", "28 \n", &intVal, 28}, // was: "unexpected newline"
|
| 302 |
|
|
{"%v", "0", &intVal, 0}, // was: "EOF"; 0 was taken as base prefix and not counted.
|
| 303 |
|
|
{"%v", "0", &uintVal, uint(0)}, // was: "EOF"; 0 was taken as base prefix and not counted.
|
| 304 |
|
|
}
|
| 305 |
|
|
|
| 306 |
|
|
var overflowTests = []ScanTest{
|
| 307 |
|
|
{"128", &int8Val, 0},
|
| 308 |
|
|
{"32768", &int16Val, 0},
|
| 309 |
|
|
{"-129", &int8Val, 0},
|
| 310 |
|
|
{"-32769", &int16Val, 0},
|
| 311 |
|
|
{"256", &uint8Val, 0},
|
| 312 |
|
|
{"65536", &uint16Val, 0},
|
| 313 |
|
|
{"1e100", &float32Val, 0},
|
| 314 |
|
|
{"1e500", &float64Val, 0},
|
| 315 |
|
|
{"(1e100+0i)", &complex64Val, 0},
|
| 316 |
|
|
{"(1+1e100i)", &complex64Val, 0},
|
| 317 |
|
|
{"(1-1e500i)", &complex128Val, 0},
|
| 318 |
|
|
}
|
| 319 |
|
|
|
| 320 |
|
|
var i, j, k int
|
| 321 |
|
|
var f float64
|
| 322 |
|
|
var s, t string
|
| 323 |
|
|
var c complex128
|
| 324 |
|
|
var x, y Xs
|
| 325 |
|
|
var z IntString
|
| 326 |
|
|
var r1, r2, r3 rune
|
| 327 |
|
|
|
| 328 |
|
|
var multiTests = []ScanfMultiTest{
|
| 329 |
|
|
{"", "", []interface{}{}, []interface{}{}, ""},
|
| 330 |
|
|
{"%d", "23", args(&i), args(23), ""},
|
| 331 |
|
|
{"%2s%3s", "22333", args(&s, &t), args("22", "333"), ""},
|
| 332 |
|
|
{"%2d%3d", "44555", args(&i, &j), args(44, 555), ""},
|
| 333 |
|
|
{"%2d.%3d", "66.777", args(&i, &j), args(66, 777), ""},
|
| 334 |
|
|
{"%d, %d", "23, 18", args(&i, &j), args(23, 18), ""},
|
| 335 |
|
|
{"%3d22%3d", "33322333", args(&i, &j), args(333, 333), ""},
|
| 336 |
|
|
{"%6vX=%3fY", "3+2iX=2.5Y", args(&c, &f), args((3 + 2i), 2.5), ""},
|
| 337 |
|
|
{"%d%s", "123abc", args(&i, &s), args(123, "abc"), ""},
|
| 338 |
|
|
{"%c%c%c", "2\u50c2X", args(&r1, &r2, &r3), args('2', '\u50c2', 'X'), ""},
|
| 339 |
|
|
|
| 340 |
|
|
// Custom scanners.
|
| 341 |
|
|
{"%e%f", "eefffff", args(&x, &y), args(Xs("ee"), Xs("fffff")), ""},
|
| 342 |
|
|
{"%4v%s", "12abcd", args(&z, &s), args(IntString{12, "ab"}, "cd"), ""},
|
| 343 |
|
|
|
| 344 |
|
|
// Errors
|
| 345 |
|
|
{"%t", "23 18", args(&i), nil, "bad verb"},
|
| 346 |
|
|
{"%d %d %d", "23 18", args(&i, &j), args(23, 18), "too few operands"},
|
| 347 |
|
|
{"%d %d", "23 18 27", args(&i, &j, &k), args(23, 18), "too many operands"},
|
| 348 |
|
|
{"%c", "\u0100", args(&int8Val), nil, "overflow"},
|
| 349 |
|
|
{"X%d", "10X", args(&intVal), nil, "input does not match format"},
|
| 350 |
|
|
|
| 351 |
|
|
// Bad UTF-8: should see every byte.
|
| 352 |
|
|
{"%c%c%c", "\xc2X\xc2", args(&r1, &r2, &r3), args(utf8.RuneError, 'X', utf8.RuneError), ""},
|
| 353 |
|
|
}
|
| 354 |
|
|
|
| 355 |
|
|
func testScan(name string, t *testing.T, scan func(r io.Reader, a ...interface{}) (int, error)) {
|
| 356 |
|
|
for _, test := range scanTests {
|
| 357 |
|
|
var r io.Reader
|
| 358 |
|
|
if name == "StringReader" {
|
| 359 |
|
|
r = strings.NewReader(test.text)
|
| 360 |
|
|
} else {
|
| 361 |
|
|
r = newReader(test.text)
|
| 362 |
|
|
}
|
| 363 |
|
|
n, err := scan(r, test.in)
|
| 364 |
|
|
if err != nil {
|
| 365 |
|
|
m := ""
|
| 366 |
|
|
if n > 0 {
|
| 367 |
|
|
m = Sprintf(" (%d fields ok)", n)
|
| 368 |
|
|
}
|
| 369 |
|
|
t.Errorf("%s got error scanning %q: %s%s", name, test.text, err, m)
|
| 370 |
|
|
continue
|
| 371 |
|
|
}
|
| 372 |
|
|
if n != 1 {
|
| 373 |
|
|
t.Errorf("%s count error on entry %q: got %d", name, test.text, n)
|
| 374 |
|
|
continue
|
| 375 |
|
|
}
|
| 376 |
|
|
// The incoming value may be a pointer
|
| 377 |
|
|
v := reflect.ValueOf(test.in)
|
| 378 |
|
|
if p := v; p.Kind() == reflect.Ptr {
|
| 379 |
|
|
v = p.Elem()
|
| 380 |
|
|
}
|
| 381 |
|
|
val := v.Interface()
|
| 382 |
|
|
if !reflect.DeepEqual(val, test.out) {
|
| 383 |
|
|
t.Errorf("%s scanning %q: expected %#v got %#v, type %T", name, test.text, test.out, val, val)
|
| 384 |
|
|
}
|
| 385 |
|
|
}
|
| 386 |
|
|
}
|
| 387 |
|
|
|
| 388 |
|
|
func TestScan(t *testing.T) {
|
| 389 |
|
|
testScan("StringReader", t, Fscan)
|
| 390 |
|
|
}
|
| 391 |
|
|
|
| 392 |
|
|
func TestMyReaderScan(t *testing.T) {
|
| 393 |
|
|
testScan("myStringReader", t, Fscan)
|
| 394 |
|
|
}
|
| 395 |
|
|
|
| 396 |
|
|
func TestScanln(t *testing.T) {
|
| 397 |
|
|
testScan("StringReader", t, Fscanln)
|
| 398 |
|
|
}
|
| 399 |
|
|
|
| 400 |
|
|
func TestMyReaderScanln(t *testing.T) {
|
| 401 |
|
|
testScan("myStringReader", t, Fscanln)
|
| 402 |
|
|
}
|
| 403 |
|
|
|
| 404 |
|
|
func TestScanf(t *testing.T) {
|
| 405 |
|
|
for _, test := range scanfTests {
|
| 406 |
|
|
n, err := Sscanf(test.text, test.format, test.in)
|
| 407 |
|
|
if err != nil {
|
| 408 |
|
|
t.Errorf("got error scanning (%q, %q): %s", test.format, test.text, err)
|
| 409 |
|
|
continue
|
| 410 |
|
|
}
|
| 411 |
|
|
if n != 1 {
|
| 412 |
|
|
t.Errorf("count error on entry (%q, %q): got %d", test.format, test.text, n)
|
| 413 |
|
|
continue
|
| 414 |
|
|
}
|
| 415 |
|
|
// The incoming value may be a pointer
|
| 416 |
|
|
v := reflect.ValueOf(test.in)
|
| 417 |
|
|
if p := v; p.Kind() == reflect.Ptr {
|
| 418 |
|
|
v = p.Elem()
|
| 419 |
|
|
}
|
| 420 |
|
|
val := v.Interface()
|
| 421 |
|
|
if !reflect.DeepEqual(val, test.out) {
|
| 422 |
|
|
t.Errorf("scanning (%q, %q): expected %#v got %#v, type %T", test.format, test.text, test.out, val, val)
|
| 423 |
|
|
}
|
| 424 |
|
|
}
|
| 425 |
|
|
}
|
| 426 |
|
|
|
| 427 |
|
|
func TestScanOverflow(t *testing.T) {
|
| 428 |
|
|
// different machines and different types report errors with different strings.
|
| 429 |
|
|
re := regexp.MustCompile("overflow|too large|out of range|not representable")
|
| 430 |
|
|
for _, test := range overflowTests {
|
| 431 |
|
|
_, err := Sscan(test.text, test.in)
|
| 432 |
|
|
if err == nil {
|
| 433 |
|
|
t.Errorf("expected overflow scanning %q", test.text)
|
| 434 |
|
|
continue
|
| 435 |
|
|
}
|
| 436 |
|
|
if !re.MatchString(err.Error()) {
|
| 437 |
|
|
t.Errorf("expected overflow error scanning %q: %s", test.text, err)
|
| 438 |
|
|
}
|
| 439 |
|
|
}
|
| 440 |
|
|
}
|
| 441 |
|
|
|
| 442 |
|
|
func verifyNaN(str string, t *testing.T) {
|
| 443 |
|
|
var f float64
|
| 444 |
|
|
var f32 float32
|
| 445 |
|
|
var f64 float64
|
| 446 |
|
|
text := str + " " + str + " " + str
|
| 447 |
|
|
n, err := Fscan(strings.NewReader(text), &f, &f32, &f64)
|
| 448 |
|
|
if err != nil {
|
| 449 |
|
|
t.Errorf("got error scanning %q: %s", text, err)
|
| 450 |
|
|
}
|
| 451 |
|
|
if n != 3 {
|
| 452 |
|
|
t.Errorf("count error scanning %q: got %d", text, n)
|
| 453 |
|
|
}
|
| 454 |
|
|
if !math.IsNaN(float64(f)) || !math.IsNaN(float64(f32)) || !math.IsNaN(f64) {
|
| 455 |
|
|
t.Errorf("didn't get NaNs scanning %q: got %g %g %g", text, f, f32, f64)
|
| 456 |
|
|
}
|
| 457 |
|
|
}
|
| 458 |
|
|
|
| 459 |
|
|
func TestNaN(t *testing.T) {
|
| 460 |
|
|
for _, s := range []string{"nan", "NAN", "NaN"} {
|
| 461 |
|
|
verifyNaN(s, t)
|
| 462 |
|
|
}
|
| 463 |
|
|
}
|
| 464 |
|
|
|
| 465 |
|
|
func verifyInf(str string, t *testing.T) {
|
| 466 |
|
|
var f float64
|
| 467 |
|
|
var f32 float32
|
| 468 |
|
|
var f64 float64
|
| 469 |
|
|
text := str + " " + str + " " + str
|
| 470 |
|
|
n, err := Fscan(strings.NewReader(text), &f, &f32, &f64)
|
| 471 |
|
|
if err != nil {
|
| 472 |
|
|
t.Errorf("got error scanning %q: %s", text, err)
|
| 473 |
|
|
}
|
| 474 |
|
|
if n != 3 {
|
| 475 |
|
|
t.Errorf("count error scanning %q: got %d", text, n)
|
| 476 |
|
|
}
|
| 477 |
|
|
sign := 1
|
| 478 |
|
|
if str[0] == '-' {
|
| 479 |
|
|
sign = -1
|
| 480 |
|
|
}
|
| 481 |
|
|
if !math.IsInf(float64(f), sign) || !math.IsInf(float64(f32), sign) || !math.IsInf(f64, sign) {
|
| 482 |
|
|
t.Errorf("didn't get right Infs scanning %q: got %g %g %g", text, f, f32, f64)
|
| 483 |
|
|
}
|
| 484 |
|
|
}
|
| 485 |
|
|
|
| 486 |
|
|
func TestInf(t *testing.T) {
|
| 487 |
|
|
for _, s := range []string{"inf", "+inf", "-inf", "INF", "-INF", "+INF", "Inf", "-Inf", "+Inf"} {
|
| 488 |
|
|
verifyInf(s, t)
|
| 489 |
|
|
}
|
| 490 |
|
|
}
|
| 491 |
|
|
|
| 492 |
|
|
func testScanfMulti(name string, t *testing.T) {
|
| 493 |
|
|
sliceType := reflect.TypeOf(make([]interface{}, 1))
|
| 494 |
|
|
for _, test := range multiTests {
|
| 495 |
|
|
var r io.Reader
|
| 496 |
|
|
if name == "StringReader" {
|
| 497 |
|
|
r = strings.NewReader(test.text)
|
| 498 |
|
|
} else {
|
| 499 |
|
|
r = newReader(test.text)
|
| 500 |
|
|
}
|
| 501 |
|
|
n, err := Fscanf(r, test.format, test.in...)
|
| 502 |
|
|
if err != nil {
|
| 503 |
|
|
if test.err == "" {
|
| 504 |
|
|
t.Errorf("got error scanning (%q, %q): %q", test.format, test.text, err)
|
| 505 |
|
|
} else if strings.Index(err.Error(), test.err) < 0 {
|
| 506 |
|
|
t.Errorf("got wrong error scanning (%q, %q): %q; expected %q", test.format, test.text, err, test.err)
|
| 507 |
|
|
}
|
| 508 |
|
|
continue
|
| 509 |
|
|
}
|
| 510 |
|
|
if test.err != "" {
|
| 511 |
|
|
t.Errorf("expected error %q error scanning (%q, %q)", test.err, test.format, test.text)
|
| 512 |
|
|
}
|
| 513 |
|
|
if n != len(test.out) {
|
| 514 |
|
|
t.Errorf("count error on entry (%q, %q): expected %d got %d", test.format, test.text, len(test.out), n)
|
| 515 |
|
|
continue
|
| 516 |
|
|
}
|
| 517 |
|
|
// Convert the slice of pointers into a slice of values
|
| 518 |
|
|
resultVal := reflect.MakeSlice(sliceType, n, n)
|
| 519 |
|
|
for i := 0; i < n; i++ {
|
| 520 |
|
|
v := reflect.ValueOf(test.in[i]).Elem()
|
| 521 |
|
|
resultVal.Index(i).Set(v)
|
| 522 |
|
|
}
|
| 523 |
|
|
result := resultVal.Interface()
|
| 524 |
|
|
if !reflect.DeepEqual(result, test.out) {
|
| 525 |
|
|
t.Errorf("scanning (%q, %q): expected %#v got %#v", test.format, test.text, test.out, result)
|
| 526 |
|
|
}
|
| 527 |
|
|
}
|
| 528 |
|
|
}
|
| 529 |
|
|
|
| 530 |
|
|
func TestScanfMulti(t *testing.T) {
|
| 531 |
|
|
testScanfMulti("StringReader", t)
|
| 532 |
|
|
}
|
| 533 |
|
|
|
| 534 |
|
|
func TestMyReaderScanfMulti(t *testing.T) {
|
| 535 |
|
|
testScanfMulti("myStringReader", t)
|
| 536 |
|
|
}
|
| 537 |
|
|
|
| 538 |
|
|
func TestScanMultiple(t *testing.T) {
|
| 539 |
|
|
var a int
|
| 540 |
|
|
var s string
|
| 541 |
|
|
n, err := Sscan("123abc", &a, &s)
|
| 542 |
|
|
if n != 2 {
|
| 543 |
|
|
t.Errorf("Sscan count error: expected 2: got %d", n)
|
| 544 |
|
|
}
|
| 545 |
|
|
if err != nil {
|
| 546 |
|
|
t.Errorf("Sscan expected no error; got %s", err)
|
| 547 |
|
|
}
|
| 548 |
|
|
if a != 123 || s != "abc" {
|
| 549 |
|
|
t.Errorf("Sscan wrong values: got (%d %q) expected (123 \"abc\")", a, s)
|
| 550 |
|
|
}
|
| 551 |
|
|
n, err = Sscan("asdf", &s, &a)
|
| 552 |
|
|
if n != 1 {
|
| 553 |
|
|
t.Errorf("Sscan count error: expected 1: got %d", n)
|
| 554 |
|
|
}
|
| 555 |
|
|
if err == nil {
|
| 556 |
|
|
t.Errorf("Sscan expected error; got none: %s", err)
|
| 557 |
|
|
}
|
| 558 |
|
|
if s != "asdf" {
|
| 559 |
|
|
t.Errorf("Sscan wrong values: got %q expected \"asdf\"", s)
|
| 560 |
|
|
}
|
| 561 |
|
|
}
|
| 562 |
|
|
|
| 563 |
|
|
// Empty strings are not valid input when scanning a string.
|
| 564 |
|
|
func TestScanEmpty(t *testing.T) {
|
| 565 |
|
|
var s1, s2 string
|
| 566 |
|
|
n, err := Sscan("abc", &s1, &s2)
|
| 567 |
|
|
if n != 1 {
|
| 568 |
|
|
t.Errorf("Sscan count error: expected 1: got %d", n)
|
| 569 |
|
|
}
|
| 570 |
|
|
if err == nil {
|
| 571 |
|
|
t.Error("Sscan expected error; got none")
|
| 572 |
|
|
}
|
| 573 |
|
|
if s1 != "abc" {
|
| 574 |
|
|
t.Errorf("Sscan wrong values: got %q expected \"abc\"", s1)
|
| 575 |
|
|
}
|
| 576 |
|
|
n, err = Sscan("", &s1, &s2)
|
| 577 |
|
|
if n != 0 {
|
| 578 |
|
|
t.Errorf("Sscan count error: expected 0: got %d", n)
|
| 579 |
|
|
}
|
| 580 |
|
|
if err == nil {
|
| 581 |
|
|
t.Error("Sscan expected error; got none")
|
| 582 |
|
|
}
|
| 583 |
|
|
// Quoted empty string is OK.
|
| 584 |
|
|
n, err = Sscanf(`""`, "%q", &s1)
|
| 585 |
|
|
if n != 1 {
|
| 586 |
|
|
t.Errorf("Sscanf count error: expected 1: got %d", n)
|
| 587 |
|
|
}
|
| 588 |
|
|
if err != nil {
|
| 589 |
|
|
t.Errorf("Sscanf expected no error with quoted string; got %s", err)
|
| 590 |
|
|
}
|
| 591 |
|
|
}
|
| 592 |
|
|
|
| 593 |
|
|
func TestScanNotPointer(t *testing.T) {
|
| 594 |
|
|
r := strings.NewReader("1")
|
| 595 |
|
|
var a int
|
| 596 |
|
|
_, err := Fscan(r, a)
|
| 597 |
|
|
if err == nil {
|
| 598 |
|
|
t.Error("expected error scanning non-pointer")
|
| 599 |
|
|
} else if strings.Index(err.Error(), "pointer") < 0 {
|
| 600 |
|
|
t.Errorf("expected pointer error scanning non-pointer, got: %s", err)
|
| 601 |
|
|
}
|
| 602 |
|
|
}
|
| 603 |
|
|
|
| 604 |
|
|
func TestScanlnNoNewline(t *testing.T) {
|
| 605 |
|
|
var a int
|
| 606 |
|
|
_, err := Sscanln("1 x\n", &a)
|
| 607 |
|
|
if err == nil {
|
| 608 |
|
|
t.Error("expected error scanning string missing newline")
|
| 609 |
|
|
} else if strings.Index(err.Error(), "newline") < 0 {
|
| 610 |
|
|
t.Errorf("expected newline error scanning string missing newline, got: %s", err)
|
| 611 |
|
|
}
|
| 612 |
|
|
}
|
| 613 |
|
|
|
| 614 |
|
|
func TestScanlnWithMiddleNewline(t *testing.T) {
|
| 615 |
|
|
r := strings.NewReader("123\n456\n")
|
| 616 |
|
|
var a, b int
|
| 617 |
|
|
_, err := Fscanln(r, &a, &b)
|
| 618 |
|
|
if err == nil {
|
| 619 |
|
|
t.Error("expected error scanning string with extra newline")
|
| 620 |
|
|
} else if strings.Index(err.Error(), "newline") < 0 {
|
| 621 |
|
|
t.Errorf("expected newline error scanning string with extra newline, got: %s", err)
|
| 622 |
|
|
}
|
| 623 |
|
|
}
|
| 624 |
|
|
|
| 625 |
|
|
// Special Reader that counts reads at end of file.
|
| 626 |
|
|
type eofCounter struct {
|
| 627 |
|
|
reader *strings.Reader
|
| 628 |
|
|
eofCount int
|
| 629 |
|
|
}
|
| 630 |
|
|
|
| 631 |
|
|
func (ec *eofCounter) Read(b []byte) (n int, err error) {
|
| 632 |
|
|
n, err = ec.reader.Read(b)
|
| 633 |
|
|
if n == 0 {
|
| 634 |
|
|
ec.eofCount++
|
| 635 |
|
|
}
|
| 636 |
|
|
return
|
| 637 |
|
|
}
|
| 638 |
|
|
|
| 639 |
|
|
// Verify that when we scan, we see at most EOF once per call to a Scan function,
|
| 640 |
|
|
// and then only when it's really an EOF
|
| 641 |
|
|
func TestEOF(t *testing.T) {
|
| 642 |
|
|
ec := &eofCounter{strings.NewReader("123\n"), 0}
|
| 643 |
|
|
var a int
|
| 644 |
|
|
n, err := Fscanln(ec, &a)
|
| 645 |
|
|
if err != nil {
|
| 646 |
|
|
t.Error("unexpected error", err)
|
| 647 |
|
|
}
|
| 648 |
|
|
if n != 1 {
|
| 649 |
|
|
t.Error("expected to scan one item, got", n)
|
| 650 |
|
|
}
|
| 651 |
|
|
if ec.eofCount != 0 {
|
| 652 |
|
|
t.Error("expected zero EOFs", ec.eofCount)
|
| 653 |
|
|
ec.eofCount = 0 // reset for next test
|
| 654 |
|
|
}
|
| 655 |
|
|
n, err = Fscanln(ec, &a)
|
| 656 |
|
|
if err == nil {
|
| 657 |
|
|
t.Error("expected error scanning empty string")
|
| 658 |
|
|
}
|
| 659 |
|
|
if n != 0 {
|
| 660 |
|
|
t.Error("expected to scan zero items, got", n)
|
| 661 |
|
|
}
|
| 662 |
|
|
if ec.eofCount != 1 {
|
| 663 |
|
|
t.Error("expected one EOF, got", ec.eofCount)
|
| 664 |
|
|
}
|
| 665 |
|
|
}
|
| 666 |
|
|
|
| 667 |
|
|
// Verify that we see an EOF error if we run out of input.
|
| 668 |
|
|
// This was a buglet: we used to get "expected integer".
|
| 669 |
|
|
func TestEOFAtEndOfInput(t *testing.T) {
|
| 670 |
|
|
var i, j int
|
| 671 |
|
|
n, err := Sscanf("23", "%d %d", &i, &j)
|
| 672 |
|
|
if n != 1 || i != 23 {
|
| 673 |
|
|
t.Errorf("Sscanf expected one value of 23; got %d %d", n, i)
|
| 674 |
|
|
}
|
| 675 |
|
|
if err != io.EOF {
|
| 676 |
|
|
t.Errorf("Sscanf expected EOF; got %q", err)
|
| 677 |
|
|
}
|
| 678 |
|
|
n, err = Sscan("234", &i, &j)
|
| 679 |
|
|
if n != 1 || i != 234 {
|
| 680 |
|
|
t.Errorf("Sscan expected one value of 234; got %d %d", n, i)
|
| 681 |
|
|
}
|
| 682 |
|
|
if err != io.EOF {
|
| 683 |
|
|
t.Errorf("Sscan expected EOF; got %q", err)
|
| 684 |
|
|
}
|
| 685 |
|
|
// Trailing space is tougher.
|
| 686 |
|
|
n, err = Sscan("234 ", &i, &j)
|
| 687 |
|
|
if n != 1 || i != 234 {
|
| 688 |
|
|
t.Errorf("Sscan expected one value of 234; got %d %d", n, i)
|
| 689 |
|
|
}
|
| 690 |
|
|
if err != io.EOF {
|
| 691 |
|
|
t.Errorf("Sscan expected EOF; got %q", err)
|
| 692 |
|
|
}
|
| 693 |
|
|
}
|
| 694 |
|
|
|
| 695 |
|
|
var eofTests = []struct {
|
| 696 |
|
|
format string
|
| 697 |
|
|
v interface{}
|
| 698 |
|
|
}{
|
| 699 |
|
|
{"%s", &stringVal},
|
| 700 |
|
|
{"%q", &stringVal},
|
| 701 |
|
|
{"%x", &stringVal},
|
| 702 |
|
|
{"%v", &stringVal},
|
| 703 |
|
|
{"%v", &bytesVal},
|
| 704 |
|
|
{"%v", &intVal},
|
| 705 |
|
|
{"%v", &uintVal},
|
| 706 |
|
|
{"%v", &boolVal},
|
| 707 |
|
|
{"%v", &float32Val},
|
| 708 |
|
|
{"%v", &complex64Val},
|
| 709 |
|
|
{"%v", &renamedStringVal},
|
| 710 |
|
|
{"%v", &renamedBytesVal},
|
| 711 |
|
|
{"%v", &renamedIntVal},
|
| 712 |
|
|
{"%v", &renamedUintVal},
|
| 713 |
|
|
{"%v", &renamedBoolVal},
|
| 714 |
|
|
{"%v", &renamedFloat32Val},
|
| 715 |
|
|
{"%v", &renamedComplex64Val},
|
| 716 |
|
|
}
|
| 717 |
|
|
|
| 718 |
|
|
func TestEOFAllTypes(t *testing.T) {
|
| 719 |
|
|
for i, test := range eofTests {
|
| 720 |
|
|
if _, err := Sscanf("", test.format, test.v); err != io.EOF {
|
| 721 |
|
|
t.Errorf("#%d: %s %T not eof on empty string: %s", i, test.format, test.v, err)
|
| 722 |
|
|
}
|
| 723 |
|
|
if _, err := Sscanf(" ", test.format, test.v); err != io.EOF {
|
| 724 |
|
|
t.Errorf("#%d: %s %T not eof on trailing blanks: %s", i, test.format, test.v, err)
|
| 725 |
|
|
}
|
| 726 |
|
|
}
|
| 727 |
|
|
}
|
| 728 |
|
|
|
| 729 |
|
|
// Verify that, at least when using bufio, successive calls to Fscan do not lose runes.
|
| 730 |
|
|
func TestUnreadRuneWithBufio(t *testing.T) {
|
| 731 |
|
|
r := bufio.NewReader(strings.NewReader("123αb"))
|
| 732 |
|
|
var i int
|
| 733 |
|
|
var a string
|
| 734 |
|
|
n, err := Fscanf(r, "%d", &i)
|
| 735 |
|
|
if n != 1 || err != nil {
|
| 736 |
|
|
t.Errorf("reading int expected one item, no errors; got %d %q", n, err)
|
| 737 |
|
|
}
|
| 738 |
|
|
if i != 123 {
|
| 739 |
|
|
t.Errorf("expected 123; got %d", i)
|
| 740 |
|
|
}
|
| 741 |
|
|
n, err = Fscanf(r, "%s", &a)
|
| 742 |
|
|
if n != 1 || err != nil {
|
| 743 |
|
|
t.Errorf("reading string expected one item, no errors; got %d %q", n, err)
|
| 744 |
|
|
}
|
| 745 |
|
|
if a != "αb" {
|
| 746 |
|
|
t.Errorf("expected αb; got %q", a)
|
| 747 |
|
|
}
|
| 748 |
|
|
}
|
| 749 |
|
|
|
| 750 |
|
|
type TwoLines string
|
| 751 |
|
|
|
| 752 |
|
|
// Attempt to read two lines into the object. Scanln should prevent this
|
| 753 |
|
|
// because it stops at newline; Scan and Scanf should be fine.
|
| 754 |
|
|
func (t *TwoLines) Scan(state ScanState, verb rune) error {
|
| 755 |
|
|
chars := make([]rune, 0, 100)
|
| 756 |
|
|
for nlCount := 0; nlCount < 2; {
|
| 757 |
|
|
c, _, err := state.ReadRune()
|
| 758 |
|
|
if err != nil {
|
| 759 |
|
|
return err
|
| 760 |
|
|
}
|
| 761 |
|
|
chars = append(chars, c)
|
| 762 |
|
|
if c == '\n' {
|
| 763 |
|
|
nlCount++
|
| 764 |
|
|
}
|
| 765 |
|
|
}
|
| 766 |
|
|
*t = TwoLines(string(chars))
|
| 767 |
|
|
return nil
|
| 768 |
|
|
}
|
| 769 |
|
|
|
| 770 |
|
|
func TestMultiLine(t *testing.T) {
|
| 771 |
|
|
input := "abc\ndef\n"
|
| 772 |
|
|
// Sscan should work
|
| 773 |
|
|
var tscan TwoLines
|
| 774 |
|
|
n, err := Sscan(input, &tscan)
|
| 775 |
|
|
if n != 1 {
|
| 776 |
|
|
t.Errorf("Sscan: expected 1 item; got %d", n)
|
| 777 |
|
|
}
|
| 778 |
|
|
if err != nil {
|
| 779 |
|
|
t.Errorf("Sscan: expected no error; got %s", err)
|
| 780 |
|
|
}
|
| 781 |
|
|
if string(tscan) != input {
|
| 782 |
|
|
t.Errorf("Sscan: expected %q; got %q", input, tscan)
|
| 783 |
|
|
}
|
| 784 |
|
|
// Sscanf should work
|
| 785 |
|
|
var tscanf TwoLines
|
| 786 |
|
|
n, err = Sscanf(input, "%s", &tscanf)
|
| 787 |
|
|
if n != 1 {
|
| 788 |
|
|
t.Errorf("Sscanf: expected 1 item; got %d", n)
|
| 789 |
|
|
}
|
| 790 |
|
|
if err != nil {
|
| 791 |
|
|
t.Errorf("Sscanf: expected no error; got %s", err)
|
| 792 |
|
|
}
|
| 793 |
|
|
if string(tscanf) != input {
|
| 794 |
|
|
t.Errorf("Sscanf: expected %q; got %q", input, tscanf)
|
| 795 |
|
|
}
|
| 796 |
|
|
// Sscanln should not work
|
| 797 |
|
|
var tscanln TwoLines
|
| 798 |
|
|
n, err = Sscanln(input, &tscanln)
|
| 799 |
|
|
if n != 0 {
|
| 800 |
|
|
t.Errorf("Sscanln: expected 0 items; got %d: %q", n, tscanln)
|
| 801 |
|
|
}
|
| 802 |
|
|
if err == nil {
|
| 803 |
|
|
t.Error("Sscanln: expected error; got none")
|
| 804 |
|
|
} else if err != io.ErrUnexpectedEOF {
|
| 805 |
|
|
t.Errorf("Sscanln: expected io.ErrUnexpectedEOF (ha!); got %s", err)
|
| 806 |
|
|
}
|
| 807 |
|
|
}
|
| 808 |
|
|
|
| 809 |
|
|
// RecursiveInt accepts an string matching %d.%d.%d....
|
| 810 |
|
|
// and parses it into a linked list.
|
| 811 |
|
|
// It allows us to benchmark recursive descent style scanners.
|
| 812 |
|
|
type RecursiveInt struct {
|
| 813 |
|
|
i int
|
| 814 |
|
|
next *RecursiveInt
|
| 815 |
|
|
}
|
| 816 |
|
|
|
| 817 |
|
|
func (r *RecursiveInt) Scan(state ScanState, verb rune) (err error) {
|
| 818 |
|
|
_, err = Fscan(state, &r.i)
|
| 819 |
|
|
if err != nil {
|
| 820 |
|
|
return
|
| 821 |
|
|
}
|
| 822 |
|
|
next := new(RecursiveInt)
|
| 823 |
|
|
_, err = Fscanf(state, ".%v", next)
|
| 824 |
|
|
if err != nil {
|
| 825 |
|
|
if err == errors.New("input does not match format") || err == io.ErrUnexpectedEOF {
|
| 826 |
|
|
err = nil
|
| 827 |
|
|
}
|
| 828 |
|
|
return
|
| 829 |
|
|
}
|
| 830 |
|
|
r.next = next
|
| 831 |
|
|
return
|
| 832 |
|
|
}
|
| 833 |
|
|
|
| 834 |
|
|
// Perform the same scanning task as RecursiveInt.Scan
|
| 835 |
|
|
// but without recurring through scanner, so we can compare
|
| 836 |
|
|
// performance more directly.
|
| 837 |
|
|
func scanInts(r *RecursiveInt, b *bytes.Buffer) (err error) {
|
| 838 |
|
|
r.next = nil
|
| 839 |
|
|
_, err = Fscan(b, &r.i)
|
| 840 |
|
|
if err != nil {
|
| 841 |
|
|
return
|
| 842 |
|
|
}
|
| 843 |
|
|
c, _, err := b.ReadRune()
|
| 844 |
|
|
if err != nil {
|
| 845 |
|
|
if err == io.EOF {
|
| 846 |
|
|
err = nil
|
| 847 |
|
|
}
|
| 848 |
|
|
return
|
| 849 |
|
|
}
|
| 850 |
|
|
if c != '.' {
|
| 851 |
|
|
return
|
| 852 |
|
|
}
|
| 853 |
|
|
next := new(RecursiveInt)
|
| 854 |
|
|
err = scanInts(next, b)
|
| 855 |
|
|
if err == nil {
|
| 856 |
|
|
r.next = next
|
| 857 |
|
|
}
|
| 858 |
|
|
return
|
| 859 |
|
|
}
|
| 860 |
|
|
|
| 861 |
|
|
func makeInts(n int) []byte {
|
| 862 |
|
|
var buf bytes.Buffer
|
| 863 |
|
|
Fprintf(&buf, "1")
|
| 864 |
|
|
for i := 1; i < n; i++ {
|
| 865 |
|
|
Fprintf(&buf, ".%d", i+1)
|
| 866 |
|
|
}
|
| 867 |
|
|
return buf.Bytes()
|
| 868 |
|
|
}
|
| 869 |
|
|
|
| 870 |
|
|
func TestScanInts(t *testing.T) {
|
| 871 |
|
|
testScanInts(t, scanInts)
|
| 872 |
|
|
testScanInts(t, func(r *RecursiveInt, b *bytes.Buffer) (err error) {
|
| 873 |
|
|
_, err = Fscan(b, r)
|
| 874 |
|
|
return
|
| 875 |
|
|
})
|
| 876 |
|
|
}
|
| 877 |
|
|
|
| 878 |
|
|
// 800 is small enough to not overflow the stack when using gccgo on a
|
| 879 |
|
|
// platform that does not support split stack.
|
| 880 |
|
|
const intCount = 800
|
| 881 |
|
|
|
| 882 |
|
|
func testScanInts(t *testing.T, scan func(*RecursiveInt, *bytes.Buffer) error) {
|
| 883 |
|
|
r := new(RecursiveInt)
|
| 884 |
|
|
ints := makeInts(intCount)
|
| 885 |
|
|
buf := bytes.NewBuffer(ints)
|
| 886 |
|
|
err := scan(r, buf)
|
| 887 |
|
|
if err != nil {
|
| 888 |
|
|
t.Error("unexpected error", err)
|
| 889 |
|
|
}
|
| 890 |
|
|
i := 1
|
| 891 |
|
|
for ; r != nil; r = r.next {
|
| 892 |
|
|
if r.i != i {
|
| 893 |
|
|
t.Fatalf("bad scan: expected %d got %d", i, r.i)
|
| 894 |
|
|
}
|
| 895 |
|
|
i++
|
| 896 |
|
|
}
|
| 897 |
|
|
if i-1 != intCount {
|
| 898 |
|
|
t.Fatalf("bad scan count: expected %d got %d", intCount, i-1)
|
| 899 |
|
|
}
|
| 900 |
|
|
}
|
| 901 |
|
|
|
| 902 |
|
|
func BenchmarkScanInts(b *testing.B) {
|
| 903 |
|
|
b.ResetTimer()
|
| 904 |
|
|
ints := makeInts(intCount)
|
| 905 |
|
|
var r RecursiveInt
|
| 906 |
|
|
for i := b.N - 1; i >= 0; i-- {
|
| 907 |
|
|
buf := bytes.NewBuffer(ints)
|
| 908 |
|
|
b.StartTimer()
|
| 909 |
|
|
scanInts(&r, buf)
|
| 910 |
|
|
b.StopTimer()
|
| 911 |
|
|
}
|
| 912 |
|
|
}
|
| 913 |
|
|
|
| 914 |
|
|
func BenchmarkScanRecursiveInt(b *testing.B) {
|
| 915 |
|
|
b.ResetTimer()
|
| 916 |
|
|
ints := makeInts(intCount)
|
| 917 |
|
|
var r RecursiveInt
|
| 918 |
|
|
for i := b.N - 1; i >= 0; i-- {
|
| 919 |
|
|
buf := bytes.NewBuffer(ints)
|
| 920 |
|
|
b.StartTimer()
|
| 921 |
|
|
Fscan(buf, &r)
|
| 922 |
|
|
b.StopTimer()
|
| 923 |
|
|
}
|
| 924 |
|
|
}
|