mirror of
https://github.com/HDT3213/godis.git
synced 2025-09-26 21:01:17 +08:00
46 lines
744 B
Go
46 lines
744 B
Go
package gnet
|
|
|
|
import (
|
|
"bufio"
|
|
"net"
|
|
"testing"
|
|
|
|
"github.com/hdt3213/godis/database"
|
|
)
|
|
|
|
func TestListenAndServe(t *testing.T) {
|
|
var err error
|
|
listener, err := net.Listen("tcp", ":0")
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
addr := listener.Addr().String()
|
|
db := database.NewStandaloneServer()
|
|
server := NewGnetServer(db)
|
|
go server.Run(addr)
|
|
|
|
conn, err := net.Dial("tcp", addr)
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
_, err = conn.Write([]byte("PING\r\n"))
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
bufReader := bufio.NewReader(conn)
|
|
line, _, err := bufReader.ReadLine()
|
|
if err != nil {
|
|
t.Error(err)
|
|
return
|
|
}
|
|
if string(line) != "+PONG" {
|
|
t.Error("get wrong response")
|
|
return
|
|
}
|
|
conn.Close()
|
|
server.Close()
|
|
}
|