Add UT for vespublish

Change-Id: Ic429214cc3eba9779b0f48ca76cdd1ced030a983
Issue-ID: MULTICLOUD-358
Signed-off-by: Xiaohua Zhang <xiaohua.zhang@windriver.com>

diff --git a/windriver/titanium_cloud/vesagent/tests/__init__.py b/windriver/titanium_cloud/vesagent/tests/__init__.py
new file mode 100644 (file)
index 0000000..e4fe7a0
--- /dev/null
+++ b/windriver/titanium_cloud/vesagent/tests/__init__.py
@@ -0,0 +1,16 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# Copyright (c) 2017-2018 Wind River Systems, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file 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.
+

diff --git a/windriver/titanium_cloud/vesagent/event_domain/tests_fault_vm.py b/windriver/titanium_cloud/vesagent/tests/tests_fault_vm.py
similarity index 100%
rename from windriver/titanium_cloud/vesagent/event_domain/tests_fault_vm.py
rename to windriver/titanium_cloud/vesagent/tests/tests_fault_vm.py

diff --git a/windriver/titanium_cloud/vesagent/tests_tasks.py b/windriver/titanium_cloud/vesagent/tests/tests_tasks.py
similarity index 100%
rename from windriver/titanium_cloud/vesagent/tests_tasks.py
rename to windriver/titanium_cloud/vesagent/tests/tests_tasks.py

diff --git a/windriver/titanium_cloud/vesagent/tests.py b/windriver/titanium_cloud/vesagent/tests/tests_vesagent_ctrl.py
similarity index 100%
rename from windriver/titanium_cloud/vesagent/tests.py
rename to windriver/titanium_cloud/vesagent/tests/tests_vesagent_ctrl.py

diff --git a/windriver/titanium_cloud/vesagent/tests/tests_vespublish.py b/windriver/titanium_cloud/vesagent/tests/tests_vespublish.py
new file mode 100644 (file)
index 0000000..0178aae
--- /dev/null
+++ b/windriver/titanium_cloud/vesagent/tests/tests_vespublish.py
@@ -0,0 +1,54 @@
+# Copyright (c) 2017-2018 Wind River Systems, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file 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.
+
+import mock
+
+import unittest
+import json
+import urllib2
+
+from titanium_cloud.vesagent import vespublish
+
+MOCK_VESENDPOINT = {
+    "endpoint" : "MOCKED_VES_COLLECTOR_EP1",
+    "username" : "MOCKED_VES_COLLECTOR_USER1",
+    "password" : "MOCKED_VES_COLLECTOR_PASSWD1",
+}
+MOCK_VESPUBLISH_EVENT1 = [{"name":"event1"}]
+
+class VespublishTest(unittest.TestCase):
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+
+    @mock.patch.object(urllib2, 'urlopen')
+    @mock.patch.object(urllib2, 'Request')
+    def test_publishAnyEventToVES(self, mock_Request, mock_urlopen):
+        mock_request = mock.Mock()
+
+        mock_Request.side_effect= [
+            mock_request
+                ]
+
+        mock_response = mock.Mock(["read"])
+        mock_response.read.return_value = "MOCKED_VESPUBLISH_RESPONSE_MESSAGE"
+        mock_urlopen.side_effect= [
+            mock_response
+                ]
+
+        vespublish.publishAnyEventToVES(MOCK_VESENDPOINT, MOCK_VESPUBLISH_EVENT1)
+
+        pass