Additional changes to the DigitalOcean tests to avoid cross test

pollution and related failures when running tests in parallel.
apache · Apr 16, 2024 · b78b55b · b78b55b
1 parent 841dea7
commit b78b55b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/libcloud/test/compute/test_digitalocean_v2.py b/libcloud/test/compute/test_digitalocean_v2.py
@@ -24,7 +24,7 @@
 from libcloud.test.secrets import DIGITALOCEAN_v1_PARAMS, DIGITALOCEAN_v2_PARAMS
 from libcloud.utils.iso8601 import UTC
 from libcloud.test.file_fixtures import ComputeFileFixtures
-from libcloud.common.digitalocean import DigitalOcean_v1_Error
+from libcloud.common.digitalocean import DigitalOcean_v1_Error, DigitalOceanBaseDriver
 from libcloud.compute.drivers.digitalocean import DigitalOceanNodeDriver
 
 try:
@@ -36,13 +36,16 @@
 # class DigitalOceanTests(unittest.TestCase, TestCaseMixin):
 class DigitalOcean_v2_Tests(LibcloudTestCase):
     def setUp(self):
+        DigitalOceanBaseDriver.connectionCls.conn_class = DigitalOceanComputeMockHttp
         DigitalOceanNodeDriver.connectionCls.conn_class = DigitalOceanComputeMockHttp
         DigitalOceanComputeMockHttp.type = None
         self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
 
     def tearDown(self):
         LibcloudConnection.type = None
         DigitalOceanComputeMockHttp.type = None
+        DigitalOceanBaseDriver.connectionCls.conn_class = LibcloudConnection
+        DigitalOceanNodeDriver.connectionCls.conn_class = LibcloudConnection
 
     def test_v1_Error(self):
         self.assertRaises(

diff --git a/libcloud/test/dns/test_digitalocean.py b/libcloud/test/dns/test_digitalocean.py
@@ -21,18 +21,22 @@
 from libcloud.utils.py3 import httplib
 from libcloud.test.secrets import DIGITALOCEAN_v2_PARAMS
 from libcloud.test.file_fixtures import DNSFileFixtures
+from libcloud.common.digitalocean import DigitalOceanBaseDriver
 from libcloud.dns.drivers.digitalocean import DigitalOceanDNSDriver
 
 
 class DigitalOceanDNSTests(LibcloudTestCase):
     def setUp(self):
+        DigitalOceanBaseDriver.connectionCls.conn_class = DigitalOceanDNSMockHttp
         DigitalOceanDNSDriver.connectionCls.conn_class = DigitalOceanDNSMockHttp
         DigitalOceanDNSMockHttp.type = None
         self.driver = DigitalOceanDNSDriver(*DIGITALOCEAN_v2_PARAMS)
 
     def tearDown(self):
         LibcloudConnection.type = None
         DigitalOceanDNSMockHttp.type = None
+        DigitalOceanBaseDriver.connectionCls.conn_class = LibcloudConnection
+        DigitalOceanDNSDriver.connectionCls.conn_class = LibcloudConnection
 
     def test_list_zones(self):
         zones = self.driver.list_zones()