bbb7b84bb6
* Add Scan and Value methods for non sql.Int64 and sql.Float64 types * Change NullInt to use int in the code opposed to int64 * Rename Float to Float64 * Add support for all missing key int and float types: float32, int8, int16, int32, int64, uint8, uint16, uint32, uint64 * Update README to include new additions/changes
178 lines
3.7 KiB
Go
178 lines
3.7 KiB
Go
package null
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
intJSON = []byte(`12345`)
|
|
nullIntJSON = []byte(`{"Int":12345,"Valid":true}`)
|
|
)
|
|
|
|
func TestIntFrom(t *testing.T) {
|
|
i := IntFrom(12345)
|
|
assertInt(t, i, "IntFrom()")
|
|
|
|
zero := IntFrom(0)
|
|
if !zero.Valid {
|
|
t.Error("IntFrom(0)", "is invalid, but should be valid")
|
|
}
|
|
}
|
|
|
|
func TestIntFromPtr(t *testing.T) {
|
|
n := int(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, "sq.NullInt 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(float64JSON, &i)
|
|
if err == nil {
|
|
panic("err should be present; non-integer number coerced to int")
|
|
}
|
|
}
|
|
|
|
func TestTextUnmarshalInt(t *testing.T) {
|
|
var i Int
|
|
err := i.UnmarshalText([]byte("12345"))
|
|
maybePanic(err)
|
|
assertInt(t, i, "UnmarshalText() 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 null
|
|
null := NewInt(0, false)
|
|
data, err = json.Marshal(null)
|
|
maybePanic(err)
|
|
assertJSONEquals(t, data, "null", "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 null
|
|
null := NewInt(0, false)
|
|
data, err = null.MarshalText()
|
|
maybePanic(err)
|
|
assertJSONEquals(t, data, "", "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 false")
|
|
}
|
|
}
|
|
|
|
func TestIntSetValid(t *testing.T) {
|
|
change := NewInt(0, false)
|
|
assertNullInt(t, change, "SetValid()")
|
|
change.SetValid(12345)
|
|
assertInt(t, change, "SetValid()")
|
|
}
|
|
|
|
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 assertInt(t *testing.T, i Int, from string) {
|
|
if i.Int != 12345 {
|
|
t.Errorf("bad %s int: %d ≠ %d\n", from, i.Int, 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")
|
|
}
|
|
}
|