Add tps and clients closing to `capped_transport_buffer_should_not_re…

…sult_in_hang`
skycoin · Jun 20, 2019 · 8daa622 · 8daa622
1 parent e2345a4
commit 8daa622
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/pkg/dmsg/server_test.go b/pkg/dmsg/server_test.go
@@ -742,7 +742,7 @@ func TestServer_Serve(t *testing.T) {
 		// block on `Write`
 		go func() {
 			_, err = aWrTransport.Write(msg)
-			require.NoError(t, err)
+			require.Error(t, err)
 		}()
 
 		// wait till it's definitely blocked
@@ -766,7 +766,23 @@ func TestServer_Serve(t *testing.T) {
 
 			require.Equal(t, recBuff, msg)
 		}
+
+		err = aWrTransport.Close()
+		require.NoError(t, err)
+
+		err = bWrTransport.Close()
+		require.NoError(t, err)
+
+		err = aRdTransport.Close()
+		require.NoError(t, err)
+
+		err = a.Close()
+		require.NoError(t, err)
+
+		err = b.Close()
+		require.NoError(t, err)
 	})
+
 }
 
 // Given two client instances (a & b) and a server instance (s),