fix tests

Author: blink1073

test/asynchronous/utils.py

@@ -242,6 +242,7 @@ def __init__(self, address, options, is_sdam=False, client_id=None):
         self.opts = options
         self.operation_count = 0
         self.conns = []
+        self.state = 0
 
     def stale_generation(self, gen, service_id):
         return self.gen.stale(gen, service_id)

test/connection_logging/connection-logging.json

@@ -713,6 +713,9 @@
                 "durationMS": {
                    "$$type": "int"
                 },
+                "attempt": {
+                   "$$type": "int"
+                },
                 "serverPort": {
                   "$$type": [
                     "int",

test/discovery_and_monitoring/unified/backoff-network-error.json

@@ -132,9 +132,24 @@
           "arguments": {
             "client": "client",
             "event": {
-              "poolBackoffEvent": {}
+              "poolBackoffEvent": {
+                "attempt": 1
+              }
+            },
+            "count": 1
+          }
+        },
+        {
+          "name": "waitForEvent",
+          "object": "testRunner",
+          "arguments": {
+            "client": "client",
+            "event": {
+              "poolBackoffEvent": {
+                "attempt": 2
+              }
             },
-            "count": 2
+             "count": 1
           }
         }
       ],

test/unified_format_shared.py

@@ -622,6 +622,7 @@ def match_event(self, expectation, actual):
         elif name == "poolBackoffEvent":
             self.test.assertIsInstance(actual, PoolBackoffEvent)
             self.test.assertIsInstance(actual.duration_ms, int)
+            self.test.assertIsInstance(actual.attempt, int)
         elif name == "connectionCreatedEvent":
             self.test.assertIsInstance(actual, ConnectionCreatedEvent)
         elif name == "connectionReadyEvent":

test/utils.py

@@ -240,6 +240,7 @@ def __init__(self, address, options, is_sdam=False, client_id=None):
         self.opts = options
         self.operation_count = 0
         self.conns = []
+        self.state = 0
 
     def stale_generation(self, gen, service_id):
         return self.gen.stale(gen, service_id)