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']})