Limit chat message size to 8192 bytes

[rbdr/mobius] / hotline / file_header_test.go

diff --git a/hotline/file_header_test.go b/hotline/file_header_test.go
index 0194a605c3ee4d09fcf2145921a515639cb482ee..308d6b9525046aa970f68d7840ebfc4bb84376be 100644 (file)
--- a/hotline/file_header_test.go
+++ b/hotline/file_header_test.go
@@ -1,6 +1,7 @@
 package hotline
 
 import (
+       "io"
        "reflect"
        "testing"
 )
@@ -22,8 +23,8 @@ func TestNewFileHeader(t *testing.T) {
                                isDir:    false,
                        },
                        want: FileHeader{
-                               Size:     []byte{0x00, 0x0a},
-                               Type:     []byte{0x00, 0x00},
+                               Size:     [2]byte{0x00, 0x0a},
+                               Type:     [2]byte{0x00, 0x00},
                                FilePath: EncodeFilePath("foo"),
                        },
                },
@@ -34,8 +35,8 @@ func TestNewFileHeader(t *testing.T) {
                                isDir:    true,
                        },
                        want: FileHeader{
-                               Size:     []byte{0x00, 0x0a},
-                               Type:     []byte{0x00, 0x01},
+                               Size:     [2]byte{0x00, 0x0a},
+                               Type:     [2]byte{0x00, 0x01},
                                FilePath: EncodeFilePath("foo"),
                        },
                },
@@ -51,8 +52,8 @@ func TestNewFileHeader(t *testing.T) {
 
 func TestFileHeader_Payload(t *testing.T) {
        type fields struct {
-               Size     []byte
-               Type     []byte
+               Size     [2]byte
+               Type     [2]byte
                FilePath []byte
        }
        tests := []struct {
@@ -63,8 +64,8 @@ func TestFileHeader_Payload(t *testing.T) {
                {
                        name: "has expected payload bytes",
                        fields: fields{
-                               Size:     []byte{0x00, 0x0a},
-                               Type:     []byte{0x00, 0x00},
+                               Size:     [2]byte{0x00, 0x0a},
+                               Type:     [2]byte{0x00, 0x00},
                                FilePath: EncodeFilePath("foo"),
                        },
                        want: []byte{
@@ -84,8 +85,9 @@ func TestFileHeader_Payload(t *testing.T) {
                                Type:     tt.fields.Type,
                                FilePath: tt.fields.FilePath,
                        }
-                       if got := fh.Payload(); !reflect.DeepEqual(got, tt.want) {
-                               t.Errorf("Payload() = %v, want %v", got, tt.want)
+                       got, _ := io.ReadAll(fh)
+                       if !reflect.DeepEqual(got, tt.want) {
+                               t.Errorf("Read() = %v, want %v", got, tt.want)
                        }
                })
        }