Fix tests for membership Info (filecoin-project#176)

protofire · May 8, 2023 · 7573c02 · 7573c02
1 parent 1f7bd31
commit 7573c02
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/chain/consensus/mir/manager.go b/chain/consensus/mir/manager.go
@@ -446,7 +446,7 @@ func waitForMembershipInfo(
 		case <-ctx.Done():
 			return nil, nil, ErrWaitForMembershipTimeout
 		case <-next.C:
-			log.With("validator", id).Info("Attempt to retrieve membership information")
+			logger.With("validator", id).Info("Attempt to retrieve membership information")
 			info, m, err := getMembershipInfo(id, r)
 			if errors.Is(err, ErrMissingOwnIdentityInMembership) {
 				continue

diff --git a/chain/consensus/mir/manager_test.go b/chain/consensus/mir/manager_test.go
@@ -20,13 +20,13 @@ func TestWaitForMembership(t *testing.T) {
 
 	logger := golog.Logger("test-logger")
 
-	info, nodes, err := waitForMembershipInfo(ctx, "testID", mb, logger, 3*time.Second)
+	info, nodes, err := waitForMembershipInfo(ctx, "not-existing-ID", mb, logger, 6*time.Second)
+	require.ErrorIs(t, err, ErrWaitForMembershipTimeout)
 	require.Nil(t, info)
 	require.Nil(t, nodes)
-	require.ErrorIs(t, err, ErrWaitForMembershipTimeout)
 
-	info, nodes, err = waitForMembershipInfo(ctx, "t1wpixt5mihkj75lfhrnaa6v56n27epvlgwparujy", mb, logger, 3*time.Second)
+	info, nodes, err = waitForMembershipInfo(ctx, "t1wpixt5mihkj75lfhrnaa6v56n27epvlgwparujy", mb, logger, 6*time.Second)
+	require.NoError(t, err)
 	require.NotNil(t, info)
 	require.NotNil(t, nodes)
-	require.NoError(t, err)
 }