fix

Author: tdq45gj

internal/verifier/change_stream_test.go

@@ -3,15 +3,13 @@ package verifier
 import (
 	"context"
 	"strings"
-	"testing"
 	"time"
 
 	"github.com/10gen/migration-verifier/internal/testutil"
 	"github.com/10gen/migration-verifier/internal/util"
 	"github.com/10gen/migration-verifier/mslices"
 	"github.com/pkg/errors"
 	"github.com/samber/lo"
-	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
@@ -20,20 +18,16 @@ import (
 	"go.mongodb.org/mongo-driver/mongo/readconcern"
 )
 
-func TestChangeStreamFilter(t *testing.T) {
-	verifier := Verifier{
-		nsMap: NewNSMap(),
-	}
-	verifier.SetMetaDBName("metadb")
-	verifier.initializeChangeStreamReaders()
-	assert.Contains(t,
+func (suite *IntegrationTestSuite) TestChangeStreamFilter() {
+	verifier := suite.BuildVerifier()
+	suite.Assert().Contains(
 		verifier.srcChangeStreamReader.GetChangeStreamFilter(),
 		bson.D{
-			{"$match", bson.D{{"ns.db", bson.D{{"$ne", "metadb"}}}}},
+			{"$match", bson.D{{"ns.db", bson.D{{"$ne", metaDBName}}}}},
 		},
 	)
 	verifier.srcChangeStreamReader.namespaces = []string{"foo.bar", "foo.baz", "test.car", "test.chaz"}
-	assert.Contains(t,
+	suite.Assert().Contains(
 		verifier.srcChangeStreamReader.GetChangeStreamFilter(),
 		bson.D{{"$match", bson.D{
 			{"$or", []bson.D{

internal/verifier/migration_verifier_test.go

@@ -1387,7 +1387,7 @@ func (suite *IntegrationTestSuite) TestGenerationalRechecking() {
 	// wait for generation to finish
 	suite.Require().NoError(runner.AwaitGenerationEnd())
 	status = waitForTasks()
-	// there should be no failures now, since they are are equivalent at this point in time
+	// there should be no failures now, since they are equivalent at this point in time
 	suite.Require().Equal(VerificationStatus{TotalTasks: 1, CompletedTasks: 1}, *status)
 
 	// now insert in the source, this should come up next generation
@@ -1415,7 +1415,7 @@ func (suite *IntegrationTestSuite) TestGenerationalRechecking() {
 	suite.Require().NoError(runner.AwaitGenerationEnd())
 	status = waitForTasks()
 
-	// there should be no failures now, since they are are equivalent at this point in time
+	// there should be no failures now, since they are equivalent at this point in time
 	suite.Assert().Equal(VerificationStatus{TotalTasks: 1, CompletedTasks: 1}, *status)
 
 	// We could just abandon this verifier, but we might as well shut it down
@@ -1613,6 +1613,7 @@ func (suite *IntegrationTestSuite) TestPartitionWithFilter() {
 	// Set up the verifier for testing.
 	verifier := suite.BuildVerifier()
 	verifier.SetSrcNamespaces([]string{dbname + ".testColl1"})
+	verifier.SetDstNamespaces([]string{dbname + ".testColl1"})
 	verifier.SetNamespaceMap()
 	verifier.globalFilter = filter
 	// Use a small partition size so that we can test creating multiple partitions.