X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=tests%2Funit%2Ftest_ocloud.py;h=a29c2a60b021784628ad74649d2b85af1d352477;hb=7c167626f2692556b1fe073f87150f54a8c9910a;hp=95bd9cd717aa6f40a4e86e9837405a27f620cc1f;hpb=6bc82510fcc73a6dc2f817caa2ce74b52daa345c;p=pti%2Fo2.git diff --git a/tests/unit/test_ocloud.py b/tests/unit/test_ocloud.py index 95bd9cd..a29c2a6 100644 --- a/tests/unit/test_ocloud.py +++ b/tests/unit/test_ocloud.py @@ -13,8 +13,10 @@ # limitations under the License. import uuid +from unittest.mock import MagicMock +# from o2dms.domain import dms -from o2ims.domain import ocloud +from o2ims.domain import ocloud, subscription_obj, configuration_obj from o2ims.domain import resource_type as rt from o2ims.views import ocloud_view from o2common.config import config @@ -71,7 +73,7 @@ def test_new_resource(): resource_type_id1 = str(uuid.uuid4()) resource_pool_id1 = str(uuid.uuid4()) resource1 = ocloud.Resource( - resource_id1, resource_type_id1, resource_pool_id1) + resource_id1, resource_type_id1, resource_pool_id1, 'resource1') assert resource_id1 is not None and resource1.resourceId == resource_id1 @@ -87,35 +89,49 @@ def test_new_deployment_manager(): def test_new_subscription(): subscription_id1 = str(uuid.uuid4()) - subscription1 = ocloud.Subscription( + subscription1 = subscription_obj.Subscription( subscription_id1, "https://callback/uri/write/here") assert subscription_id1 is not None and\ subscription1.subscriptionId == subscription_id1 +def test_new_configuration(): + configuration_id1 = str(uuid.uuid4()) + configuration1 = configuration_obj.Configuration( + configuration_id1, "https://callback/uri/write/here", + "SMO") + assert configuration_id1 is not None and\ + configuration1.configurationId == configuration_id1 + + def test_view_olcouds(mock_uow): session, uow = mock_uow ocloud1_UUID = str(uuid.uuid4) - session.return_value.execute.return_value = [ - {"oCloudId": ocloud1_UUID}] + ocloud1 = MagicMock() + ocloud1.serialize.return_value = { + 'oCloudId': ocloud1_UUID, 'name': 'ocloud1'} + session.return_value.query.return_value = [ocloud1] ocloud_list = ocloud_view.oclouds(uow) - assert str(ocloud_list[0].get("oCloudId")) == ocloud1_UUID + # assert str(ocloud_list[0].get("oCloudId")) == ocloud1_UUID + assert len(ocloud_list) == 1 def test_view_olcoud_one(mock_uow): session, uow = mock_uow ocloud1_UUID = str(uuid.uuid4) - session.return_value.execute.return_value.first.return_value = None + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = None # Query return None ocloud_res = ocloud_view.ocloud_one(ocloud1_UUID, uow) assert ocloud_res is None - session.return_value.execute.return_value.first.return_value = { - "oCloudId": ocloud1_UUID} + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = { + "oCloudId": ocloud1_UUID} ocloud_res = ocloud_view.ocloud_one(ocloud1_UUID, uow) assert str(ocloud_res.get("oCloudId")) == ocloud1_UUID @@ -125,9 +141,10 @@ def test_view_resource_types(mock_uow): session, uow = mock_uow resource_type_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value = [ - {"resourceTypeId": resource_type_id1} - ] + restype1 = MagicMock() + restype1.serialize.return_value = { + "resourceTypeId": resource_type_id1} + session.return_value.query.return_value = [restype1] resource_type_list = ocloud_view.resource_types(uow) assert str(resource_type_list[0].get( @@ -138,15 +155,17 @@ def test_view_resource_type_one(mock_uow): session, uow = mock_uow resource_type_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value.first.return_value = None + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = None # Query return None resource_type_res = ocloud_view.resource_type_one( resource_type_id1, uow) assert resource_type_res is None - session.return_value.execute.return_value.first.return_value = { - "resourceTypeId": resource_type_id1} + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = { + "resourceTypeId": resource_type_id1} resource_type_res = ocloud_view.resource_type_one(resource_type_id1, uow) assert str(resource_type_res.get("resourceTypeId")) == resource_type_id1 @@ -156,9 +175,10 @@ def test_view_resource_pools(mock_uow): session, uow = mock_uow resource_pool_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value = [ - {"resourcePoolId": resource_pool_id1} - ] + respool1 = MagicMock() + respool1.serialize.return_value = { + "resourcePoolId": resource_pool_id1} + session.return_value.query.return_value = [respool1] resource_pool_list = ocloud_view.resource_pools(uow) assert str(resource_pool_list[0].get( @@ -169,16 +189,18 @@ def test_view_resource_pool_one(mock_uow): session, uow = mock_uow resource_pool_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value.first.return_value = None + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = None # Query return None resource_pool_res = ocloud_view.resource_pool_one( resource_pool_id1, uow) assert resource_pool_res is None - session.return_value.execute.return_value.first.return_value = { - "resourcePoolId": resource_pool_id1 - } + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = { + "resourcePoolId": resource_pool_id1 + } resource_pool_res = ocloud_view.resource_pool_one(resource_pool_id1, uow) assert str(resource_pool_res.get("resourcePoolId")) == resource_pool_id1 @@ -189,10 +211,12 @@ def test_view_resources(mock_uow): resource_id1 = str(uuid.uuid4()) resource_pool_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value = [{ + res1 = MagicMock() + res1.serialize.return_value = { "resourceId": resource_id1, "resourcePoolId": resource_pool_id1 - }] + } + session.return_value.query.return_value.filter_by.return_value = [res1] resource_list = ocloud_view.resources(resource_pool_id1, uow) assert str(resource_list[0].get("resourceId")) == resource_id1 @@ -204,16 +228,18 @@ def test_view_resource_one(mock_uow): resource_id1 = str(uuid.uuid4()) resource_pool_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value.first.return_value = None + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = None # Query return None resource_res = ocloud_view.resource_one(resource_id1, uow) assert resource_res is None - session.return_value.execute.return_value.first.return_value = { - "resourceId": resource_id1, - "resourcePoolId": resource_pool_id1 - } + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = { + "resourceId": resource_id1, + "resourcePoolId": resource_pool_id1 + } resource_res = ocloud_view.resource_one(resource_id1, uow) assert str(resource_res.get("resourceId")) == resource_id1 @@ -223,9 +249,11 @@ def test_view_deployment_managers(mock_uow): session, uow = mock_uow deployment_manager_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value = [{ + dm1 = MagicMock() + dm1.serialize.return_value = { "deploymentManagerId": deployment_manager_id1, - }] + } + session.return_value.query.return_value = [dm1] deployment_manager_list = ocloud_view.deployment_managers(uow) assert str(deployment_manager_list[0].get( @@ -236,30 +264,65 @@ def test_view_deployment_manager_one(mock_uow): session, uow = mock_uow deployment_manager_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value.first.return_value = None + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = None # Query return None deployment_manager_res = ocloud_view.deployment_manager_one( deployment_manager_id1, uow) assert deployment_manager_res is None - session.return_value.execute.return_value.first.return_value = { - "deploymentManagerId": deployment_manager_id1, - } + dms_endpoint = "http://o2:30205/o2dms/v1/uuid" + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = { + "deploymentManagerId": deployment_manager_id1, + "deploymentManagementServiceEndpoint": dms_endpoint, + "profile": {} + } + # profile default deployment_manager_res = ocloud_view.deployment_manager_one( deployment_manager_id1, uow) assert str(deployment_manager_res.get( "deploymentManagerId")) == deployment_manager_id1 + assert str(deployment_manager_res.get( + 'deploymentManagementServiceEndpoint')) == dms_endpoint + assert deployment_manager_res.get('profile') is None + + # profile sol018 + profileName = ocloud.DeploymentManagerProfileSOL018 + cluster_endpoint = "https://test_k8s:6443" + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value['profile'] = { + "cluster_api_endpoint": cluster_endpoint + } + deployment_manager_res = ocloud_view.deployment_manager_one( + deployment_manager_id1, uow, profile=profileName) + assert str(deployment_manager_res.get( + 'deploymentManagementServiceEndpoint')) == cluster_endpoint + assert str(deployment_manager_res.get( + "profileName")) == profileName + + # profile wrong name + profileName = 'wrong_profile' + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value['profile'] = { + "cluster_api_endpoint": cluster_endpoint + } + deployment_manager_res = ocloud_view.deployment_manager_one( + deployment_manager_id1, uow, profile=profileName) + assert deployment_manager_res is None def test_view_subscriptions(mock_uow): session, uow = mock_uow subscription_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value = [{ + sub1 = MagicMock() + sub1.serialize.return_value = { "subscriptionId": subscription_id1, - }] + } + session.return_value.query.return_value = [sub1] subscription_list = ocloud_view.subscriptions(uow) assert str(subscription_list[0].get( @@ -270,16 +333,18 @@ def test_view_subscription_one(mock_uow): session, uow = mock_uow subscription_id1 = str(uuid.uuid4()) - session.return_value.execute.return_value.first.return_value = None + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = None # Query return None subscription_res = ocloud_view.subscription_one( subscription_id1, uow) assert subscription_res is None - session.return_value.execute.return_value.first.return_value = { - "subscriptionId": subscription_id1, - } + session.return_value.query.return_value.filter_by.return_value.first.\ + return_value.serialize.return_value = { + "subscriptionId": subscription_id1, + } subscription_res = ocloud_view.subscription_one( subscription_id1, uow) @@ -289,7 +354,7 @@ def test_view_subscription_one(mock_uow): def test_flask_get_list(mock_flask_uow): session, app = mock_flask_uow - session.return_value.execute.return_value = [] + session.query.return_value = [] apibase = config.get_o2ims_api_base() with app.test_client() as client: @@ -315,7 +380,9 @@ def test_flask_get_list(mock_flask_uow): def test_flask_get_one(mock_flask_uow): session, app = mock_flask_uow - session.return_value.execute.return_value.first.return_value = None + + session.return_value.query.return_value.filter_by.return_value.\ + first.return_value = None apibase = config.get_o2ims_api_base() with app.test_client() as client: