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 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 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 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 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 }