def __init__(self, engine):
"""
Constructor
:param engine: DxMaskingEngine object
"""
DatabaseConnector.__init__(self)
self.__is_database = None
self.__engine = engine
self.__logger = logging.getLogger()
self.__logger.debug("creating DxConnector object")
def update(self):
"""
Update connector on engine
Return None if OK
"""
api_instance = DatabaseConnectorApi(self.__engine.api_client)
body = DatabaseConnector()
for k in self.attribute_map.keys():
if getattr(self, k) is not None:
setattr(body, k, getattr(self, k))
try:
self.__logger.debug("update connector input %s" % str(self))
response = api_instance.update_database_connector(
self.database_connector_id,
body,
_request_timeout=self.__engine.get_timeout())
self.__logger.debug("update connector response %s" % str(response))
self.database_connector_id = response.database_connector_id
print_message("Connector %s updated" % self.connector_name)
except ApiException as e:
print_error(e.body)
self.__logger.error(e)
return 1
def test_connector_add(self, get_session):
dbresposne = DatabaseConnector(database_connector_id=1,
connector_name="DB connector",
environment_id=1,
database_type="ORACLE")
with mock.patch.object(DatabaseConnectorApi,
'create_database_connector',
return_value=dbresposne) as mock_method:
params = {
'envname': 'Env1',
'schemaName': 'SCOTT',
'host': '10.10.10.10',
'port': 1521,
'password': '******',
'username': '******',
'connname': 'DB new',
'sid': 'ORCL',
'type': 'oracle',
'jdbc': None,
'instancename': None,
'databasename': None
}
ret = connector_add(None, params)
name, args, kwargs = mock_method.mock_calls[0]
print args[0]
self.assertEqual(1, args[0].environment_id)
self.assertEqual(1521, args[0].port)
self.assertEqual(0, ret)
def dbconnector_load(a, **kwargs):
"""
Create an output for get_all_database_connectors call
"""
pi = PageInfo(number_on_page=2, total=2)
dbconnector = [DatabaseConnector(database_connector_id=1, connector_name="DB connector", environment_id=1, database_type="ORACLE")]
dbcpo = DatabaseConnectorList(page_info=pi, response_list=dbconnector)
return dbcpo