73 lines
1.1 KiB
Go
73 lines
1.1 KiB
Go
package net
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func Test(t *testing.T) {
|
|
p := make(chan int, 1)
|
|
go server(t, p)
|
|
<-p
|
|
client(t)
|
|
}
|
|
|
|
func server(t *testing.T, prepare chan<- int) {
|
|
l, err := ListenRCON("localhost:25575")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer l.Close()
|
|
|
|
prepare <- 1
|
|
|
|
for {
|
|
conn, err := l.Accept()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
go func(conn RCONServerConn) {
|
|
err := conn.AcceptLogin("RightPassword")
|
|
if err != nil {
|
|
t.Fatal("password wrong")
|
|
}
|
|
|
|
for {
|
|
cmd, err := conn.AcceptCmd()
|
|
if err != nil {
|
|
t.Log(err)
|
|
return
|
|
}
|
|
resp := handleCommand(cmd)
|
|
err = conn.RespCmd(resp)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}(conn)
|
|
}
|
|
}
|
|
|
|
func handleCommand(cmd string) (resp string) {
|
|
return fmt.Sprintf("your command is %q", cmd)
|
|
}
|
|
|
|
func client(t *testing.T) {
|
|
conn, err := DialRCON("localhost:25575", "RightPassword")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer conn.Close()
|
|
|
|
err = conn.Cmd("TEST COMMAND")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
resp, err := conn.Resp()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Logf("Server response: %q", resp)
|
|
}
|