def test_get_url(self):
# Leave Oracle SID as it is
config = {
'username': "******",
'password': "******",
'host': "host",
'port': 1234,
'database': 'SID'
}
store = OracleDatastore(config)
self.assertEqual("oracle+cx_oracle://username:password@host:1234/SID",
str(store.connection_config['url']))
# Handle Oracle service name
config = {
'username': "******",
'password': "******",
'host': "host",
'port': "1234",
'database': 'x.y.z'
}
store = OracleDatastore(config)
self.assertEqual(
"oracle+cx_oracle://username:password@host:1234/?service_name=x.y.z",
str(store.connection_config['url']))
def test_get_dsn(self):
args = {'host': 'localhost', 'port': '1234', 'database': 'db'}
address_list = [
DSN_FAILOVER_ADDRESS_TEMPLATE.format(host='localhost',
port='1234'),
]
exp_res = '(DESCRIPTION=(FAILOVER=off)(LOAD_BALANCE=off)(CONNECT_TIMEOUT=3)(RETRY_COUNT=3)' \
f'(ADDRESS_LIST={address_list[0]})' \
'(CONNECT_DATA=(SERVICE_NAME=db)))'
self.assertEqual(OracleDatastore._get_dsn(**args), exp_res)
args['host'] = 'localhost1,localhost2'
address_list = [
DSN_FAILOVER_ADDRESS_TEMPLATE.format(host='localhost1',
port='1234'),
DSN_FAILOVER_ADDRESS_TEMPLATE.format(host='localhost2',
port='1234')
]
exp_res = '(DESCRIPTION=(FAILOVER=on)(LOAD_BALANCE=off)(CONNECT_TIMEOUT=3)(RETRY_COUNT=3)' \
f'(ADDRESS_LIST={"".join(address_list)})' \
'(CONNECT_DATA=(SERVICE_NAME=db)))'
self.assertEqual(exp_res, OracleDatastore._get_dsn(**args))
args['host'] = 'localhost'
args['port'] = '1234,1235'
address_list = [
DSN_FAILOVER_ADDRESS_TEMPLATE.format(host='localhost',
port='1234'),
DSN_FAILOVER_ADDRESS_TEMPLATE.format(host='localhost', port='1235')
]
exp_res = '(DESCRIPTION=(FAILOVER=on)(LOAD_BALANCE=off)(CONNECT_TIMEOUT=3)(RETRY_COUNT=3)' \
f'(ADDRESS_LIST={"".join(address_list)})' \
'(CONNECT_DATA=(SERVICE_NAME=db)))'
self.assertEqual(exp_res, OracleDatastore._get_dsn(**args))
def test_output_type_handler(self):
cursor = type('Cursor', (object, ), {
'arraysize': 20,
'var': lambda x, arraysize: x + '_' + str(arraysize)
})
store = OracleDatastore({}, {})
self.assertEqual(
'LONG_STRING_20',
store._output_type_handler(cursor, 'name', 'CLOB', 0, 0, 0))
self.assertIsNone(
store._output_type_handler(cursor, 'name', 'SOME_OTHER_TYPE', 0, 0,
0))
def test_makedict(self):
cursor = type('MockCursor', (object, ),
{'description': ['a', 'b', 'c']})
args = [1, 2, 3, 4]
store = OracleDatastore({}, {})
createrowfunc = store._makedict(cursor)
self.assertEqual({
'a': 1,
'b': 2,
'c': 3,
}, createrowfunc(*args))
def test_disconnect(self):
store = OracleDatastore({})
connection = MagicMock()
store.connection = connection
store.disconnect()
connection.close.assert_called_once()
self.assertIsNone(store.connection)
store.disconnect()
def test_not_implemented_methods(self):
methods = [
('write_rows', ['the table', []]),
('execute', ['the query']),
('list_tables_for_schema', ['the schema']),
('rename_schema', ['old', 'new']),
]
store = OracleDatastore({}, {})
for method, args in methods:
with self.assertRaises(NotImplementedError):
getattr(store, method)(*args)
def test_query(self):
query = "SELECT this FROM that WHERE this=that"
store = OracleDatastore({}, {})
store.connection = MockConnection([{"id": 1}, {"id": 2}])
store._output_type_handler = MagicMock()
store._makedict = MagicMock()
list(store.query(query))
self.assertEqual(query, store.connection.cursor_obj.executed)
self.assertEqual(store._output_type_handler,
store.connection.outputtypehandler)
self.assertEqual(store._makedict.return_value,
store.connection.cursor_obj.rowfactory)
def test_connect(self, mock_cx_oracle):
config = {
'username': '******',
'database': 'db',
'password': '******',
'port': 9999,
'host': 'localhost',
'name': 'configname',
}
mock_cx_oracle.Connection.return_value = {"connected": True}
store = OracleDatastore(config)
store.connect()
self.assertEqual("(user@db)", store.user)
self.assertEqual({"connected": True}, store.connection)
address_list = DSN_FAILOVER_ADDRESS_TEMPLATE.format(
host=config['host'], port=config['port'])
dsn = '(DESCRIPTION=(FAILOVER=off)(LOAD_BALANCE=off)(CONNECT_TIMEOUT=3)(RETRY_COUNT=3)' \
f'(ADDRESS_LIST={"".join(address_list)})' \
'(CONNECT_DATA=(SERVICE_NAME=db)))'
mock_cx_oracle.Connection.assert_called_with(user='******',
password='******',
dsn=dsn)
config = {
'username': '******',
'database': 'db',
'port': 9999,
'host': 'localhost',
'name': 'configname',
}
store = OracleDatastore(config)
with self.assertRaises(GOBException):
store.connect()