fix test and lint

filecoin-project · Sep 2, 2020 · 8e44630 · 8e44630
1 parent 34e8aef
commit 8e44630
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/retrievalmarket/impl/integration_test.go b/retrievalmarket/impl/integration_test.go
@@ -74,7 +74,8 @@ func TestProvider_Stop(t *testing.T) {
 	client, expectedCIDs, _, _, retrievalPeer, provider := requireSetupTestClientAndProvider(bgCtx, t, payChAddr)
 	require.NoError(t, provider.Stop())
 	_, err := client.Query(bgCtx, retrievalPeer, expectedCIDs[0], retrievalmarket.QueryParams{})
-	assert.EqualError(t, err, "protocol not supported")
+
+	assert.EqualError(t, err, "exhausted 5 attempts but failed to open stream, err: protocol not supported")
 }
 
 func requireSetupTestClientAndProvider(bgCtx context.Context, t *testing.T, payChAddr address.Address) (retrievalmarket.RetrievalClient,

diff --git a/storagemarket/network/libp2p_impl_test.go b/storagemarket/network/libp2p_impl_test.go
@@ -63,17 +63,21 @@ func TestOpenStreamWithRetries(t *testing.T) {
 		require.NoError(t, err)
 		achan <- readq
 	}}
+
+	var err error
+
 	go func() {
 		select {
-		case _ = <-time.After(3 * time.Second):
-			toNetwork.SetDelegate(tr2)
+		case <-time.After(3 * time.Second):
+			err = toNetwork.SetDelegate(tr2)
 		case <-ctx.Done():
 			return
 		}
 	}()
 
 	// setup query stream host1 --> host 2
 	assertAskRequestReceived(ctx, t, fromNetwork, toHost, achan)
+	assert.NoError(t, err)
 }
 
 func TestAskStreamSendReceiveAskRequest(t *testing.T) {