From c49dfd57de1e8e6d5555ec7ae514e31c0663fb92 Mon Sep 17 00:00:00 2001
From: Lauri Tulmin <ltulmin@splunk.com>
Date: Tue, 13 Apr 2021 11:49:27 +0300
Subject: [PATCH] Fix mongo latest test (#2790)

---
 .../src/main/groovy/AbstractMongoClientTest.groovy            | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/instrumentation/mongo/mongo-testing/src/main/groovy/AbstractMongoClientTest.groovy b/instrumentation/mongo/mongo-testing/src/main/groovy/AbstractMongoClientTest.groovy
index bfcf42522c..c042957578 100644
--- a/instrumentation/mongo/mongo-testing/src/main/groovy/AbstractMongoClientTest.groovy
+++ b/instrumentation/mongo/mongo-testing/src/main/groovy/AbstractMongoClientTest.groovy
@@ -133,6 +133,7 @@ abstract class AbstractMongoClientTest<T> extends AgentInstrumentationSpecificat
         basicSpan(it, 0, "parent")
         mongoSpan(it, 1, "count", collectionName, dbName, span(0)) {
           assert it == "{\"count\":\"$collectionName\",\"query\":{}}" ||
+            it == "{\"count\":\"$collectionName\"}" ||
             it == "{\"count\": \"$collectionName\", \"query\": {}, \"\$db\": \"?\", \"\$readPreference\": {\"mode\": \"?\"}}"
           true
         }
@@ -163,6 +164,7 @@ abstract class AbstractMongoClientTest<T> extends AgentInstrumentationSpecificat
         }
         mongoSpan(it, 2, "count", collectionName, dbName, span(0)) {
           assert it == "{\"count\":\"$collectionName\",\"query\":{}}" ||
+            it == "{\"count\":\"$collectionName\"}" ||
             it == "{\"count\": \"$collectionName\", \"query\": {}, \"\$db\": \"?\", \"\$readPreference\": {\"mode\": \"?\"}}"
           true
         }
@@ -193,6 +195,7 @@ abstract class AbstractMongoClientTest<T> extends AgentInstrumentationSpecificat
         }
         mongoSpan(it, 2, "count", collectionName, dbName, span(0)) {
           assert it == "{\"count\":\"$collectionName\",\"query\":{}}" ||
+            it == "{\"count\":\"$collectionName\"}" ||
             it == "{\"count\": \"$collectionName\", \"query\": {}, \"\$db\": \"?\", \"\$readPreference\": {\"mode\": \"?\"}}"
           true
         }
@@ -223,6 +226,7 @@ abstract class AbstractMongoClientTest<T> extends AgentInstrumentationSpecificat
         }
         mongoSpan(it, 2, "count", collectionName, dbName, span(0)) {
           assert it == "{\"count\":\"$collectionName\",\"query\":{}}" ||
+            it == "{\"count\":\"$collectionName\"}" ||
             it == "{\"count\": \"$collectionName\", \"query\": {}, \"\$db\": \"?\", \"\$readPreference\": {\"mode\": \"?\"}}"
           true
         }