mirror of
https://github.com/bolucat/Archive.git
synced 2026-04-23 00:17:16 +08:00
github-action[bot]
74 lines
2.1 KiB
Go
74 lines
2.1 KiB
Go
// Copyright (C) 2023 mieru authors
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package protocol
|
|
|
|
import (
|
|
mrand "math/rand"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestSessionStruct(t *testing.T) {
|
|
s := &sessionStruct{
|
|
baseStruct: baseStruct{
|
|
protocol: uint8(closeSessionRequest),
|
|
},
|
|
sessionID: mrand.Uint32(),
|
|
statusCode: uint8(mrand.Uint32()),
|
|
seq: mrand.Uint32(),
|
|
payloadLen: uint16(mrand.Uint32()),
|
|
suffixLen: uint8(mrand.Uint32()),
|
|
}
|
|
b := s.Marshal()
|
|
s2 := &sessionStruct{}
|
|
if err := s2.Unmarshal(b); err != nil {
|
|
t.Fatalf("Unmarshal() failed: %v", err)
|
|
}
|
|
if !reflect.DeepEqual(s, s2) {
|
|
t.Errorf("Not equal:\n%v\n====\n%v", s, s2)
|
|
}
|
|
if s.String() != s2.String() {
|
|
t.Errorf("Not equal:\n%s\n====\n%s", s.String(), s2.String())
|
|
}
|
|
}
|
|
|
|
func TestDataAckStruct(t *testing.T) {
|
|
s := &dataAckStruct{
|
|
baseStruct: baseStruct{
|
|
protocol: uint8(dataServerToClient),
|
|
},
|
|
sessionID: mrand.Uint32(),
|
|
seq: mrand.Uint32(),
|
|
unAckSeq: mrand.Uint32(),
|
|
windowSize: uint16(mrand.Uint32()),
|
|
fragment: uint8(mrand.Uint32()),
|
|
prefixLen: uint8(mrand.Uint32()),
|
|
payloadLen: uint16(mrand.Uint32()),
|
|
suffixLen: uint8(mrand.Uint32()),
|
|
}
|
|
b := s.Marshal()
|
|
s2 := &dataAckStruct{}
|
|
if err := s2.Unmarshal(b); err != nil {
|
|
t.Fatalf("Unmarshal() failed: %v", err)
|
|
}
|
|
if !reflect.DeepEqual(s, s2) {
|
|
t.Errorf("Not equal:\n%v\n====\n%v", s, s2)
|
|
}
|
|
if s.String() != s2.String() {
|
|
t.Errorf("Not equal:\n%s\n====\n%s", s.String(), s2.String())
|
|
}
|
|
}
|