mirror of
https://github.com/CPunch/gopenfusion.git
synced 2024-11-14 12:00:05 +00:00
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package testutil
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/CPunch/gopenfusion/cnet"
|
|
"github.com/CPunch/gopenfusion/cnet/protocol"
|
|
"github.com/matryer/is"
|
|
)
|
|
|
|
type DummyPeer struct {
|
|
Recv chan *cnet.PacketEvent
|
|
Peer *cnet.Peer
|
|
is *is.I
|
|
}
|
|
|
|
// MakeDummyPeer creates a new dummy peer and returns it
|
|
func MakeDummyPeer(ctx context.Context, is *is.I, port int) *DummyPeer {
|
|
conn, err := net.Dial("tcp", fmt.Sprintf("127.0.0.1:%d", port))
|
|
is.NoErr(err)
|
|
|
|
recv := make(chan *cnet.PacketEvent)
|
|
peer := cnet.NewPeer(ctx, conn)
|
|
go func() {
|
|
peer.Handler(recv)
|
|
}()
|
|
|
|
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
|
|
func (dp *DummyPeer) SendAndRecv(sID, rID uint32, out, in interface{}) {
|
|
// send out packet
|
|
err := dp.Peer.Send(sID, out)
|
|
dp.is.NoErr(err) // peer.Send() should not return an error
|
|
|
|
// receive response
|
|
evnt := <-dp.Recv
|
|
defer protocol.PutBuffer(evnt.Pkt)
|
|
|
|
dp.is.Equal(evnt.PktID, rID) // should receive expected type
|
|
dp.is.NoErr(protocol.NewPacket(evnt.Pkt).Decode(in)) // packet.Decode() should not return an error
|
|
}
|
|
|
|
// Kill closes the peer's connection
|
|
func (dp *DummyPeer) Kill() {
|
|
dp.Peer.Kill()
|
|
}
|