Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
2021-08-01 12:51:38 +03:00
parent 03fe6da356
commit 9496807a27
7 changed files with 93 additions and 44 deletions

View File

@@ -65,14 +65,6 @@ func TestTCPServer(t *testing.T) {
t.Fatalf("Expected 1 node got %d: %+v", len(service[0].Nodes), service[0].Nodes)
}
go func() {
<-h.done
// stop server
if err := srv.Stop(); err != nil {
t.Fatal(err)
}
}()
c, err := net.DialTimeout("tcp", srv.Options().Address, 5*time.Second)
if err != nil {
t.Fatal(err)
@@ -82,6 +74,12 @@ func TestTCPServer(t *testing.T) {
if _, err = c.Write([]byte("test")); err != nil {
t.Fatal(err)
}
<-h.done
// stop server
if err := srv.Stop(); err != nil {
t.Fatal(err)
}
}
func (h *testHandler) Serve(c net.Conn) {