Initial squashed commit

1 files changed, 394 insertions, 0 deletions

diff --git a/transaction_handlers_test.go b/transaction_handlers_test.go
new file mode 100644
index 0000000..3042da2
--- /dev/null
+++ b/transaction_handlers_test.go
@@ -0,0 +1,394 @@
+package hotline
+
+import (
+	"github.com/stretchr/testify/assert"
+	"math/rand"
+	"reflect"
+	"testing"
+)
+
+func TestHandleSetChatSubject(t *testing.T) {
+	type args struct {
+		cc *ClientConn
+		t  *Transaction
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    []Transaction
+		wantErr bool
+	}{
+		{
+			name: "sends chat subject to private chat members",
+			args: args{
+				cc: &ClientConn{
+					UserName: &[]byte{0x00, 0x01},
+					Server: &Server{
+						PrivateChats: map[uint32]*PrivateChat{
+							uint32(1): {
+								Subject: "unset",
+								ClientConn: map[uint16]*ClientConn{
+									uint16(1): {
+										Account: &Account{
+											Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+										},
+										ID: &[]byte{0, 1},
+									},
+									uint16(2): {
+										Account: &Account{
+											Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+										},
+										ID: &[]byte{0, 2},
+									},
+								},
+							},
+						},
+						Clients: map[uint16]*ClientConn{
+							uint16(1): {
+								Account: &Account{
+									Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+								},
+								ID: &[]byte{0, 1},
+							},
+							uint16(2): {
+								Account: &Account{
+									Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+								},
+								ID: &[]byte{0, 2},
+							},
+						},
+					},
+				},
+				t: &Transaction{
+					Flags:     0x00,
+					IsReply:   0x00,
+					Type:      []byte{0, 0x6a},
+					ID:        []byte{0, 0, 0, 1},
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(fieldChatID, []byte{0, 0, 0, 1}),
+						NewField(fieldChatSubject, []byte("Test Subject")),
+					},
+				},
+			},
+			want: []Transaction{
+				{
+					clientID:  &[]byte{0, 1},
+					Flags:     0x00,
+					IsReply:   0x00,
+					Type:      []byte{0, 0x77},
+					ID:        []byte{0x9a, 0xcb, 0x04, 0x42}, // Random ID from rand.Seed(1)
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(fieldChatID, []byte{0, 0, 0, 1}),
+						NewField(fieldChatSubject, []byte("Test Subject")),
+					},
+				},
+				{
+					clientID:  &[]byte{0, 2},
+					Flags:     0x00,
+					IsReply:   0x00,
+					Type:      []byte{0, 0x77},
+					ID:        []byte{0xf0, 0xc5, 0x34, 0x1e}, // Random ID from rand.Seed(1)
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(fieldChatID, []byte{0, 0, 0, 1}),
+						NewField(fieldChatSubject, []byte("Test Subject")),
+					},
+				},
+			},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		rand.Seed(1) // reset seed between tests to make transaction IDs predictable
+
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := HandleSetChatSubject(tt.args.cc, tt.args.t)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("HandleSetChatSubject() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !assert.Equal(t, tt.want, got) {
+				t.Errorf("HandleSetChatSubject() got = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestHandleLeaveChat(t *testing.T) {
+	type args struct {
+		cc *ClientConn
+		t  *Transaction
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    []Transaction
+		wantErr bool
+	}{
+		{
+			name: "returns expected transactions",
+			args: args{
+				cc: &ClientConn{
+					ID: &[]byte{0, 2},
+					Server: &Server{
+						PrivateChats: map[uint32]*PrivateChat{
+							uint32(1): {
+								ClientConn: map[uint16]*ClientConn{
+									uint16(1): {
+										Account: &Account{
+											Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+										},
+										ID: &[]byte{0, 1},
+									},
+									uint16(2): {
+										Account: &Account{
+											Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+										},
+										ID: &[]byte{0, 2},
+									},
+								},
+							},
+						},
+						Clients: map[uint16]*ClientConn{
+							uint16(1): {
+								Account: &Account{
+									Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+								},
+								ID: &[]byte{0, 1},
+							},
+							uint16(2): {
+								Account: &Account{
+									Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+								},
+								ID: &[]byte{0, 2},
+							},
+						},
+					},
+				},
+				t: NewTransaction(tranDeleteUser,nil, NewField(fieldChatID, []byte{0, 0, 0, 1})),
+			},
+			want: []Transaction{
+				{
+					clientID:  &[]byte{0, 1},
+					Flags:     0x00,
+					IsReply:   0x00,
+					Type:      []byte{0, 0x76},
+					ID:        []byte{0x9a, 0xcb, 0x04, 0x42}, // Random ID from rand.Seed(1)
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(fieldChatID, []byte{0, 0, 0, 1}),
+						NewField(fieldUserID, []byte{0, 2}),
+					},
+				},
+			},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		rand.Seed(1)
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := HandleLeaveChat(tt.args.cc, tt.args.t)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("HandleLeaveChat() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !assert.Equal(t, tt.want, got) {
+				t.Errorf("HandleLeaveChat() got = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+
+func TestHandleGetUserNameList(t *testing.T) {
+	type args struct {
+		cc *ClientConn
+		t  *Transaction
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    []Transaction
+		wantErr bool
+	}{
+		{
+			name: "replies with userlist transaction",
+			args: args{
+				cc: &ClientConn{
+
+					ID: &[]byte{1, 1},
+					Server: &Server{
+						Clients: map[uint16]*ClientConn{
+							uint16(1): {
+								ID:       &[]byte{0, 1},
+								Icon:     &[]byte{0, 2},
+								Flags:    &[]byte{0, 3},
+								UserName: &[]byte{0, 4},
+							},
+						},
+					},
+				},
+				t: &Transaction{
+					ID:   []byte{0, 0, 0, 1},
+					Type: []byte{0, 1},
+				},
+			},
+			want: []Transaction{
+				{
+					clientID:  &[]byte{1, 1},
+					Flags:     0x00,
+					IsReply:   0x01,
+					Type:      []byte{0, 1},
+					ID:        []byte{0, 0, 0, 1},
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(
+							fieldUsernameWithInfo,
+							[]byte{00, 01, 00, 02, 00, 03, 00, 02, 00, 04},
+						),
+					},
+				},
+			},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := HandleGetUserNameList(tt.args.cc, tt.args.t)
+			if (err != nil) != tt.wantErr {
+				t.Errorf("HandleGetUserNameList() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !reflect.DeepEqual(got, tt.want) {
+				t.Errorf("HandleGetUserNameList() got = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}
+
+func TestHandleChatSend(t *testing.T) {
+	type args struct {
+		cc *ClientConn
+		t  *Transaction
+	}
+	tests := []struct {
+		name    string
+		args    args
+		want    []Transaction
+		wantErr bool
+	}{
+		{
+			name: "sends chat msg transaction to all clients",
+			args: args{
+				cc: &ClientConn{
+					UserName: &[]byte{0x00, 0x01},
+					Server: &Server{
+						Clients: map[uint16]*ClientConn{
+							uint16(1): {
+								Account: &Account{
+									Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+								},
+								ID: &[]byte{0, 1},
+							},
+							uint16(2): {
+								Account: &Account{
+									Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+								},
+								ID: &[]byte{0, 2},
+							},
+						},
+					},
+				},
+				t: &Transaction{
+					Fields: []Field{
+						NewField(fieldData, []byte("hai")),
+					},
+				},
+			},
+			want: []Transaction{
+				{
+					clientID:  &[]byte{0, 1},
+					Flags:     0x00,
+					IsReply:   0x00,
+					Type:      []byte{0, 0x6a},
+					ID:        []byte{0x9a, 0xcb, 0x04, 0x42}, // Random ID from rand.Seed(1)
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(fieldData, []byte{0x0d, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x00, 0x01, 0x3a, 0x20, 0x20, 0x68, 0x61, 0x69}),
+					},
+				},
+				{
+					clientID:  &[]byte{0, 2},
+					Flags:     0x00,
+					IsReply:   0x00,
+					Type:      []byte{0, 0x6a},
+					ID:        []byte{0xf0, 0xc5, 0x34, 0x1e}, // Random ID from rand.Seed(1)
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(fieldData, []byte{0x0d, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x00, 0x01, 0x3a, 0x20, 0x20, 0x68, 0x61, 0x69}),
+					},
+				},
+			},
+			wantErr: false,
+		},
+		{
+			name: "only sends chat msg to clients with accessReadChat permission",
+			args: args{
+				cc: &ClientConn{
+					UserName: &[]byte{0x00, 0x01},
+					Server: &Server{
+						Clients: map[uint16]*ClientConn{
+							uint16(1): {
+								Account: &Account{
+									Access: &[]byte{255, 255, 255, 255, 255, 255, 255, 255},
+								},
+								ID: &[]byte{0, 1},
+							},
+							uint16(2): {
+								Account: &Account{
+									Access: &[]byte{0, 0, 0, 0, 0, 0, 0, 0},
+								},
+								ID: &[]byte{0, 2},
+							},
+						},
+					},
+				},
+				t: &Transaction{
+					Fields: []Field{
+						NewField(fieldData, []byte("hai")),
+					},
+				},
+			},
+			want: []Transaction{
+				{
+					clientID:  &[]byte{0, 1},
+					Flags:     0x00,
+					IsReply:   0x00,
+					Type:      []byte{0, 0x6a},
+					ID:        []byte{0x9a, 0xcb, 0x04, 0x42}, // Random ID from rand.Seed(1)
+					ErrorCode: []byte{0, 0, 0, 0},
+					Fields: []Field{
+						NewField(fieldData, []byte{0x0d, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x20, 0x00, 0x01, 0x3a, 0x20, 0x20, 0x68, 0x61, 0x69}),
+					},
+				},
+			},
+			wantErr: false,
+		},
+	}
+	for _, tt := range tests {
+		rand.Seed(1) // reset seed between tests to make transaction IDs predictable
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := HandleChatSend(tt.args.cc, tt.args.t)
+
+			if (err != nil) != tt.wantErr {
+				t.Errorf("HandleChatSend() error = %v, wantErr %v", err, tt.wantErr)
+				return
+			}
+			if !assert.Equal(t, tt.want, got) {
+				t.Errorf("HandleChatSend() got = %v, want %v", got, tt.want)
+			}
+		})
+	}
+}