import logging
import uuid
import requests
+import json
from rest_framework import status
+from requests.auth import HTTPBasicAuth
from catalog.packages import const
from catalog.pub.database.models import VnfPackageModel, VnfPkgSubscriptionModel, NsdmSubscriptionModel
import catalog.pub.utils.timeutil
serialized_data = self.notifyserializer(data=notification)
if not serialized_data.is_valid():
logger.error('Notification Data is invalid:%s.' % serialized_data.errors)
- self.post_notification(callbackuri, notification)
-
- def post_notification(self, callbackuri, notification):
- """
- params = auth_info.get("paramsBasic", {})
- username, password = params.get("userName"), params.get("password")
- logger.info("Sending notification to %s, %s", callbackuri, params)
- resp = None
- if username:
- resp = requests.post(callbackuri,
- data=notification,
- auth=HTTPBasicAuth(username, password))
- else:
- """
+ if sub.auth_info:
+ self.post_notification(callbackuri, notification, auth_info=json.loads(sub.auth_info))
+ else:
+ self.post_notification(callbackuri, notification)
+
+ def post_notification(self, callbackuri, notification, auth_info=None):
try:
- resp = requests.post(callbackuri, data=notification, headers={'Connection': 'close'})
+ if auth_info:
+ if const.BASIC in auth_info.get("authType", ''):
+ params = auth_info.get("paramsBasic", {})
+ username = params.get("userName")
+ password = params.get("password")
+ resp = requests.post(callbackuri, data=notification, headers={'Connection': 'close'},
+ auth=HTTPBasicAuth(username, password))
+ elif const.OAUTH2_CLIENT_CREDENTIALS in auth_info.get("authType", ''):
+ # todo
+ pass
+ else:
+ # todo
+ pass
+ else:
+ resp = requests.post(callbackuri, data=notification, headers={'Connection': 'close'})
if resp.status_code != status.HTTP_204_NO_CONTENT:
logger.error("Sending notification to %s failed: %s" % (callbackuri, resp.text))
else:
subscription_save_db = {
"subscriptionid": self.subscription_id,
"callback_uri": self.callback_uri,
- "auth_info": self.authentication,
+ "auth_info": json.dumps(self.authentication),
"links": json.dumps(links)
}
for filter_type in const.NSDM_NOTIFICATION_FILTERS:
from collections import Counter
from rest_framework import status
-
+from requests.auth import HTTPBasicAuth
from catalog.packages import const
from catalog.pub.database.models import VnfPkgSubscriptionModel
from catalog.pub.exceptions import VnfPkgSubscriptionException, \
logger.debug("SubscribeNotification-post::> Sending GET request "
"to %s" % self.callback_uri)
try:
- response = requests.get(self.callback_uri, timeout=2)
+ if self.authentication:
+ if const.BASIC in self.authentication.get("authType", ''):
+ params = self.authentication.get("paramsBasic", {})
+ username = params.get("userName")
+ password = params.get("password")
+ response = requests.get(self.callback_uri, auth=HTTPBasicAuth(username, password), timeout=2)
+ elif const.OAUTH2_CLIENT_CREDENTIALS in self.authentication.get("authType", ''):
+ # todo
+ pass
+ else:
+ # todo
+ pass
+ else:
+ response = requests.get(self.callback_uri, timeout=2)
if response.status_code != status.HTTP_204_NO_CONTENT:
raise VnfPkgSubscriptionException(
"callbackUri %s returns %s status code." % (
"BASIC"
],
"paramsBasic": {
- "userName": "string",
- "password": "string"
+ "userName": "admin",
+ "password": "pwd1234"
}
}
}
from django.test import TestCase
from rest_framework.test import APIClient
from rest_framework import status
+from requests.auth import HTTPBasicAuth
from catalog.packages.biz.nsdm_subscription import NsdmSubscription
from catalog.pub.database.models import NsdmSubscriptionModel
}
}
}
- mock_requests_post.assert_called_with(expect_callbackuri, data=expect_notification, headers={'Connection': 'close'})
+ mock_requests_post.assert_called_with(expect_callbackuri, data=expect_notification,
+ auth=HTTPBasicAuth("username", "password"),
+ headers={'Connection': 'close'})
class NotificationTest(TestCase):
from django.test import TestCase
from rest_framework import status
from rest_framework.test import APIClient
+from requests.auth import HTTPBasicAuth
import catalog.pub.utils.timeutil
from catalog.packages import const
}
}
mock_requests_post.assert_called_with(vnf_subscription_data["callbackUri"], data=expect_notification,
- headers={'Connection': 'close'})
+ headers={'Connection': 'close'}, auth=HTTPBasicAuth("admin", "pwd1234"))
def test_service_query_single_subscription_not_found(self):
try: