package zero import ( "encoding/json" "math" "strconv" "testing" ) var ( intJSON = []byte(`12345`) intStringJSON = []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 si Int err = json.Unmarshal(intStringJSON, &si) maybePanic(err) assertInt(t, si, "int string json") var ni Int err = json.Unmarshal(nullIntJSON, &ni) maybePanic(err) assertInt(t, ni, "sql.NullInt64 json") var bi Int err = json.Unmarshal(floatBlankJSON, &bi) maybePanic(err) assertNullInt(t, bi, "blank json string") 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") } }