Account for the root

[rbdr/mobius] / hotline / news_test.go

diff --git a/hotline/news_test.go b/hotline/news_test.go
index a2102b23c79959147d0446252857fffcc2de0c5d..d1b043edc333762066a694d62c61846757ac04e0 100644 (file)
--- a/hotline/news_test.go
+++ b/hotline/news_test.go
@@ -1,7 +1,8 @@
 package hotline
 
 import (
-       "reflect"
+       "github.com/stretchr/testify/assert"
+       "io"
        "testing"
 )
 
@@ -60,8 +61,8 @@ func TestNewsCategoryListData15_MarshalBinary(t *testing.T) {
                                0x00, 0x03,
                                0x00, 0x01,
                                0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
-                               0x00, 0x00, 0x00, 0x01,
-                               0x00, 0x00, 0x00, 0x02,
+                               0x00, 0x00, 0x00, 0x00,
+                               0x00, 0x00, 0x00, 0x00,
                                0x03,
                                0x66, 0x6f, 0x6f,
                        },
@@ -79,7 +80,7 @@ func TestNewsCategoryListData15_MarshalBinary(t *testing.T) {
                                DeleteSN: tt.fields.DeleteSN,
                                GUID:     tt.fields.GUID,
                        }
-                       gotData, err := newscat.MarshalBinary()
+                       gotData, err := io.ReadAll(newscat)
                        if newscat.Type == [2]byte{0, 3} {
                                // zero out the random GUID before comparison
                                for i := 4; i < 20; i++ {
@@ -90,7 +91,7 @@ func TestNewsCategoryListData15_MarshalBinary(t *testing.T) {
                                t.Errorf("MarshalBinary() error = %v, wantErr %v", err, tt.wantErr)
                                return
                        }
-                       if !reflect.DeepEqual(gotData, tt.wantData) {
+                       if !assert.Equal(t, tt.wantData, gotData) {
                                t.Errorf("MarshalBinary() gotData = %v, want %v", gotData, tt.wantData)
                        }
                })