DRIVERS-2951 Test that serverMonitoringMode=poll waits after a succes…

…sful heartbeat
mongodb · Aug 2, 2024 · 8692940 · 8692940
1 parent 34b9d46
commit 8692940
Show file tree

Hide file tree

Showing 2 changed files with 106 additions and 0 deletions.
diff --git a/source/server-discovery-and-monitoring/tests/unified/serverMonitoringMode.json b/source/server-discovery-and-monitoring/tests/unified/serverMonitoringMode.json
diff --git a/source/server-discovery-and-monitoring/tests/unified/serverMonitoringMode.yml b/source/server-discovery-and-monitoring/tests/unified/serverMonitoringMode.yml
@@ -171,3 +171,45 @@ tests:
       # Wait for the second serverHeartbeatStartedEvent to ensure we do not stream.
       - *waitForSecondHeartbeatStarted
     expectEvents: *pollingStartedEvents
+
+  - description: "poll waits after successful heartbeat"
+    operations:
+      - name: createEntities
+        object: testRunner
+        arguments:
+          entities:
+            - client:
+                id: client
+                uriOptions:
+                  serverMonitoringMode: "poll"
+                  heartbeatFrequencyMS: 1000000
+                useMultipleMongoses: false
+                observeEvents:
+                  - serverHeartbeatStartedEvent
+                  - serverHeartbeatSucceededEvent
+                  - serverHeartbeatFailedEvent
+            - database:
+                id: db
+                client: client
+                databaseName: sdam-tests
+      # Wait for the first serverHeartbeatSucceededEvent to ensure we start polling.
+      - name: waitForEvent
+        object: testRunner
+        arguments:
+          client: client
+          event:
+            serverHeartbeatSucceededEvent: {}
+          count: 1
+      # Wait for a bit longer to ensure we wait heartbeatFrequencyMS before starting the next check.
+      - name: wait
+        object: testRunner
+        arguments:
+          ms: 500
+      - name: assertEventCount
+        object: testRunner
+        arguments:
+          client: client
+          event:
+            serverHeartbeatStartedEvent: {}
+          count: 1
+#    expectEvents: *streamingStartedEvents