From 9234dbdfbb4b3284305b55258a505a179f0fecb2 Mon Sep 17 00:00:00 2001
From: Benjamin Rewis <32186188+benjirewis@users.noreply.github.com>
Date: Tue, 30 Nov 2021 18:13:04 -0500
Subject: [PATCH] GODRIVER-2242 Update estimatedDocumentCount test for ADL
 (#821)

---
 .../estimatedDocumentCount.json               | 21 +++++++++++++++++--
 .../estimatedDocumentCount.yml                |  7 ++++++-
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/data/atlas-data-lake-testing/estimatedDocumentCount.json b/data/atlas-data-lake-testing/estimatedDocumentCount.json
index d039a51f06..87b385208d 100644
--- a/data/atlas-data-lake-testing/estimatedDocumentCount.json
+++ b/data/atlas-data-lake-testing/estimatedDocumentCount.json
@@ -15,8 +15,25 @@
         {
           "command_started_event": {
             "command": {
-              "count": "driverdata"
-            }
+              "aggregate": "driverdata",
+              "pipeline": [
+                {
+                  "$collStats": {
+                    "count": {}
+                  }
+                },
+                {
+                  "$group": {
+                    "_id": 1,
+                    "n": {
+                      "$sum": "$count"
+                    }
+                  }
+                }
+              ]
+            },
+            "command_name": "aggregate",
+            "database_name": "test"
           }
         }
       ]
diff --git a/data/atlas-data-lake-testing/estimatedDocumentCount.yml b/data/atlas-data-lake-testing/estimatedDocumentCount.yml
index b4cecc9522..144831173f 100644
--- a/data/atlas-data-lake-testing/estimatedDocumentCount.yml
+++ b/data/atlas-data-lake-testing/estimatedDocumentCount.yml
@@ -13,4 +13,9 @@ tests:
       -
         command_started_event:
           command:
-            count: *collection_name
+            aggregate: *collection_name
+            pipeline:
+              - $collStats: { count: {} }
+              - $group: { _id: 1, n: { $sum: $count }}
+          command_name: aggregate
+          database_name: *database_name