X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/5c34f8752938764668d829fb284626b9c62c5475..af8daec6c282ca85669bfe03a7afb46c24607caa:/hotline/transaction_test.go?ds=sidebyside

diff --git a/hotline/transaction_test.go b/hotline/transaction_test.go
index 4c7d372..04bcde0 100644
--- a/hotline/transaction_test.go
+++ b/hotline/transaction_test.go
@@ -1,6 +1,7 @@
 package hotline
 
 import (
+	"fmt"
 	"github.com/stretchr/testify/assert"
 	"testing"
 )
@@ -142,16 +143,25 @@ func TestReadTransaction(t *testing.T) {
 		{
 			name: "when buf contains all bytes for a single transaction",
 			args: args{
-				buf: sampleTransaction.Payload(),
+				buf: func() []byte {
+					b, _ := sampleTransaction.MarshalBinary()
+					return b
+				}(),
 			},
-			want:    sampleTransaction,
-			want1:   len(sampleTransaction.Payload()),
+			want: sampleTransaction,
+			want1: func() int {
+				b, _ := sampleTransaction.MarshalBinary()
+				return len(b)
+			}(),
 			wantErr: false,
 		},
 		{
 			name: "when len(buf) is less than the length of the transaction",
 			args: args{
-				buf: sampleTransaction.Payload()[:len(sampleTransaction.Payload())-1],
+				buf: func() []byte {
+					b, _ := sampleTransaction.MarshalBinary()
+					return b[:len(b)-1]
+				}(),
 			},
 			want:    nil,
 			want1:   0,
@@ -174,3 +184,191 @@ func TestReadTransaction(t *testing.T) {
 		})
 	}
 }
+
+func Test_transactionScanner(t *testing.T) {
+	type args struct {
+		data []byte
+		in1  bool
+	}
+	tests := []struct {
+		name        string
+		args        args
+		wantAdvance int
+		wantToken   []byte
+		wantErr     assert.ErrorAssertionFunc
+	}{
+		{
+			name: "when too few bytes are provided to read the transaction size",
+			args: args{
+				data: []byte{},
+				in1:  false,
+			},
+			wantAdvance: 0,
+			wantToken:   []byte(nil),
+			wantErr:     assert.NoError,
+		},
+		{
+			name: "when too few bytes are provided to read the full payload",
+			args: args{
+				data: []byte{
+					0,
+					1,
+					0, 0,
+					0, 00, 00, 04,
+					00, 00, 00, 00,
+					00, 00, 00, 10,
+					00, 00, 00, 10,
+				},
+				in1: false,
+			},
+			wantAdvance: 0,
+			wantToken:   []byte(nil),
+			wantErr:     assert.NoError,
+		},
+		{
+			name: "when a full transaction is provided",
+			args: args{
+				data: []byte{
+					0,
+					1,
+					0, 0,
+					0, 00, 00, 0x04,
+					00, 00, 00, 0x00,
+					00, 00, 00, 0x10,
+					00, 00, 00, 0x10,
+					00, 02,
+					00, 0x6c, // 108 - fieldTransferSize
+					00, 02,
+					0x63, 0x3b,
+					00, 0x6b, // 107 = fieldRefNum
+					00, 0x04,
+					00, 0x02, 0x93, 0x47,
+				},
+				in1: false,
+			},
+			wantAdvance: 36,
+			wantToken: []byte{
+				0,
+				1,
+				0, 0,
+				0, 00, 00, 0x04,
+				00, 00, 00, 0x00,
+				00, 00, 00, 0x10,
+				00, 00, 00, 0x10,
+				00, 02,
+				00, 0x6c, // 108 - fieldTransferSize
+				00, 02,
+				0x63, 0x3b,
+				00, 0x6b, // 107 = fieldRefNum
+				00, 0x04,
+				00, 0x02, 0x93, 0x47,
+			},
+			wantErr: assert.NoError,
+		},
+		{
+			name: "when a full transaction plus extra bytes are provided",
+			args: args{
+				data: []byte{
+					0,
+					1,
+					0, 0,
+					0, 00, 00, 0x04,
+					00, 00, 00, 0x00,
+					00, 00, 00, 0x10,
+					00, 00, 00, 0x10,
+					00, 02,
+					00, 0x6c, // 108 - fieldTransferSize
+					00, 02,
+					0x63, 0x3b,
+					00, 0x6b, // 107 = fieldRefNum
+					00, 0x04,
+					00, 0x02, 0x93, 0x47,
+					1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
+				},
+				in1: false,
+			},
+			wantAdvance: 36,
+			wantToken: []byte{
+				0,
+				1,
+				0, 0,
+				0, 00, 00, 0x04,
+				00, 00, 00, 0x00,
+				00, 00, 00, 0x10,
+				00, 00, 00, 0x10,
+				00, 02,
+				00, 0x6c, // 108 - fieldTransferSize
+				00, 02,
+				0x63, 0x3b,
+				00, 0x6b, // 107 = fieldRefNum
+				00, 0x04,
+				00, 0x02, 0x93, 0x47,
+			},
+			wantErr: assert.NoError,
+		},
+		{
+			name: "when two full transactions are provided",
+			args: args{
+				data: []byte{
+					0,
+					1,
+					0, 0,
+					0, 00, 00, 0x04,
+					00, 00, 00, 0x00,
+					00, 00, 00, 0x10,
+					00, 00, 00, 0x10,
+					00, 02,
+					00, 0x6c, // 108 - fieldTransferSize
+					00, 02,
+					0x63, 0x3b,
+					00, 0x6b, // 107 = fieldRefNum
+					00, 0x04,
+					00, 0x02, 0x93, 0x47,
+					0,
+					1,
+					0, 0,
+					0, 00, 00, 0x04,
+					00, 00, 00, 0x00,
+					00, 00, 00, 0x10,
+					00, 00, 00, 0x10,
+					00, 02,
+					00, 0x6c, // 108 - fieldTransferSize
+					00, 02,
+					0x63, 0x3b,
+					00, 0x6b, // 107 = fieldRefNum
+					00, 0x04,
+					00, 0x02, 0x93, 0x47,
+				},
+				in1: false,
+			},
+			wantAdvance: 36,
+			wantToken: []byte{
+				0,
+				1,
+				0, 0,
+				0, 00, 00, 0x04,
+				00, 00, 00, 0x00,
+				00, 00, 00, 0x10,
+				00, 00, 00, 0x10,
+				00, 02,
+				00, 0x6c, // 108 - fieldTransferSize
+				00, 02,
+				0x63, 0x3b,
+				00, 0x6b, // 107 = fieldRefNum
+				00, 0x04,
+				00, 0x02, 0x93, 0x47,
+			},
+			wantErr: assert.NoError,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			gotAdvance, gotToken, err := transactionScanner(tt.args.data, tt.args.in1)
+			if !tt.wantErr(t, err, fmt.Sprintf("transactionScanner(%v, %v)", tt.args.data, tt.args.in1)) {
+				return
+			}
+			assert.Equalf(t, tt.wantAdvance, gotAdvance, "transactionScanner(%v, %v)", tt.args.data, tt.args.in1)
+			assert.Equalf(t, tt.wantToken, gotToken, "transactionScanner(%v, %v)", tt.args.data, tt.args.in1)
+		})
+	}
+}