WIP

filecoin-project · May 5, 2020 · 2536126 · 2536126
1 parent 9a1d95f
commit 2536126
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/impl/graphsync/graphsync_impl_test.go b/impl/graphsync/graphsync_impl_test.go
@@ -935,10 +935,14 @@ func TestDataTransferPushRoundTrip(t *testing.T) {
 	dt2 := NewGraphSyncDataTransfer(host2, gs2, gsData.StoredCounter2)
 
 	finished := make(chan struct{}, 2)
+	errChan := make(chan struct{}, 2)
 	var subscriber datatransfer.Subscriber = func(event datatransfer.Event, channelState datatransfer.ChannelState) {
 		if event.Code == datatransfer.Complete {
 			finished <- struct{}{}
 		}
+		if event.Code == datatransfer.Error {
+			errChan <- struct{}{}
+		}
 	}
 	dt1.SubscribeToEvents(subscriber)
 	dt2.SubscribeToEvents(subscriber)
@@ -954,6 +958,8 @@ func TestDataTransferPushRoundTrip(t *testing.T) {
 		case <-ctx.Done():
 			t.Fatal("Did not complete succcessful data transfer")
 		case <-finished:
+		case <-errChan:
+			t.Fatal("received error on data transfer")
 		}
 	}
 	gsData.VerifyFileTransferred(t, root, true)