Change way to start simulators
[transportpce.git] / tests / transportpce_tests / 7.1 / test_pce_400G.py
index 10fb918af0f1114a6ce437490763e8e2995f2804..189e9245bfc6362af9ddefa5b9b26439e20552ac 100644 (file)
@@ -16,7 +16,8 @@ import os
 import sys
 import time
 import requests
-from common import test_utils
+sys.path.append('transportpce_tests/common/')
+import test_utils
 
 
 class TransportPCE400Gtesting(unittest.TestCase):
@@ -67,7 +68,7 @@ class TransportPCE400Gtesting(unittest.TestCase):
     # Load port mapping
     def test_01_load_port_mapping(self):
         response = test_utils.put_jsonrequest(test_utils.URL_FULL_PORTMAPPING, self.port_mapping_data)
-        self.assertEqual(response.status_code, requests.codes.ok)
+        self.assertIn(response.status_code, (requests.codes.ok, requests.codes.created))
         time.sleep(2)
 
     # Load simple bidirectional topology