3535# class DigitalOceanTests(unittest.TestCase, TestCaseMixin):
3636class DigitalOcean_v2_Tests (LibcloudTestCase ):
3737 def setUp (self ):
38- DigitalOceanNodeDriver .connectionCls .conn_class = DigitalOceanMockHttp
39- DigitalOceanMockHttp .type = None
38+ DigitalOceanNodeDriver .connectionCls .conn_class = DigitalOceanComputeMockHttp
39+ DigitalOceanComputeMockHttp .type = None
4040 self .driver = DigitalOceanNodeDriver (* DIGITALOCEAN_v2_PARAMS )
4141
42+ def tearDown (self ):
43+ DigitalOceanComputeMockHttp .type = None
44+
4245 def test_v1_Error (self ):
4346 self .assertRaises (
4447 DigitalOcean_v1_Error ,
@@ -56,7 +59,7 @@ def test_v2_uses_v1_key(self):
5659 )
5760
5861 def test_authentication (self ):
59- DigitalOceanMockHttp .type = "UNAUTHORIZED"
62+ DigitalOceanComputeMockHttp .type = "UNAUTHORIZED"
6063 self .assertRaises (InvalidCredsError , self .driver .list_nodes )
6164
6265 def test_list_images_success (self ):
@@ -128,7 +131,7 @@ def test_create_node_invalid_size(self):
128131 size = self .driver .list_sizes ()[0 ]
129132 location = self .driver .list_locations ()[0 ]
130133
131- DigitalOceanMockHttp .type = "INVALID_IMAGE"
134+ DigitalOceanComputeMockHttp .type = "INVALID_IMAGE"
132135 expected_msg = (
133136 r"You specified an invalid image for Droplet creation."
134137 + r" \(code: (404|HTTPStatus.NOT_FOUND)\)"
@@ -146,13 +149,13 @@ def test_create_node_invalid_size(self):
146149
147150 def test_reboot_node_success (self ):
148151 node = self .driver .list_nodes ()[0 ]
149- DigitalOceanMockHttp .type = "REBOOT"
152+ DigitalOceanComputeMockHttp .type = "REBOOT"
150153 result = self .driver .reboot_node (node )
151154 self .assertTrue (result )
152155
153156 def test_create_image_success (self ):
154157 node = self .driver .list_nodes ()[0 ]
155- DigitalOceanMockHttp .type = "SNAPSHOT"
158+ DigitalOceanComputeMockHttp .type = "SNAPSHOT"
156159 result = self .driver .create_image (node , "My snapshot" )
157160 self .assertTrue (result )
158161
@@ -164,62 +167,62 @@ def test_get_image_success(self):
164167
165168 def test_delete_image_success (self ):
166169 image = self .driver .get_image (12345 )
167- DigitalOceanMockHttp .type = "DESTROY"
170+ DigitalOceanComputeMockHttp .type = "DESTROY"
168171 result = self .driver .delete_image (image )
169172 self .assertTrue (result )
170173
171174 def test_ex_power_on_node_success (self ):
172175 node = self .driver .list_nodes ()[0 ]
173- DigitalOceanMockHttp .type = "POWERON"
176+ DigitalOceanComputeMockHttp .type = "POWERON"
174177 result = self .driver .ex_power_on_node (node )
175178 self .assertTrue (result )
176179
177180 def test_ex_shutdown_node_success (self ):
178181 node = self .driver .list_nodes ()[0 ]
179- DigitalOceanMockHttp .type = "SHUTDOWN"
182+ DigitalOceanComputeMockHttp .type = "SHUTDOWN"
180183 result = self .driver .ex_shutdown_node (node )
181184 self .assertTrue (result )
182185
183186 def test_ex_hard_reboot_success (self ):
184187 node = self .driver .list_nodes ()[0 ]
185- DigitalOceanMockHttp .type = "POWERCYCLE"
188+ DigitalOceanComputeMockHttp .type = "POWERCYCLE"
186189 result = self .driver .ex_hard_reboot (node )
187190 self .assertTrue (result )
188191
189192 def test_ex_rebuild_node_success (self ):
190193 node = self .driver .list_nodes ()[0 ]
191- DigitalOceanMockHttp .type = "REBUILD"
194+ DigitalOceanComputeMockHttp .type = "REBUILD"
192195 result = self .driver .ex_rebuild_node (node )
193196 self .assertTrue (result )
194197
195198 def test_ex_resize_node_success (self ):
196199 node = self .driver .list_nodes ()[0 ]
197200 size = self .driver .list_sizes ()[0 ]
198- DigitalOceanMockHttp .type = "RESIZE"
201+ DigitalOceanComputeMockHttp .type = "RESIZE"
199202 result = self .driver .ex_resize_node (node , size )
200203 self .assertTrue (result )
201204
202205 def test_destroy_node_success (self ):
203206 node = self .driver .list_nodes ()[0 ]
204- DigitalOceanMockHttp .type = "DESTROY"
207+ DigitalOceanComputeMockHttp .type = "DESTROY"
205208 result = self .driver .destroy_node (node )
206209 self .assertTrue (result )
207210
208211 def test_ex_change_kernel_success (self ):
209212 node = self .driver .list_nodes ()[0 ]
210- DigitalOceanMockHttp .type = "KERNELCHANGE"
213+ DigitalOceanComputeMockHttp .type = "KERNELCHANGE"
211214 result = self .driver .ex_change_kernel (node , 7515 )
212215 self .assertTrue (result )
213216
214217 def test_ex_enable_ipv6_success (self ):
215218 node = self .driver .list_nodes ()[0 ]
216- DigitalOceanMockHttp .type = "ENABLEIPV6"
219+ DigitalOceanComputeMockHttp .type = "ENABLEIPV6"
217220 result = self .driver .ex_enable_ipv6 (node )
218221 self .assertTrue (result )
219222
220223 def test_ex_rename_node_success (self ):
221224 node = self .driver .list_nodes ()[0 ]
222- DigitalOceanMockHttp .type = "RENAME"
225+ DigitalOceanComputeMockHttp .type = "RENAME"
223226 result = self .driver .ex_rename_node (node , "fedora helios" )
224227 self .assertTrue (result )
225228
@@ -231,7 +234,7 @@ def test_list_key_pairs(self):
231234 self .assertEqual (keys [0 ].public_key , "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5 example" )
232235
233236 def test_create_key_pair (self ):
234- DigitalOceanMockHttp .type = "CREATE"
237+ DigitalOceanComputeMockHttp .type = "CREATE"
235238 key = self .driver .create_key_pair (
236239 name = "test1" , public_key = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQsxRiUKn example"
237240 )
@@ -250,7 +253,7 @@ def test__paginated_request_single_page(self):
250253 self .assertEqual (nodes [0 ]["size_slug" ], "s-1vcpu-1gb" )
251254
252255 def test__paginated_request_two_pages (self ):
253- DigitalOceanMockHttp .type = "PAGE_ONE"
256+ DigitalOceanComputeMockHttp .type = "PAGE_ONE"
254257 nodes = self .driver ._paginated_request ("/v2/droplets" , "droplets" )
255258 self .assertEqual (len (nodes ), 2 )
256259
@@ -264,13 +267,13 @@ def test_list_volumes(self):
264267 self .assertEqual (volume .driver , self .driver )
265268
266269 def test_list_volumes_empty (self ):
267- DigitalOceanMockHttp .type = "EMPTY"
270+ DigitalOceanComputeMockHttp .type = "EMPTY"
268271 volumes = self .driver .list_volumes ()
269272 self .assertEqual (len (volumes ), 0 )
270273
271274 def test_create_volume (self ):
272275 nyc1 = [r for r in self .driver .list_locations () if r .id == "nyc1" ][0 ]
273- DigitalOceanMockHttp .type = "CREATE"
276+ DigitalOceanComputeMockHttp .type = "CREATE"
274277 volume = self .driver .create_volume (4 , "example" , nyc1 )
275278 self .assertEqual (volume .id , "62766883-2c28-11e6-b8e6-000f53306ae1" )
276279 self .assertEqual (volume .name , "example" )
@@ -280,19 +283,19 @@ def test_create_volume(self):
280283 def test_attach_volume (self ):
281284 node = self .driver .list_nodes ()[0 ]
282285 volume = self .driver .list_volumes ()[0 ]
283- DigitalOceanMockHttp .type = "ATTACH"
286+ DigitalOceanComputeMockHttp .type = "ATTACH"
284287 resp = self .driver .attach_volume (node , volume )
285288 self .assertTrue (resp )
286289
287290 def test_detach_volume (self ):
288291 volume = self .driver .list_volumes ()[0 ]
289- DigitalOceanMockHttp .type = "DETACH"
292+ DigitalOceanComputeMockHttp .type = "DETACH"
290293 resp = self .driver .detach_volume (volume )
291294 self .assertTrue (resp )
292295
293296 def test_destroy_volume (self ):
294297 volume = self .driver .list_volumes ()[0 ]
295- DigitalOceanMockHttp .type = "DESTROY"
298+ DigitalOceanComputeMockHttp .type = "DESTROY"
296299 resp = self .driver .destroy_volume (volume )
297300 self .assertTrue (resp )
298301
@@ -307,7 +310,7 @@ def test_list_volume_snapshots(self):
307310
308311 def test_create_volume_snapshot (self ):
309312 volume = self .driver .list_volumes ()[0 ]
310- DigitalOceanMockHttp .type = "CREATE"
313+ DigitalOceanComputeMockHttp .type = "CREATE"
311314 snapshot = self .driver .create_volume_snapshot (volume , "test-snapshot" )
312315 self .assertEqual (snapshot .id , "c0def940-9324-11e6-9a56-000f533176b1" )
313316 self .assertEqual (snapshot .name , "test-snapshot" )
@@ -316,7 +319,7 @@ def test_create_volume_snapshot(self):
316319 def test_delete_volume_snapshot (self ):
317320 volume = self .driver .list_volumes ()[0 ]
318321 snapshot = self .driver .list_volume_snapshots (volume )[0 ]
319- DigitalOceanMockHttp .type = "DELETE"
322+ DigitalOceanComputeMockHttp .type = "DELETE"
320323 result = self .driver .delete_volume_snapshot (snapshot )
321324 self .assertTrue (result )
322325
@@ -396,7 +399,7 @@ def test_ex_detach_floating_ip_from_node(self):
396399 self .assertTrue (ret )
397400
398401
399- class DigitalOceanMockHttp (MockHttp ):
402+ class DigitalOceanComputeMockHttp (MockHttp ):
400403 fixtures = ComputeFileFixtures ("digitalocean_v2" )
401404
402405 def _v2_regions (self , method , url , body , headers ):
0 commit comments