summaryrefslogtreecommitdiffstats
path: root/dmaap/tests/test_consulif.py
blob: 2d26576438c9b5f0d3875a227948a5c3f86d6603 (plain)
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))