initial seed code commit VVP-5

Change-Id: I6560c87ef48a6d0d1fe8197c7c6439c7e6ad653f
Signed-off-by: Paul McGoldrick <paul.mcgoldrick@att.com>

1 files changed, 128 insertions, 0 deletions

diff --git a/django/engagementmanager/tests/test_digest_email_notifications.py b/django/engagementmanager/tests/test_digest_email_notifications.py
new file mode 100755
index 0000000..b3e75df
--- /dev/null
+++ b/django/engagementmanager/tests/test_digest_email_notifications.py
@@ -0,0 +1,128 @@
+#  
+# ============LICENSE_START========================================== 
+# org.onap.vvp/engagementmgr
+# ===================================================================
+# Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+# ===================================================================
+#
+# Unless otherwise specified, all software contained herein is licensed
+# under the Apache License, Version 2.0 (the “License”);
+# you may not use this software except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#             http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+#
+#
+# Unless otherwise specified, all documentation contained herein is licensed
+# under the Creative Commons License, Attribution 4.0 Intl. (the “License”);
+# you may not use this documentation except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#             https://creativecommons.org/licenses/by/4.0/
+#
+# Unless required by applicable law or agreed to in writing, documentation
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+# ============LICENSE_END============================================
+#
+# ECOMP is a trademark and service mark of AT&T Intellectual Property.
+import json
+import mock
+from rest_framework.status import HTTP_202_ACCEPTED
+from engagementmanager.bus.messages.activity_event_message import ActivityEventMessage
+from engagementmanager.bus.messages.daily_scheduled_message import DailyScheduledMessage
+from engagementmanager.models import Vendor
+from engagementmanager.utils.activities_data import UserJoinedEngagementActivityData
+from engagementmanager.tests.test_base_entity import TestBaseEntity
+from engagementmanager.utils.constants import Constants, EngagementStage
+from engagementmanager.apps import bus_service
+from engagementmanager.service.logging_service import LoggingServiceFactory
+
+logger = LoggingServiceFactory.get_logger()
+
+
+def digest_mock(self, bus_message):
+    DigestEmailNotificationsTestCase.is_digested_mock_sent = True
+    DigestEmailNotificationsTestCase.message = bus_message
+
+
+# engagementmanager/bus/handlers/digest_email_notification_handler.py
+@mock.patch('engagementmanager.bus.handlers.digest_email_notification_handler.'
+            'DigestEmailNotificationHandler.handle_message', digest_mock)
+class DigestEmailNotificationsTestCase(TestBaseEntity):
+    is_digested_mock_sent = False
+    message = None
+
+    def childSetup(self):
+        self.createVendors([Constants.service_provider_company_name, 'Other'])
+
+        self.createDefaultRoles()
+        vendor = Vendor.objects.get(name=Constants.service_provider_company_name)
+        self.el_user = self.creator.createUser(vendor, self.randomGenerator(
+            "main-vendor-email"), self.randomGenerator("randomNumber"), self.randomGenerator("randomString"), self.el, True)
+        vendor = Vendor.objects.get(name='Other')
+        self.user = self.creator.createUser(vendor, self.randomGenerator("email"), self.randomGenerator(
+            "randomNumber"), self.randomGenerator("randomString"), self.standard_user, True)
+        self.pruser = self.creator.createUser(vendor, self.randomGenerator("email"), self.randomGenerator(
+            "randomNumber"), self.randomGenerator("randomString"), self.standard_user, True)
+
+        self.engagement = self.creator.createEngagement(self.randomGenerator(
+            "randomString"), self.randomGenerator("randomString"), None)
+        self.engagement.engagement_team.add(self.user, self.el_user)
+        self.engagement.reviewer = self.el_user
+        self.engagement.peer_reviewer = self.pruser
+        self.engagement.save()
+
+        self.deploymentTarget = self.creator.createDeploymentTarget(
+            self.randomGenerator("randomString"), self.randomGenerator("randomString"))
+        self.vf = self.creator.createVF(self.randomGenerator("randomString"),
+                                        self.engagement, self.deploymentTarget, False, vendor)
+
+    def testDigestEmailForActivities(self):
+        """
+        Will check if the service bus deliver the message of sending digested mails
+        No need to check if the mail is sent or if the python scheduling is working
+        """
+        # Create the activities:
+        self.urlStr = self.urlPrefix + "single-engagement/" + str(self.engagement.uuid) + "/stage/@stage"
+
+        random_user = self.creator.createUser(Vendor.objects.get(name='Other'),
+                                              self.randomGenerator("email"),
+                                              self.randomGenerator("randomNumber"),
+                                              self.randomGenerator("randomString"),
+                                              self.el, True)
+
+        self.engagement.engagement_team.add(random_user)
+        self.engagement.save()
+
+        users_list = []
+        users_list.append(random_user)
+        activity_data = UserJoinedEngagementActivityData(self.vf, users_list, self.engagement)
+        bus_service.send_message(ActivityEventMessage(activity_data))
+
+        token = self.loginAndCreateSessionToken(random_user)
+        response = self.c.put(self.urlStr.replace("@stage", EngagementStage.Active.name),
+                              json.dumps(dict(), ensure_ascii=False), content_type='application/json',
+                              **{'HTTP_AUTHORIZATION': "token " + token})
+        self.assertEqual(response.status_code, HTTP_202_ACCEPTED)
+
+        urlStr = self.urlPrefix + 'engagement/${uuid}/activities/'
+        response = self.c.get(urlStr.replace('${uuid}', str(self.engagement.uuid)),
+                              **{'HTTP_AUTHORIZATION': "token " + token})
+        content = json.loads(response.content)
+        self.assertEqual(len(content), 2)
+
+        message = DailyScheduledMessage()
+        bus_service.send_message(message)
+        self.assertTrue(self.is_digested_mock_sent)
+        self.assertEqual(self.message, message)