|
38 | 38 | createNetworkRulesForVM, |
39 | 39 | verifyNetworkState) |
40 | 40 | from marvin.codes import (PASS, FAIL, FAILED, UNKNOWN, FAULT, MASTER, |
41 | | - NAT_RULE, STATIC_NAT_RULE, ERROR_CODE_530) |
| 41 | + NAT_RULE, STATIC_NAT_RULE) |
42 | 42 | import netaddr |
43 | 43 |
|
44 | 44 | import random |
@@ -270,10 +270,8 @@ def test_update_cidr_multiple_vms_not_all_inclusive(self): |
270 | 270 | except Exception as e: |
271 | 271 | self.fail("VM creation failed: %s" % e) |
272 | 272 |
|
273 | | - response = isolated_network.update(self.apiclient, guestvmcidr=guest_vm_cidr) |
274 | | - self.assertEqual(response.errorcode, ERROR_CODE_530, "Job should \ |
275 | | - have failed with error code %s, instead got response \ |
276 | | - %s" % (ERROR_CODE_530, str(response))) |
| 273 | + with self.assertRaises(Exception): |
| 274 | + isolated_network.update(self.apiclient, guestvmcidr=guest_vm_cidr) |
277 | 275 | return |
278 | 276 |
|
279 | 277 | @attr(tags=["advanced"]) |
@@ -302,10 +300,8 @@ def test_update_cidr_single_vm_not_inclusive(self): |
302 | 300 | except Exception as e: |
303 | 301 | self.fail("VM creation failed: %s" % e) |
304 | 302 |
|
305 | | - response = isolated_network.update(self.apiclient, guestvmcidr=guest_vm_cidr) |
306 | | - self.assertEqual(response.errorcode, ERROR_CODE_530, "Job should \ |
307 | | - have failed with error code %s, instead got response \ |
308 | | - %s" % (ERROR_CODE_530, str(response))) |
| 303 | + with self.assertRaises(Exception): |
| 304 | + isolated_network.update(self.apiclient, guestvmcidr=guest_vm_cidr) |
309 | 305 | return |
310 | 306 |
|
311 | 307 | @data(NAT_RULE, STATIC_NAT_RULE) |
@@ -1083,10 +1079,8 @@ def test_network_not_implemented(self): |
1083 | 1079 | else: |
1084 | 1080 | isolated_network = resultSet[1] |
1085 | 1081 |
|
1086 | | - response = isolated_network.update(self.apiclient, guestvmcidr="10.1.1.0/26") |
1087 | | - self.assertEqual(response.errorcode, ERROR_CODE_530, "Job should \ |
1088 | | - have failed with error code %s, instead got response \ |
1089 | | - %s" % (ERROR_CODE_530, str(response))) |
| 1082 | + with self.assertRaises(Exception): |
| 1083 | + response = isolated_network.update(self.apiclient, guestvmcidr="10.1.1.0/26") |
1090 | 1084 | return |
1091 | 1085 |
|
1092 | 1086 | @attr(tags=["advanced", "selfservice"]) |
|
0 commit comments