feat: updated test cases

crawlab-team · Jun 7, 2024 · 2f2fd73 · 2f2fd73
1 parent 0345917
commit 2f2fd73
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/models/service/base_service_v2_test.go b/models/service/base_service_v2_test.go
@@ -25,7 +25,10 @@ func setupTestDB() {
 
 func teardownTestDB() {
 	db := mongo.GetMongoDb("testdb")
-	db.Drop(context.Background())
+	err := db.Drop(context.Background())
+	if err != nil {
+		return
+	}
 }
 
 func TestModelServiceV2_GetById(t *testing.T) {