208 lines
4.4 KiB
Go
208 lines
4.4 KiB
Go
|
package zero
|
||
|
|
||
|
import (
|
||
|
"encoding/json"
|
||
|
"math"
|
||
|
"strconv"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
intJSON = []byte(`12345`)
|
||
|
nullIntJSON = []byte(`{"Int64":12345,"Valid":true}`)
|
||
|
zeroJSON = []byte(`0`)
|
||
|
)
|
||
|
|
||
|
func TestIntFrom(t *testing.T) {
|
||
|
i := IntFrom(12345)
|
||
|
assertInt(t, i, "IntFrom()")
|
||
|
|
||
|
zero := IntFrom(0)
|
||
|
if zero.Valid {
|
||
|
t.Error("IntFrom(0)", "is valid, but should be invalid")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestIntFromPtr(t *testing.T) {
|
||
|
n := int64(12345)
|
||
|
iptr := &n
|
||
|
i := IntFromPtr(iptr)
|
||
|
assertInt(t, i, "IntFromPtr()")
|
||
|
|
||
|
null := IntFromPtr(nil)
|
||
|
assertNullInt(t, null, "IntFromPtr(nil)")
|
||
|
}
|
||
|
|
||
|
func TestUnmarshalInt(t *testing.T) {
|
||
|
var i Int
|
||
|
err := json.Unmarshal(intJSON, &i)
|
||
|
maybePanic(err)
|
||
|
assertInt(t, i, "int json")
|
||
|
|
||
|
var ni Int
|
||
|
err = json.Unmarshal(nullIntJSON, &ni)
|
||
|
maybePanic(err)
|
||
|
assertInt(t, ni, "sql.NullInt64 json")
|
||
|
|
||
|
var zero Int
|
||
|
err = json.Unmarshal(zeroJSON, &zero)
|
||
|
maybePanic(err)
|
||
|
assertNullInt(t, zero, "zero json")
|
||
|
|
||
|
var null Int
|
||
|
err = json.Unmarshal(nullJSON, &null)
|
||
|
maybePanic(err)
|
||
|
assertNullInt(t, null, "null json")
|
||
|
|
||
|
var badType Int
|
||
|
err = json.Unmarshal(boolJSON, &badType)
|
||
|
if err == nil {
|
||
|
panic("err should not be nil")
|
||
|
}
|
||
|
assertNullInt(t, badType, "wrong type json")
|
||
|
|
||
|
var invalid Int
|
||
|
err = invalid.UnmarshalJSON(invalidJSON)
|
||
|
if _, ok := err.(*json.SyntaxError); !ok {
|
||
|
t.Errorf("expected json.SyntaxError, not %T", err)
|
||
|
}
|
||
|
assertNullInt(t, invalid, "invalid json")
|
||
|
}
|
||
|
|
||
|
func TestUnmarshalNonIntegerNumber(t *testing.T) {
|
||
|
var i Int
|
||
|
err := json.Unmarshal(floatJSON, &i)
|
||
|
if err == nil {
|
||
|
panic("err should be present; non-integer number coerced to int")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestUnmarshalInt64Overflow(t *testing.T) {
|
||
|
int64Overflow := uint64(math.MaxInt64)
|
||
|
|
||
|
// Max int64 should decode successfully
|
||
|
var i Int
|
||
|
err := json.Unmarshal([]byte(strconv.FormatUint(int64Overflow, 10)), &i)
|
||
|
maybePanic(err)
|
||
|
|
||
|
// Attempt to overflow
|
||
|
int64Overflow++
|
||
|
err = json.Unmarshal([]byte(strconv.FormatUint(int64Overflow, 10)), &i)
|
||
|
if err == nil {
|
||
|
panic("err should be present; decoded value overflows int64")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestTextUnmarshalInt(t *testing.T) {
|
||
|
var i Int
|
||
|
err := i.UnmarshalText([]byte("12345"))
|
||
|
maybePanic(err)
|
||
|
assertInt(t, i, "UnmarshalText() int")
|
||
|
|
||
|
var zero Int
|
||
|
err = zero.UnmarshalText([]byte("0"))
|
||
|
maybePanic(err)
|
||
|
assertNullInt(t, zero, "UnmarshalText() zero int")
|
||
|
|
||
|
var blank Int
|
||
|
err = blank.UnmarshalText([]byte(""))
|
||
|
maybePanic(err)
|
||
|
assertNullInt(t, blank, "UnmarshalText() empty int")
|
||
|
|
||
|
var null Int
|
||
|
err = null.UnmarshalText([]byte("null"))
|
||
|
maybePanic(err)
|
||
|
assertNullInt(t, null, `UnmarshalText() "null"`)
|
||
|
}
|
||
|
|
||
|
func TestMarshalInt(t *testing.T) {
|
||
|
i := IntFrom(12345)
|
||
|
data, err := json.Marshal(i)
|
||
|
maybePanic(err)
|
||
|
assertJSONEquals(t, data, "12345", "non-empty json marshal")
|
||
|
|
||
|
// invalid values should be encoded as 0
|
||
|
null := NewInt(0, false)
|
||
|
data, err = json.Marshal(null)
|
||
|
maybePanic(err)
|
||
|
assertJSONEquals(t, data, "0", "null json marshal")
|
||
|
}
|
||
|
|
||
|
func TestMarshalIntText(t *testing.T) {
|
||
|
i := IntFrom(12345)
|
||
|
data, err := i.MarshalText()
|
||
|
maybePanic(err)
|
||
|
assertJSONEquals(t, data, "12345", "non-empty text marshal")
|
||
|
|
||
|
// invalid values should be encoded as zero
|
||
|
null := NewInt(0, false)
|
||
|
data, err = null.MarshalText()
|
||
|
maybePanic(err)
|
||
|
assertJSONEquals(t, data, "0", "null text marshal")
|
||
|
}
|
||
|
|
||
|
func TestIntPointer(t *testing.T) {
|
||
|
i := IntFrom(12345)
|
||
|
ptr := i.Ptr()
|
||
|
if *ptr != 12345 {
|
||
|
t.Errorf("bad %s int: %#v ≠ %d\n", "pointer", ptr, 12345)
|
||
|
}
|
||
|
|
||
|
null := NewInt(0, false)
|
||
|
ptr = null.Ptr()
|
||
|
if ptr != nil {
|
||
|
t.Errorf("bad %s int: %#v ≠ %s\n", "nil pointer", ptr, "nil")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestIntIsZero(t *testing.T) {
|
||
|
i := IntFrom(12345)
|
||
|
if i.IsZero() {
|
||
|
t.Errorf("IsZero() should be false")
|
||
|
}
|
||
|
|
||
|
null := NewInt(0, false)
|
||
|
if !null.IsZero() {
|
||
|
t.Errorf("IsZero() should be true")
|
||
|
}
|
||
|
|
||
|
zero := NewInt(0, true)
|
||
|
if !zero.IsZero() {
|
||
|
t.Errorf("IsZero() should be true")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestIntScan(t *testing.T) {
|
||
|
var i Int
|
||
|
err := i.Scan(12345)
|
||
|
maybePanic(err)
|
||
|
assertInt(t, i, "scanned int")
|
||
|
|
||
|
var null Int
|
||
|
err = null.Scan(nil)
|
||
|
maybePanic(err)
|
||
|
assertNullInt(t, null, "scanned null")
|
||
|
}
|
||
|
|
||
|
func TestIntSetValid(t *testing.T) {
|
||
|
change := NewInt(0, false)
|
||
|
assertNullInt(t, change, "SetValid()")
|
||
|
change.SetValid(12345)
|
||
|
assertInt(t, change, "SetValid()")
|
||
|
}
|
||
|
|
||
|
func assertInt(t *testing.T, i Int, from string) {
|
||
|
if i.Int64 != 12345 {
|
||
|
t.Errorf("bad %s int: %d ≠ %d\n", from, i.Int64, 12345)
|
||
|
}
|
||
|
if !i.Valid {
|
||
|
t.Error(from, "is invalid, but should be valid")
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func assertNullInt(t *testing.T, i Int, from string) {
|
||
|
if i.Valid {
|
||
|
t.Error(from, "is valid, but should be invalid")
|
||
|
}
|
||
|
}
|