From 0460e7ce29e187a15df9836363ceb59f25fb79c8 Mon Sep 17 00:00:00 2001
From: Joy Wang <joy.wang@mongodb.com>
Date: Fri, 13 Sep 2024 17:21:48 -0400
Subject: [PATCH] remove tests that expect ErrClientDisconnected in collection
 and database

---
 mongo/collection_test.go | 53 ----------------------------------------
 mongo/database_test.go   | 11 ---------
 2 files changed, 64 deletions(-)

diff --git a/mongo/collection_test.go b/mongo/collection_test.go
index 648f04a46f..53d856c860 100644
--- a/mongo/collection_test.go
+++ b/mongo/collection_test.go
@@ -78,59 +78,6 @@ func TestCollection(t *testing.T) {
 		}
 		compareColls(t, expected, coll)
 	})
-	t.Run("replace topology error", func(t *testing.T) {
-		coll := setupColl("foo")
-		doc := bson.D{}
-		update := bson.D{{"$update", bson.D{{"x", 1}}}}
-
-		_, err := coll.InsertOne(bgCtx, doc)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.InsertMany(bgCtx, []interface{}{doc})
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.DeleteOne(bgCtx, doc)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.DeleteMany(bgCtx, doc)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.UpdateOne(bgCtx, doc, update)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.UpdateMany(bgCtx, doc, update)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.ReplaceOne(bgCtx, doc, doc)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.Aggregate(bgCtx, Pipeline{})
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.EstimatedDocumentCount(bgCtx)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.CountDocuments(bgCtx, doc)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		err = coll.Distinct(bgCtx, "x", doc).Err()
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = coll.Find(bgCtx, doc)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		err = coll.FindOne(bgCtx, doc).Err()
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		err = coll.FindOneAndDelete(bgCtx, doc).Err()
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		err = coll.FindOneAndReplace(bgCtx, doc, doc).Err()
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		err = coll.FindOneAndUpdate(bgCtx, doc, update).Err()
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-	})
 	t.Run("database accessor", func(t *testing.T) {
 		coll := setupColl("bar")
 		dbName := coll.Database().Name()
diff --git a/mongo/database_test.go b/mongo/database_test.go
index 6b9b8df319..f209cec542 100644
--- a/mongo/database_test.go
+++ b/mongo/database_test.go
@@ -83,17 +83,6 @@ func TestDatabase(t *testing.T) {
 			compareDbs(t, expected, got)
 		})
 	})
-	t.Run("replace topology error", func(t *testing.T) {
-		db := setupDb("foo")
-		err := db.RunCommand(bgCtx, bson.D{{"x", 1}}).Err()
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		err = db.Drop(bgCtx)
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-
-		_, err = db.ListCollections(bgCtx, bson.D{})
-		assert.Equal(t, ErrClientDisconnected, err, "expected error %v, got %v", ErrClientDisconnected, err)
-	})
 	t.Run("TransientTransactionError label", func(t *testing.T) {
 		client := setupClient(options.Client().ApplyURI("mongodb://nonexistent").SetServerSelectionTimeout(3 * time.Second))
 		err := client.connect()