X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=catalog%2Fpackages%2Ftests%2Ftest_vnf_pkg_subscription.py;h=a8fd5d55b3a5d88c3c9b991a4b77540e53be6d3d;hb=f173f592ed958f55d992c109553760e3fb12628a;hp=635b1372f74abd2baa88f31a7d12ef81aefc5181;hpb=a32c2b20207885d895bd96204cc166fca14db97b;p=modeling%2Fetsicatalog.git diff --git a/catalog/packages/tests/test_vnf_pkg_subscription.py b/catalog/packages/tests/test_vnf_pkg_subscription.py index 635b137..a8fd5d5 100644 --- a/catalog/packages/tests/test_vnf_pkg_subscription.py +++ b/catalog/packages/tests/test_vnf_pkg_subscription.py @@ -12,14 +12,26 @@ # See the License for the specific language governing permissions and # limitations under the License. +import json +import os import uuid -import mock -from rest_framework.test import APIClient +import mock from django.test import TestCase +from rest_framework import status +from rest_framework.test import APIClient +from requests.auth import HTTPBasicAuth -from catalog.pub.database.models import VnfPkgSubscriptionModel -from .const import vnf_subscription_data +import catalog.pub.utils.timeutil +from catalog.packages import const +from catalog.packages.biz.notificationsutil import PkgNotifications +from catalog.packages.biz.vnf_pkg_subscription import QuerySubscription, TerminateSubscription +from catalog.pub.config import config as pub_config +from catalog.pub.config.config import CATALOG_ROOT_PATH +from catalog.pub.database.models import VnfPkgSubscriptionModel, VnfPackageModel +from catalog.pub.exceptions import SubscriptionDoesNotExistsException +from catalog.pub.utils import toscaparser +from .const import vnf_subscription_data, vnfd_data class TestNfPackageSubscription(TestCase): @@ -181,3 +193,131 @@ class TestNfPackageSubscription(TestCase): dummy_uuid = str(uuid.uuid4()) response = self.client.delete("/api/vnfpkgm/v1/subscriptions/%s" % dummy_uuid) self.assertEqual(404, response.status_code) + + @mock.patch("requests.get") + @mock.patch.object(toscaparser, 'parse_vnfd') + @mock.patch("requests.post") + @mock.patch("uuid.uuid4") + @mock.patch.object(catalog.pub.utils.timeutil, "now_time") + def test_vnfpkg_subscript_notify(self, mock_nowtime, mock_uuid, mock_requests_post, mock_parse_vnfd, + mock_requests_get): + mock_nowtime.return_value = "2019-02-16 14:41:16" + uuid_subscriptid = "99442b18-a5c7-11e8-998c-bf1755941f13" + uuid_vnfPackageId = "3fa85f64-5717-4562-b3fc-2c963f66afa6" + uuid_vnfdid = "00342b18-a5c7-11e8-998c-bf1755941f12" + mock_uuid.side_effect = [uuid_subscriptid, "1111"] + mock_requests_get.return_value.status_code = 204 + mock_parse_vnfd.return_value = json.JSONEncoder().encode(vnfd_data) + + response = self.client.post( + "/api/vnfpkgm/v1/subscriptions", + data=vnf_subscription_data, + format='json') + self.assertEqual(201, response.status_code) + + data = {'file': open(os.path.join(CATALOG_ROOT_PATH, "empty.txt"), "rt")} + VnfPackageModel.objects.create( + vnfPackageId=uuid_vnfPackageId, + onboardingState="CREATED" + ) + + response = self.client.put("/api/vnfpkgm/v1/vnf_packages/%s/package_content" % uuid_vnfPackageId, data=data) + vnf_pkg = VnfPackageModel.objects.filter(vnfPackageId=uuid_vnfPackageId) + self.assertEqual(uuid_vnfdid, vnf_pkg[0].vnfdId) + self.assertEqual(const.PKG_STATUS.ONBOARDED, vnf_pkg[0].onboardingState) + self.assertEqual(response.status_code, status.HTTP_202_ACCEPTED) + + expect_notification = { + 'id': "1111", + 'notificationType': const.PKG_NOTIFICATION_TYPE.ONBOARDING, + 'timeStamp': "2019-02-16 14:41:16", + 'vnfPkgId': uuid_vnfPackageId, + 'vnfdId': uuid_vnfdid, + "subscriptionId": uuid_subscriptid, + '_links': { + 'subscription': { + 'href': 'http://%s:%s/%s%s' % (pub_config.MSB_SERVICE_IP, + pub_config.MSB_SERVICE_PORT, + const.VNFPKG_SUBSCRIPTION_ROOT_URI, + uuid_subscriptid)}, + 'vnfPackage': { + 'href': 'http://%s:%s/%s/vnf_packages/%s' % (pub_config.MSB_SERVICE_IP, + pub_config.MSB_SERVICE_PORT, + const.PKG_URL_PREFIX, + uuid_vnfPackageId) + } + } + } + mock_requests_post.assert_called_with(vnf_subscription_data["callbackUri"], data=expect_notification, + headers={'Connection': 'close'}, auth=HTTPBasicAuth("admin", "pwd1234")) + + def test_service_query_single_subscription_not_found(self): + try: + subscription_id = "test_not_found" + QuerySubscription().query_single_subscription(subscription_id) + except SubscriptionDoesNotExistsException as e: + self.assertEqual("Subscription with ID: %s does not exist" % subscription_id, e.args[0]) + + def test_service_delete_single_subscription_not_found(self): + try: + subscription_id = "test_not_found" + TerminateSubscription().terminate(subscription_id) + except SubscriptionDoesNotExistsException as e: + self.assertEqual("Subscription with ID: %s does not exist" % subscription_id, e.args[0]) + + +class NotificationTest(TestCase): + def setUp(self): + VnfPackageModel.objects.all().delete() + VnfPkgSubscriptionModel.objects.all().delete() + + def tearDown(self): + VnfPackageModel.objects.all().delete() + VnfPkgSubscriptionModel.objects.all().delete() + + @mock.patch("requests.post") + @mock.patch("uuid.uuid4") + @mock.patch.object(catalog.pub.utils.timeutil, "now_time") + def test_vnfpkg_manual_notify(self, mock_nowtime, mock_uuid, mock_requests_post): + VnfPackageModel(vnfPackageId="vnfpkgid1", + vnfdId="vnfdid1" + ).save() + + VnfPkgSubscriptionModel(subscription_id="1", + callback_uri="http://127.0.0.1/self", + notification_types=const.NOTIFICATION_TYPES, + vnfd_id="vnfdid1", + vnf_pkg_id="vnfpkgid1" + ).save() + mock_nowtime.return_value = "2019-12-16 14:41:16" + mock_uuid.return_value = "1111" + notify = PkgNotifications(const.PKG_NOTIFICATION_TYPE.CHANGE, "vnfpkgid1", + const.PKG_CHANGE_TYPE.OP_STATE_CHANGE, operational_state=None) + + notify.send_notification() + expect_callbackuri = "http://127.0.0.1/self" + expect_notification = { + 'id': "1111", + 'notificationType': const.PKG_NOTIFICATION_TYPE.CHANGE, + 'timeStamp': "2019-12-16 14:41:16", + 'vnfPkgId': "vnfpkgid1", + 'vnfdId': "vnfdid1", + 'changeType': const.PKG_CHANGE_TYPE.OP_STATE_CHANGE, + 'operationalState': None, + "subscriptionId": "1", + '_links': { + 'subscription': { + 'href': 'http://%s:%s/%s%s' % (pub_config.MSB_SERVICE_IP, + pub_config.MSB_SERVICE_PORT, + const.VNFPKG_SUBSCRIPTION_ROOT_URI, + "1")}, + 'vnfPackage': { + 'href': 'http://%s:%s/%s/vnf_packages/%s' % (pub_config.MSB_SERVICE_IP, + pub_config.MSB_SERVICE_PORT, + const.PKG_URL_PREFIX, + "vnfpkgid1") + } + } + } + mock_requests_post.assert_called_with(expect_callbackuri, data=expect_notification, + headers={'Connection': 'close'})