diff options
Diffstat (limited to 'catalog/packages/tests/test_vnf_pkg_subscription.py')
| -rw-r--r-- | catalog/packages/tests/test_vnf_pkg_subscription.py | 63 |
1 files changed, 62 insertions, 1 deletions
diff --git a/catalog/packages/tests/test_vnf_pkg_subscription.py b/catalog/packages/tests/test_vnf_pkg_subscription.py index 635b137..f2fe150 100644 --- a/catalog/packages/tests/test_vnf_pkg_subscription.py +++ b/catalog/packages/tests/test_vnf_pkg_subscription.py @@ -18,8 +18,12 @@ import mock from rest_framework.test import APIClient from django.test import TestCase -from catalog.pub.database.models import VnfPkgSubscriptionModel +from catalog.pub.database.models import VnfPkgSubscriptionModel, VnfPackageModel from .const import vnf_subscription_data +from catalog.packages.biz.notificationsutil import NotificationsUtil, prepare_vnfpkg_notification +from catalog.packages import const +from catalog.pub.config import config as pub_config +import catalog.pub.utils.timeutil class TestNfPackageSubscription(TestCase): @@ -181,3 +185,60 @@ 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) + + +class NotificationTest(TestCase): + def setUp(self): + 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() + + 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_notify(self, mock_nowtime, mock_uuid, mock_requests_post): + mock_nowtime.return_value = "nowtime()" + mock_uuid.return_value = "1111" + notification_content = prepare_vnfpkg_notification("vnfpkgid1", const.PKG_NOTIFICATION_TYPE.CHANGE, + const.PKG_CHANGE_TYPE.OP_STATE_CHANGE, operational_state=None) + filters = { + 'vnfdId': 'vnfd_id', + 'vnfPkgId': 'vnf_pkg_id' + } + NotificationsUtil().send_notification(notification_content, filters, True) + expect_callbackuri = "http://127.0.0.1/self" + expect_notification = { + 'id': "1111", + 'notificationType': const.PKG_NOTIFICATION_TYPE.CHANGE, + 'timeStamp': "nowtime()", + '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'}) |