1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
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))
|