improve tests

Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
2020-11-05 22:38:29 +03:00
parent c9255752a0
commit f484976f55
3 changed files with 37 additions and 8 deletions

View File

@@ -22,8 +22,17 @@ type testHandler struct {
func TestTCPServer(t *testing.T) {
ctx := context.Background()
logger.DefaultLogger = logger.NewLogger(logger.WithLevel(logger.TraceLevel))
reg := rmemory.NewRegistry()
if err := reg.Init(); err != nil {
t.Fatal(err)
}
brk := bmemory.NewBroker(broker.Registry(reg))
if err := brk.Init(); err != nil {
t.Fatal(err)
}
// create server
srv := tcp.NewServer(server.Registry(reg), server.Broker(brk), server.Address("127.0.0.1:65000"))
@@ -35,6 +44,10 @@ func TestTCPServer(t *testing.T) {
t.Fatal(err)
}
if err := srv.Init(); err != nil {
t.Fatal(err)
}
// start server
if err := srv.Start(); err != nil {
t.Fatal(err)
@@ -86,7 +99,7 @@ func (h *testHandler) Serve(c net.Conn) {
if err != nil && err == io.EOF {
return
} else if err != nil {
logger.Fatal(err)
logger.Fatal(err.Error())
}
fmt.Printf("%s", buf[:n])
close(h.done)