CPunch/gopenfusion
1
0
Fork 0
mirror of https://github.com/CPunch/gopenfusion.git synced 2024-11-21 15:00:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor protocol_test.go refactor

Browse Source
This commit is contained in:
CPunch 2023-12-02 17:03:56 -06:00
parent 44f3b31965
commit 76e9bdf7e7
1 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
cnet/protocol/protocol_test.go
View File

@ -61,7 +61,7 @@ func TestPacketEncode(t *testing.T) {
err := pkt.Encode(testStruct) err := pkt.Encode(testStruct)
is.NoErr(err) is.NoErr(err)
is.True(bytes.Equal(buf.Bytes(), testData[:])) // encoded data should match expected data is.Equal(buf.Bytes(), testData[:]) // encoded data should match expected data
} }
func TestPacketDecode(t *testing.T) { func TestPacketDecode(t *testing.T) {
@ -73,7 +73,7 @@ func TestPacketDecode(t *testing.T) {
var test TestPacketData var test TestPacketData
err := pkt.Decode(&test) err := pkt.Decode(&test)
is.NoErr(err) is.NoErr(err)
is.True(test == testStruct) // decoded data should match testStruct is.Equal(test, testStruct) // decoded data should match testStruct
} }
func TestDataEncrypt(t *testing.T) { func TestDataEncrypt(t *testing.T) {
@ -82,8 +82,7 @@ func TestDataEncrypt(t *testing.T) {
copy(buf, testData[:]) copy(buf, testData[:])
protocol.EncryptData(buf, []byte(protocol.DEFAULT_KEY)) protocol.EncryptData(buf, []byte(protocol.DEFAULT_KEY))
is.Equal(buf, encTestData) // encrypted data should match expected data
is.True(bytes.Equal(buf, encTestData)) // encrypted data should match expected data
} }
func TestDataDecrypt(t *testing.T) { func TestDataDecrypt(t *testing.T) {
@ -92,13 +91,12 @@ func TestDataDecrypt(t *testing.T) {
copy(buf, encTestData) copy(buf, encTestData)
protocol.DecryptData(buf, []byte(protocol.DEFAULT_KEY)) protocol.DecryptData(buf, []byte(protocol.DEFAULT_KEY))
is.Equal(buf, testData[:]) // decrypted data should match expected data
is.True(bytes.Equal(buf, testData[:])) // decrypted data should match expected data
} }
func TestCreateNewKey(t *testing.T) { func TestCreateNewKey(t *testing.T) {
is := is.New(t) is := is.New(t)
key := protocol.CreateNewKey(123456789, 0x1234567890abcdef, 0x1234567890abcdef) key := protocol.CreateNewKey(123456789, 0x1234567890abcdef, 0x1234567890abcdef)
is.True(bytes.Equal(key, []byte{0x0, 0x31, 0xb8, 0xcd, 0xd, 0xc3, 0xad, 0x67})) // key should match expected data is.Equal(key, []byte{0x0, 0x31, 0xb8, 0xcd, 0xd, 0xc3, 0xad, 0x67}) // key should match expected data
} }