class TestConnection(BaseTestCase):
def tearDown(self):
Connection._instances.clear()
def setUp(self):
self.conn = Connection()
def test_create_table(self):
self.conn.create_table('contact', {
'd': {
'max_versions': 10,
'block_cache_enabled': True
},
'm': {}
})
table = self.conn.table('contact')
self.assertEqual(table.families(), {
'd': {
'block_cache_enabled': True,
'bloom_filter_nb_hashes': 0,
'bloom_filter_type': 'NONE',
'bloom_filter_vector_size': 0,
'compression': 'NONE',
'in_memory': False,
'max_versions': 10,
'name': 'd',
'time_to_live': -1
},
'm': {
'block_cache_enabled': False,
'bloom_filter_nb_hashes': 0,
'bloom_filter_type': 'NONE',
'bloom_filter_vector_size': 0,
'compression': 'NONE',
'in_memory': False,
'max_versions': 3,
'name': 'm',
'time_to_live': -1
}
})
def test_enable_disable_table(self):
self.conn.create_table('mytable', {'d': dict()})
self.assertTrue(self.conn.is_table_enabled('mytable'))
self.conn.disable_table('mytable')
self.assertFalse(self.conn.is_table_enabled('mytable'))
self.conn.enable_table('mytable')
self.assertTrue(self.conn.is_table_enabled('mytable'))
def test_enable_non_existing_table(self):
with self.assertRaises(IOError):
self.conn.enable_table('no_such_table')
def test_disable_non_existing_table(self):
with self.assertRaises(IOError):
self.conn.disable_table('no_such_table')
def test_is_non_existing_table_enabled(self):
# Although it's odd, Connection.is_table_enabled() returns true on
# non-existing table
self.assertTrue(self.conn.is_table_enabled('no_such_table'))
def test_list_tables(self):
families = {'d': dict()}
self.conn.create_table('apple', families)
self.conn.create_table('book', families)
self.conn.create_table('cat', families)
self.assertEqual(self.conn.tables(), ['apple', 'book', 'cat'])
def test_list_tables_with_prefix(self):
self.conn = Connection(table_prefix='abc')
self.test_list_tables()
def test_get_non_existing_table(self):
# In HappyBase, you still can get a non-existing table from a
# connection, but you will get an error if you operate on that table
table = self.conn.table('no_such_table')
with self.assertRaises(IOError):
table.put('test', {})
self.assertFalse(self.conn.tables())
def test_delete_table(self):
families = {'d': dict()}
self.conn.create_table('dog', families)
self.conn.create_table('fox', families)
self.assertEqual(self.conn.tables(), ['dog', 'fox'])
self.conn.delete_table('dog', disable=True)
self.assertEqual(self.conn.tables(), ['fox'])
def test_delete_table_with_prefix(self):
conn = Connection(table_prefix='123')
conn.create_table('jacket', {'d': dict()})
conn.delete_table('jacket', disable=True)
def test_delete_non_existing_table(self):
with self.assertRaises(IOError):
self.conn.delete_table('no_such_table')
def test_delete_enabled_table(self):
families = {'d': dict()}
self.conn.create_table('game', families)
with self.assertRaises(IOError):
self.conn.delete_table('game')
def test_open_close(self):
# TODO: open() and close() don't really do anything
self.conn.open()
self.conn.close()
def test_compact(self):
# TODO: compact_table() doesn't really do anything
self.conn.compact_table('no_such_table')
def test_delete_table_with_prefix(self):
conn = Connection(table_prefix='123')
conn.create_table('jacket', {'d': dict()})
conn.delete_table('jacket', disable=True)