def test_basic_hbase_usage(self):
local_config = DotDict({
'hbase_host': 'host',
'database_name': 'name',
'hbase_port': 9090,
'hbase_timeout': 9000,
'number_of_retries': 2,
'logger': SilentFakeLogger(),
'executor_identity': lambda: 'dwight' # bogus thread id
})
a_fake_hbase_connection = FakeHB_Connection(local_config)
with mock.patch.object(
connection_context, 'HBaseConnection',
mock.Mock(return_value=a_fake_hbase_connection)):
hb_context = connection_context.HBaseConnectionContext(
local_config)
# open a connection
with hb_context() as conn:
pass
eq_(a_fake_hbase_connection.close_counter, 1)
# open another connection again
with hb_context() as conn:
pass
eq_(a_fake_hbase_connection.close_counter, 2)
# get a named connection
with hb_context('fred') as conn:
pass
eq_(a_fake_hbase_connection.close_counter, 3)
# close all connections
hb_context.close()
eq_(a_fake_hbase_connection.close_counter, 3)
def test_hbase_usage_with_transaction(self):
local_config = DotDict({
'hbase_host': 'host',
'database_name': 'name',
'hbase_port': 9090,
'hbase_timeout': 9000,
'number_of_retries': 2,
'logger': SilentFakeLogger(),
'executor_identity': lambda: 'dwight' # bogus thread id
})
a_fake_hbase_connection = FakeHB_Connection(local_config)
with mock.patch.object(
connection_context, 'HBaseConnection',
mock.Mock(return_value=a_fake_hbase_connection)):
hb_context = connection_context.HBaseConnectionContext(
local_config)
def all_ok(connection, dummy):
eq_(dummy, 'hello')
return True
transaction = TransactionExecutor(local_config, hb_context)
result = transaction(all_ok, 'hello')
ok_(result)
eq_(a_fake_hbase_connection.close_counter, 1)
eq_(a_fake_hbase_connection.rollback_counter, 0)
eq_(a_fake_hbase_connection.commit_counter, 1)
def bad_deal(connection, dummy):
raise KeyError('fred')
assert_raises(KeyError, transaction, bad_deal, 'hello')
eq_(a_fake_hbase_connection.close_counter, 2)
eq_(a_fake_hbase_connection.commit_counter, 1)
hb_context.close()
eq_(a_fake_hbase_connection.close_counter, 2)