Merge pull request #23 from ymt2/feat/implement-TextUnmarshaler

Author: iwata

bps/serial.go

@@ -3,11 +3,16 @@ package bps
 import (
 	"database/sql"
 	"database/sql/driver"
+	"encoding"
 	"errors"
 )
 
-// make sure a *BPS implements the sql.Scanner interface.
-var _ sql.Scanner = (*BPS)(nil)
+// make sure that the *BPS implements some interfaces.
+var _ interface {
+	sql.Scanner
+	driver.Valuer
+	encoding.TextUnmarshaler
+} = (*BPS)(nil)
 
 // Scan implements the sql.Scanner interface for database deserialization.
 func (b *BPS) Scan(value interface{}) error {
@@ -52,10 +57,23 @@ func (b *BPS) Scan(value interface{}) error {
 	return errors.New("BPS.Scan: invalid type, supporting only integer or string")
 }
 
-// make sure a *BPS implements the driver.Valuer interface.
-var _ driver.Valuer = (*BPS)(nil)
-
 // Value implements the driver.Valuer interface for database serialization.
 func (b *BPS) Value() (driver.Value, error) {
 	return b.String(), nil
 }
+
+// UnmarshalText implements the encoding.TextUnmarshaler interface.
+func (b *BPS) UnmarshalText(text []byte) error {
+	v := string(text)
+	if v == "" {
+		return errors.New("BPS.UnmarshalText: no data")
+	}
+
+	n, err := NewFromString(v)
+	if err != nil {
+		return err
+	}
+	b.value = n.value
+
+	return nil
+}

bps/serial_test.go

@@ -118,3 +118,57 @@ func TestBPS_Scan(t *testing.T) {
 		})
 	}
 }
+
+func TestBPS_UnmarshalText(t *testing.T) {
+	t.Parallel()
+	tests := map[string]struct {
+		value   string
+		want    *bps.BPS
+		wantErr bool
+	}{
+		"If value is valid string with the integer part abbreviating, it should set value via NewFromString": {
+			".15",
+			bps.NewFromPercentage(15),
+			false,
+		},
+		"If value is valid string, it should set value via NewFromString": {
+			"1.15",
+			bps.NewFromPercentage(115),
+			false,
+		},
+		"If value is negative, it should set value via NewFromString": {
+			"-123.456",
+			bps.NewFromBasisPoint(-1234560),
+			false,
+		},
+		"If value is zero, it should set value via NewFromString": {
+			"0",
+			bps.NewFromAmount(0),
+			false,
+		},
+		"If value is invalid string, it should return an error": {
+			"a15",
+			&bps.BPS{},
+			true,
+		},
+		"If value is empty, it should return an error": {
+			"",
+			&bps.BPS{},
+			true,
+		},
+	}
+	for name, tt := range tests {
+		name := name
+		tt := tt
+		t.Run(name, func(t *testing.T) {
+			t.Parallel()
+			b := &bps.BPS{}
+			if err := b.UnmarshalText([]byte(tt.value)); (err != nil) != tt.wantErr {
+				t.Errorf("BPS.UnmarshalText() error = %v, wantErr %v", err, tt.wantErr)
+			}
+			if !reflect.DeepEqual(b, tt.want) {
+				t.Errorf("BPS.Value() = %v, want %v", b, tt.want)
+			}
+		})
+	}
+}