docstore/mongodocstore: Make TestLowerCaseFields re-runnable (google#…

…3375)
Simprints · Jan 25, 2024 · 8d700d1 · 8d700d1
1 parent 3130713
commit 8d700d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/docstore/drivertest/driverbenchmark.go b/docstore/drivertest/driverbenchmark.go
@@ -27,7 +27,7 @@ import (
 // RunBenchmarks runs benchmarks for docstore drivers.
 func RunBenchmarks(b *testing.B, coll *docstore.Collection) {
 	defer coll.Close()
-	clearCollection(b, coll)
+	ClearCollection(b, coll)
 	b.Run("BenchmarkSingleActionPut", func(b *testing.B) {
 		benchmarkSingleActionPut(10, b, coll)
 	})
@@ -40,7 +40,7 @@ func RunBenchmarks(b *testing.B, coll *docstore.Collection) {
 	b.Run("BenchmarkActionListGet", func(b *testing.B) {
 		benchmarkActionListGet(100, b, coll)
 	})
-	clearCollection(b, coll)
+	ClearCollection(b, coll)
 }
 
 func benchmarkSingleActionPut(n int, b *testing.B, coll *docstore.Collection) {

diff --git a/docstore/mongodocstore/mongo_test.go b/docstore/mongodocstore/mongo_test.go
@@ -238,6 +238,7 @@ func TestLowercaseFields(t *testing.T) {
 	}
 	coll := docstore.NewCollection(dc)
 	defer coll.Close()
+	defer drivertest.ClearCollection(t, coll)
 
 	type S struct {
 		ID, F, G         int