[20190329]: unit test code for dmaap plugin

add unit test code for dmaap plugin. Conaul call in consul
interface module is replace by mock function.

Issue-ID: CCSDK-794
Change-Id: I01f8bbe4f3aba8f6b3bc1a17d2aa8a63781a575f
Signed-off-by: xufeiliu <xl085b@att.com>

1 files changed, 120 insertions, 0 deletions

diff --git a/dmaap/tests/test_consulif.py b/dmaap/tests/test_consulif.py
new file mode 100644
index 0000000..2d26576
--- /dev/null
+++ b/dmaap/tests/test_consulif.py
@@ -0,0 +1,120 @@
+# ============LICENSE_START=======================================================
+# org.onap.dcae
+# ================================================================================
+# Copyright (c) 2017-2018 AT&T Intellectual Property. All rights reserved.
+# ================================================================================
+# 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.
+# ============LICENSE_END=========================================================
+#
+# ECOMP is a trademark and service mark of AT&T Intellectual Property.
+
+
+import pytest
+from cloudify.exceptions import NonRecoverableError
+import os
+from consulif.consulif import ConsulHandle
+
+
+#When run unit test, please update the consul interface parameters based on local consul configuration. 
+CONSUL_HOST = "consul"                      # Should always be a local consul agent on Cloudify Manager
+#CONSUL_PORT = '8510'
+CONSUL_PORT = '8500'
+DBCL_KEY_NAME = "dmaap_dbcl_info"           # Consul key containing DMaaP data bus credentials
+DBC_SERVICE_NAME= "dmaap_bus_controller"    # Name under which the DMaaP bus controller is registered
+
+
+def _fake_get_config(self, key):
+    fix ={'dmaap': {'username': 'm06814@controller.dcae.ecomp.att.com',
+                'url': 'https://dcae-dbcl.d54852.dyh1b.tci.att.com:8443/webapi',
+                'password' : 'Dca3wr3x',
+                'owner': 'dcaeorch' }}
+    return fix
+
+def _fake_service(self, key):
+    service_address = "myAddress"
+    service_port= "8443"
+    return service_address, service_port
+
+def test_get_config_service(monkeypatch):
+  monkeypatch.setattr(ConsulHandle, 'get_config', _fake_get_config)
+  monkeypatch.setattr(ConsulHandle, 'get_service', _fake_service)
+  try:
+    err_msg = "Error getting ConsulHandle when configuring dmaap plugin: {0}"
+    _ch = ConsulHandle("http://{0}:{1}".format(CONSUL_HOST, CONSUL_PORT), None, None, None)
+    assert None != _ch
+    
+    err_msg = "Error getting config for '{0}' from ConsulHandle when configuring dmaap plugin: ".format(DBCL_KEY_NAME) + "{0}"
+    config = _ch.get_config(DBCL_KEY_NAME)
+
+    err_msg = "Error setting DMAAP_USER while configuring dmaap plugin: {0}"
+    DMAAP_USER = config['dmaap']['username']
+
+    err_msg = "Error setting DMAAP_PASS while configuring dmaap plugin: {0}"
+    DMAAP_PASS = config['dmaap']['password']
+
+    err_msg = "Error setting DMAAP_OWNER while configuring dmaap plugin: {0}"
+    DMAAP_OWNER = config['dmaap']['owner']
+
+    err_msg = "Error setting DMAAP_PROTOCOL while configuring dmaap plugin: {0}"
+    if 'protocol' in config['dmaap']:
+        DMAAP_PROTOCOL = config['dmaap']['protocol']
+    else:
+        DMAAP_PROTOCOL = 'https'    # Default to https (service discovery should give us this but doesn't
+
+    err_msg = "Error setting DMAAP_PATH while configuring dmaap plugin: {0}"
+    if 'path' in config['dmaap']:
+        DMAAP_PATH = config['dmaap']['path']
+    else:
+        DMAAP_PATH = 'webapi'       # SHould come from service discovery but Consul doesn't support it
+
+    err_msg = "Error getting service_address and service_port for '{0}' from ConsulHandle when configuring dmaap plugin: ".format(DBC_SERVICE_NAME) + "{0}"
+    service_address, service_port = _ch.get_service(DBC_SERVICE_NAME)
+
+    err_msg = "Error setting DMAAP_API_URL while configuring dmaap plugin: {0}"
+    DMAAP_API_URL = '{0}://{1}:{2}/{3}'.format(DMAAP_PROTOCOL, service_address, service_port, DMAAP_PATH)
+    assert DMAAP_API_URL != None
+    dmaap_config = {'DMAAP_USER':DMAAP_USER, 'DMAAP_API_URL':DMAAP_API_URL, 'DMAAP_PASS':DMAAP_PASS, 'DMAAP_OWNER':DMAAP_OWNER}
+    print "return dmaap config info from consul: {0}".format(dmaap_config)
+    return dmaap_config
+
+  except Exception as e:
+    raise NonRecoverableError(err_msg.format(e))
+
+def _fake_add_to_entry(self, key, add_name, add_value):
+    return True
+
+def _fake_delete_entry(self, entry_name):
+    return True
+
+def test_add_entry(monkeypatch):
+  monkeypatch.setattr(ConsulHandle, 'get_config', _fake_get_config)
+  monkeypatch.setattr(ConsulHandle, 'add_to_entry', _fake_add_to_entry)
+  monkeypatch.setattr(ConsulHandle, 'delete_entry', _fake_delete_entry)
+  try:
+    _ch = ConsulHandle("http://{0}:{1}".format(CONSUL_HOST, CONSUL_PORT), None, None, None)
+    assert None != _ch
+
+    key = 'DMAAP_TEST'
+    name = 'dmaap_test_name'
+    value = 'dmaap_test_value'
+    _ch.add_to_entry(key, name, value)
+
+    name = "dmaap_test_name_2"
+    value = 'dmaap_test_value_2'
+    _ch.add_to_entry(key, name, value)
+
+    _ch.delete_entry(key)
+
+  except Exception as e:
+    raise NonRecoverableError("Error in test_add_entry: {0}".format(e))
+