testutil: DummyPeer now holds onto the *is.Is

makes SendAndRecv a bit cleaner imo
This commit is contained in:
CPunch 2024-02-04 11:26:48 -06:00
parent cd93a058ce
commit 79f68187bf
2 changed files with 11 additions and 10 deletions

View File

@ -19,6 +19,7 @@ import (
type DummyPeer struct { type DummyPeer struct {
Recv chan *cnet.PacketEvent Recv chan *cnet.PacketEvent
Peer *cnet.Peer Peer *cnet.Peer
is *is.I
} }
// MakeDummyPeer creates a new dummy peer and returns it // MakeDummyPeer creates a new dummy peer and returns it
@ -32,21 +33,21 @@ func MakeDummyPeer(ctx context.Context, is *is.I, port int) *DummyPeer {
peer.Handler(recv) peer.Handler(recv)
}() }()
return &DummyPeer{Recv: recv, Peer: peer} return &DummyPeer{Recv: recv, Peer: peer, is: is}
} }
// SendAndRecv sends a packet (sID & out), waits for the expected response (rID) and decodes it into in // SendAndRecv sends a packet (sID & out), waits for the expected response (rID) and decodes it into in
func (dp *DummyPeer) SendAndRecv(is *is.I, sID, rID uint32, out, in interface{}) { func (dp *DummyPeer) SendAndRecv(sID, rID uint32, out, in interface{}) {
// send out packet // send out packet
err := dp.Peer.Send(sID, out) err := dp.Peer.Send(sID, out)
is.NoErr(err) // peer.Send() should not return an error dp.is.NoErr(err) // peer.Send() should not return an error
// receive response // receive response
evnt := <-dp.Recv evnt := <-dp.Recv
defer protocol.PutBuffer(evnt.Pkt) defer protocol.PutBuffer(evnt.Pkt)
is.Equal(evnt.PktID, rID) // should receive expected type dp.is.Equal(evnt.PktID, rID) // should receive expected type
is.NoErr(protocol.NewPacket(evnt.Pkt).Decode(in)) // packet.Decode() should not return an error dp.is.NoErr(protocol.NewPacket(evnt.Pkt).Decode(in)) // packet.Decode() should not return an error
} }
// Kill closes the peer's connection // Kill closes the peer's connection

View File

@ -87,7 +87,7 @@ func TestLoginSuccSequence(t *testing.T) {
// send login request (this should create an account) // send login request (this should create an account)
var resp protocol.SP_LS2CL_REP_LOGIN_SUCC var resp protocol.SP_LS2CL_REP_LOGIN_SUCC
dummy.SendAndRecv(is, protocol.P_CL2LS_REQ_LOGIN, protocol.P_LS2CL_REP_LOGIN_SUCC, dummy.SendAndRecv(protocol.P_CL2LS_REQ_LOGIN, protocol.P_LS2CL_REP_LOGIN_SUCC,
protocol.SP_CL2LS_REQ_LOGIN{ protocol.SP_CL2LS_REQ_LOGIN{
SzID: "testLoginSequence", SzID: "testLoginSequence",
SzPassword: "test", SzPassword: "test",
@ -113,7 +113,7 @@ func TestLoginFailSequence(t *testing.T) {
// send login request (this should not create an account) // send login request (this should not create an account)
var resp protocol.SP_LS2CL_REP_LOGIN_FAIL var resp protocol.SP_LS2CL_REP_LOGIN_FAIL
dummy.SendAndRecv(is, protocol.P_CL2LS_REQ_LOGIN, protocol.P_LS2CL_REP_LOGIN_FAIL, dummy.SendAndRecv(protocol.P_CL2LS_REQ_LOGIN, protocol.P_LS2CL_REP_LOGIN_FAIL,
protocol.SP_CL2LS_REQ_LOGIN{ protocol.SP_CL2LS_REQ_LOGIN{
SzID: "", SzID: "",
SzPassword: "", SzPassword: "",
@ -135,7 +135,7 @@ func TestCharacterSequence(t *testing.T) {
// send login request (this should create an account) // send login request (this should create an account)
var resp protocol.SP_LS2CL_REP_LOGIN_SUCC var resp protocol.SP_LS2CL_REP_LOGIN_SUCC
dummy.SendAndRecv(is, protocol.P_CL2LS_REQ_LOGIN, protocol.P_LS2CL_REP_LOGIN_SUCC, dummy.SendAndRecv(protocol.P_CL2LS_REQ_LOGIN, protocol.P_LS2CL_REP_LOGIN_SUCC,
protocol.SP_CL2LS_REQ_LOGIN{ protocol.SP_CL2LS_REQ_LOGIN{
SzID: "testCharacterSequence", SzID: "testCharacterSequence",
SzPassword: "test", SzPassword: "test",
@ -159,7 +159,7 @@ func TestCharacterSequence(t *testing.T) {
// send character name check request // send character name check request
var charResp protocol.SP_LS2CL_REP_SAVE_CHAR_NAME_SUCC var charResp protocol.SP_LS2CL_REP_SAVE_CHAR_NAME_SUCC
dummy.SendAndRecv(is, protocol.P_CL2LS_REQ_SAVE_CHAR_NAME, protocol.P_LS2CL_REP_SAVE_CHAR_NAME_SUCC, dummy.SendAndRecv(protocol.P_CL2LS_REQ_SAVE_CHAR_NAME, protocol.P_LS2CL_REP_SAVE_CHAR_NAME_SUCC,
protocol.SP_CL2LS_REQ_SAVE_CHAR_NAME{ protocol.SP_CL2LS_REQ_SAVE_CHAR_NAME{
ISlotNum: 1, ISlotNum: 1,
IGender: 1, IGender: 1,
@ -180,7 +180,7 @@ func TestCharacterSequence(t *testing.T) {
charCreate := testCharCreate charCreate := testCharCreate
charCreate.PCStyle.IPC_UID = charResp.IPC_UID charCreate.PCStyle.IPC_UID = charResp.IPC_UID
var charCreateResp protocol.SP_LS2CL_REP_CHAR_CREATE_SUCC var charCreateResp protocol.SP_LS2CL_REP_CHAR_CREATE_SUCC
dummy.SendAndRecv(is, protocol.P_CL2LS_REQ_CHAR_CREATE, protocol.P_LS2CL_REP_CHAR_CREATE_SUCC, dummy.SendAndRecv(protocol.P_CL2LS_REQ_CHAR_CREATE, protocol.P_LS2CL_REP_CHAR_CREATE_SUCC,
charCreate, &charCreateResp) charCreate, &charCreateResp)
// verify response // verify response