X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/95159e5585762c06c654945070ba54262b7dcec9..3e4a52ed34b130743164d6d77f31a1fc67f161c8:/hotline/field_test.go

diff --git a/hotline/field_test.go b/hotline/field_test.go
index 317c2d9..213d9c3 100644
--- a/hotline/field_test.go
+++ b/hotline/field_test.go
@@ -90,12 +90,142 @@ func Test_fieldScanner(t *testing.T) {
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
-			gotAdvance, gotToken, err := fieldScanner(tt.args.data, tt.args.in1)
-			if !tt.wantErr(t, err, fmt.Sprintf("fieldScanner(%v, %v)", tt.args.data, tt.args.in1)) {
+			gotAdvance, gotToken, err := FieldScanner(tt.args.data, tt.args.in1)
+			if !tt.wantErr(t, err, fmt.Sprintf("FieldScanner(%v, %v)", tt.args.data, tt.args.in1)) {
 				return
 			}
-			assert.Equalf(t, tt.wantAdvance, gotAdvance, "fieldScanner(%v, %v)", tt.args.data, tt.args.in1)
-			assert.Equalf(t, tt.wantToken, gotToken, "fieldScanner(%v, %v)", tt.args.data, tt.args.in1)
+			assert.Equalf(t, tt.wantAdvance, gotAdvance, "FieldScanner(%v, %v)", tt.args.data, tt.args.in1)
+			assert.Equalf(t, tt.wantToken, gotToken, "FieldScanner(%v, %v)", tt.args.data, tt.args.in1)
+		})
+	}
+}
+
+func TestField_Read(t *testing.T) {
+	type fields struct {
+		ID         [2]byte
+		FieldSize  [2]byte
+		Data       []byte
+		readOffset int
+	}
+	type args struct {
+		p []byte
+	}
+	tests := []struct {
+		name      string
+		fields    fields
+		args      args
+		want      int
+		wantErr   assert.ErrorAssertionFunc
+		wantBytes []byte
+	}{
+		{
+			name: "returns field bytes",
+			fields: fields{
+				ID:        [2]byte{0x00, 0x62},
+				FieldSize: [2]byte{0x00, 0x03},
+				Data:      []byte("hai!"),
+			},
+			args: args{
+				p: make([]byte, 512),
+			},
+			want:    8,
+			wantErr: assert.NoError,
+			wantBytes: []byte{
+				0x00, 0x62,
+				0x00, 0x03,
+				0x68, 0x61, 0x69, 0x21,
+			},
+		},
+		{
+			name: "returns field bytes from readOffset",
+			fields: fields{
+				ID:         [2]byte{0x00, 0x62},
+				FieldSize:  [2]byte{0x00, 0x03},
+				Data:       []byte("hai!"),
+				readOffset: 4,
+			},
+			args: args{
+				p: make([]byte, 512),
+			},
+			want:    4,
+			wantErr: assert.NoError,
+			wantBytes: []byte{
+				0x68, 0x61, 0x69, 0x21,
+			},
+		},
+		{
+			name: "returns io.EOF when all bytes read",
+			fields: fields{
+				ID:         [2]byte{0x00, 0x62},
+				FieldSize:  [2]byte{0x00, 0x03},
+				Data:       []byte("hai!"),
+				readOffset: 8,
+			},
+			args: args{
+				p: make([]byte, 512),
+			},
+			want:      0,
+			wantErr:   assert.Error,
+			wantBytes: []byte{},
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			f := &Field{
+				Type:       tt.fields.ID,
+				FieldSize:  tt.fields.FieldSize,
+				Data:       tt.fields.Data,
+				readOffset: tt.fields.readOffset,
+			}
+			got, err := f.Read(tt.args.p)
+			if !tt.wantErr(t, err, fmt.Sprintf("Read(%v)", tt.args.p)) {
+				return
+			}
+			assert.Equalf(t, tt.want, got, "Read(%v)", tt.args.p)
+			assert.Equalf(t, tt.wantBytes, tt.args.p[:got], "Read(%v)", tt.args.p)
+		})
+	}
+}
+
+func TestField_DecodeInt(t *testing.T) {
+	type fields struct {
+		Data []byte
+	}
+	tests := []struct {
+		name    string
+		fields  fields
+		want    int
+		wantErr assert.ErrorAssertionFunc
+	}{
+		{
+			name:    "with 2 bytes of input",
+			fields:  fields{Data: []byte{0, 1}},
+			want:    1,
+			wantErr: assert.NoError,
+		},
+		{
+			name:    "with 4 bytes of input",
+			fields:  fields{Data: []byte{0, 1, 0, 0}},
+			want:    65536,
+			wantErr: assert.NoError,
+		},
+		{
+			name:    "with invalid number of bytes of input",
+			fields:  fields{Data: []byte{1, 0, 0, 0, 0, 0, 0, 0}},
+			want:    0,
+			wantErr: assert.Error,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			f := &Field{
+				Data: tt.fields.Data,
+			}
+			got, err := f.DecodeInt()
+			if !tt.wantErr(t, err) {
+				return
+			}
+			assert.Equalf(t, tt.want, got, "DecodeInt()")
 		})
 	}
 }