def test_column_families(self):
'''
Test for Return existing column families for all keyspaces
'''
mock_keyspaces = ['A', 'B']
class MockSystemManager(object):
def list_keyspaces(self):
return mock_keyspaces
def get_keyspace_column_families(self, keyspace):
if keyspace == 'A':
return {'a': 'saltines', 'b': 'biscuits'}
if keyspace == 'B':
return {'c': 'cheese', 'd': 'crackers'}
mock_sys_mgr = MagicMock(return_value=MockSystemManager())
with patch.object(cassandra, '_sys_mgr', mock_sys_mgr):
self.assertEqual(cassandra.column_families('Z'), None)
if six.PY3:
self.assertCountEqual(cassandra.column_families('A'),
['a', 'b'])
self.assertCountEqual(cassandra.column_families(), {
'A': ['a', 'b'],
'B': ['c', 'd']
})
else:
self.assertEqual(cassandra.column_families('A'), ['a', 'b'])
self.assertEqual(cassandra.column_families(), {
'A': ['a', 'b'],
'B': ['c', 'd']
})
def test_column_families(self):
"""
Test for Return existing column families for all keyspaces
"""
mock_keyspaces = ["A", "B"]
class MockSystemManager(object):
def list_keyspaces(self):
return mock_keyspaces
def get_keyspace_column_families(self, keyspace):
if keyspace == "A":
return {"a": "saltines", "b": "biscuits"}
if keyspace == "B":
return {"c": "cheese", "d": "crackers"}
mock_sys_mgr = MagicMock(return_value=MockSystemManager())
with patch.object(cassandra, "_sys_mgr", mock_sys_mgr):
self.assertEqual(cassandra.column_families("Z"), None)
if six.PY3:
self.assertCountEqual(cassandra.column_families("A"), ["a", "b"])
self.assertCountEqual(
cassandra.column_families(), {"A": ["a", "b"], "B": ["c", "d"]}
)
else:
self.assertEqual(sorted(cassandra.column_families("A")), ["a", "b"])
column_families = cassandra.column_families()
for key in ("A", "B"):
column_families[key] = sorted(column_families[key])
self.assertEqual(column_families, {"A": ["a", "b"], "B": ["c", "d"]})
def test_column_families(self):
"""
Test for Return existing column families for all keyspaces
"""
mock_keyspaces = ["A", "B"]
class MockSystemManager(object):
def list_keyspaces(self):
return mock_keyspaces
def get_keyspace_column_families(self, keyspace):
if keyspace == "A":
return {"a": "saltines", "b": "biscuits"}
if keyspace == "B":
return {"c": "cheese", "d": "crackers"}
mock_sys_mgr = MagicMock(return_value=MockSystemManager())
with patch.object(cassandra, "_sys_mgr", mock_sys_mgr):
self.assertEqual(cassandra.column_families("Z"), None)
if six.PY3:
self.assertCountEqual(cassandra.column_families("A"), ["a", "b"])
self.assertCountEqual(cassandra.column_families(), {"A": ["a", "b"], "B": ["c", "d"]})
else:
self.assertEqual(cassandra.column_families("A"), ["a", "b"])
self.assertEqual(cassandra.column_families(), {"A": ["a", "b"], "B": ["c", "d"]})
def test_column_families(self):
'''
Test for Return existing column families for all keyspaces
'''
mock_keyspaces = ['A', 'B']
class MockSystemManager(object):
def list_keyspaces(self):
return mock_keyspaces
def get_keyspace_column_families(self, keyspace):
if keyspace == 'A':
return {'a': 'saltines', 'b': 'biscuits'}
if keyspace == 'B':
return {'c': 'cheese', 'd': 'crackers'}
mock_sys_mgr = MagicMock(return_value=MockSystemManager())
with patch.object(cassandra, '_sys_mgr', mock_sys_mgr):
self.assertEqual(cassandra.column_families('A'),
['a', 'b'])
self.assertEqual(cassandra.column_families('Z'),
None)
self.assertEqual(cassandra.column_families(),
{'A': ['a', 'b'], 'B': ['c', 'd']})