PYTHON-2800 Add Atlas connectivity tests for MongoDB Serverless (#669)

mongodb · Jul 12, 2021 · 948ebb2 · 948ebb2
1 parent 834500d
commit 948ebb2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/.evergreen/config.yml b/.evergreen/config.yml
@@ -450,11 +450,13 @@ functions:
           export ATLAS_SHRD='${atlas_shrd}'
           export ATLAS_TLS11='${atlas_tls11}'
           export ATLAS_TLS12='${atlas_tls12}'
+          export ATLAS_SERVERLESS='${atlas_serverless}'
           export ATLAS_SRV_FREE='${atlas_srv_free}'
           export ATLAS_SRV_REPL='${atlas_srv_repl}'
           export ATLAS_SRV_SHRD='${atlas_srv_shrd}'
           export ATLAS_SRV_TLS11='${atlas_srv_tls11}'
           export ATLAS_SRV_TLS12='${atlas_srv_tls12}'
+          export ATLAS_SRV_SERVERLESS='${atlas_srv_serverless}'
           EOT
     - command: shell.exec
       type: test

diff --git a/test/atlas/test_connection.py b/test/atlas/test_connection.py
@@ -38,11 +38,13 @@
     "ATLAS_FREE": os.environ.get("ATLAS_FREE"),
     "ATLAS_TLS11": os.environ.get("ATLAS_TLS11"),
     "ATLAS_TLS12": os.environ.get("ATLAS_TLS12"),
+    "ATLAS_SERVERLESS": os.environ.get("ATLAS_SERVERLESS"),
     "ATLAS_SRV_REPL": os.environ.get("ATLAS_SRV_REPL"),
     "ATLAS_SRV_SHRD": os.environ.get("ATLAS_SRV_SHRD"),
     "ATLAS_SRV_FREE": os.environ.get("ATLAS_SRV_FREE"),
     "ATLAS_SRV_TLS11": os.environ.get("ATLAS_SRV_TLS11"),
     "ATLAS_SRV_TLS12": os.environ.get("ATLAS_SRV_TLS12"),
+    "ATLAS_SRV_SERVERLESS": os.environ.get("ATLAS_SRV_SERVERLESS"),
 }
 
 # Set this variable to true to run the SRV tests even when dnspython is not
@@ -77,6 +79,9 @@ def test_tls_11(self):
     def test_tls_12(self):
         connect(URIS['ATLAS_TLS12'])
 
+    def test_serverless(self):
+        connect(URIS['ATLAS_SERVERLESS'])
+
     def connect_srv(self, uri):
         connect(uri)
         self.assertIn('mongodb+srv://', uri)
@@ -102,6 +107,10 @@ def test_srv_tls_11(self):
     def test_srv_tls_12(self):
         self.connect_srv(URIS['ATLAS_SRV_TLS12'])
 
+    @unittest.skipUnless(HAS_DNS or MUST_TEST_SRV, 'SRV requires dnspython')
+    def test_srv_serverless(self):
+        self.connect_srv(URIS['ATLAS_SRV_SERVERLESS'])
+
     def test_uniqueness(self):
         """Ensure that we don't accidentally duplicate the test URIs."""
         uri_to_names = defaultdict(list)
@@ -114,6 +123,5 @@ def test_uniqueness(self):
                                      'duplicate values: %s' % (duplicates,))
 
 
-
 if __name__ == '__main__':
     unittest.main()