db: strconv.ParseInt specify base 10

closes #99
This commit is contained in:
Eric Chiang 2015-12-09 14:47:21 -08:00
parent cec3cd4451
commit 9f3bf0d160
2 changed files with 32 additions and 1 deletions

View file

@ -52,7 +52,7 @@ func parseToken(token string) (int64, []byte, error) {
if len(parts) != 2 {
return -1, nil, refresh.ErrorInvalidToken
}
id, err := strconv.ParseInt(parts[0], 0, 64)
id, err := strconv.ParseInt(parts[0], 10, 64)
if err != nil {
return -1, nil, refresh.ErrorInvalidToken
}

31
db/refresh_test.go Normal file
View file

@ -0,0 +1,31 @@
package db
import (
"bytes"
"testing"
)
func TestBuildAndParseToken(t *testing.T) {
tests := []struct {
id int64
payload []byte
}{
{11111, []byte("may the force be with you")},
{123213, []byte("If we can hit that bullseye the rest of the dominoes will fall like a house of cards, checkmate!")},
{1, []byte{0xd3, 0x22, 0xa8, 0x44, 0x34, 0x94, 0xd8}},
}
for i, tt := range tests {
id, payload, err := parseToken(buildToken(tt.id, tt.payload))
if err != nil {
t.Errorf("case %d: failed to parse token: %v", i, err)
continue
}
if tt.id != id {
t.Errorf("case %d: want id=%d, got id=%d", i, tt.id, id)
}
if bytes.Compare(tt.payload, payload) != 0 {
t.Errorf("case %d: want payload=%x, got payload=%x", i, tt.payload, payload)
}
}
}