Finish test for failed accept

skycoin · Jun 16, 2019 · 59f2159 · 59f2159
1 parent 1852048
commit 59f2159
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/pkg/dmsg/server_test.go b/pkg/dmsg/server_test.go
@@ -603,12 +603,32 @@ func TestServer_Serve(t *testing.T) {
 			}
 		}()
 
-		ctx, cancel := context.WithTimeout(context.Background(), 1*time.Second)
+		ctx, cancel := context.WithTimeout(context.Background(), 100*time.Second)
 		defer cancel()
 
 		_, err = a.Dial(ctx, bPK)
+		require.Error(t, err)
+
+		time.Sleep(2 * time.Second)
+
+		close(aTpDone)
+		close(bTpDone)
+
 		tpReadWriteWG.Wait()
-		//require.Error(t, err)
+		require.NoError(t, aErr)
+		require.NoError(t, bErr)
+
+		err = aTransport.Close()
+		require.NoError(t, err)
+
+		err = bTransport.Close()
+		require.NoError(t, err)
+
+		err = a.Close()
+		require.NoError(t, err)
+
+		err = b.Close()
+		require.NoError(t, err)
 	})
 }