Improvements to the OSDF changes for new PEF APIs

[optf/osdf.git] / test / test_PolicyCalls.py

diff --git a/test/test_PolicyCalls.py b/test/test_PolicyCalls.py
index 4c9366a..0b17081 100644 (file)
--- a/test/test_PolicyCalls.py
+++ b/test/test_PolicyCalls.py
@@ -24,7 +24,7 @@ from osdf.adapters.policy import interface
 from osdf.utils.interfaces import RestClient, json_from_file
 import yaml
 from mock import patch
-from osdf.optimizers.placementopt.conductor import translation
+from apps.placement.optimizers.conductor import translation
 from osdf.operation.exceptions import BusinessException
 
 
@@ -63,7 +63,7 @@ class TestPolicyCalls(unittest.TestCase):
                                                       "test/placement-tests/policy_response2.json")
         with patch('osdf.adapters.policy.interface.policy_api_call', return_value=policy_response):
             policy_list = interface.remote_api(req_json, osdf_config, service_type="placement")
-            policy_type = [policy['content']['policyType'] for policy in policy_list]
+            policy_type = [policy[list(policy.keys())[0]]['type'] for policy in policy_list]
             #self.assertEqual(set(policy_type), {'hpaPolicy', 'SubscriberPolicy'})
 
     def failure_policy_call(self, req_json_file, resp_json_file):
@@ -99,7 +99,8 @@ class TestPolicyCalls(unittest.TestCase):
         req_json = "./test/placement-tests/request.json"
         req_json = json.loads(open(req_json).read())
         # need to run this only on vnf policies
-        vnf_policies = [x for x in self.policies if x["content"]["policyType"] == "vnfPolicy"]
+        vnf_policies = [x for x in self.policies if x[list(x.keys())[0]]["type"] ==
+                        "onap.policies.optimization.VnfPolicy"]
         gen_demands = translation.gen_demands(req_json, vnf_policies)
         for action in req_json['placementInfo']['placementDemands']:
             actions_list.append(action['resourceModuleName'])