test(firestore): Cleanup documents created during test run (#10188)

googleapis · May 24, 2024 · 727da30 · 727da30
1 parent 2d4fc81
commit 727da30
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/firestore/integration_test.go b/firestore/integration_test.go
@@ -2478,10 +2478,15 @@ func TestIntegration_AggregationQueries(t *testing.T) {
 		{"weight": 7.1, "height": 93, "model": "C"},
 		{"weight": 8.2, "height": 93, "model": "A"},
 	}
+	docRefs := []*DocumentRef{}
 	for _, doc := range docs {
 		newDoc := coll.NewDoc()
+		docRefs = append(docRefs, newDoc)
 		h.mustCreate(newDoc, doc)
 	}
+	t.Cleanup(func() {
+		deleteDocuments(docRefs)
+	})
 
 	query := coll.Where("weight", ">=", 1)