def test_keys_types(self):
"""Ensure that View._keys() returns the correct type & keys."""
view = self.object
keys = view._keys()
self.assert_(hasattr(keys, '__iter__'))
for key in tuple(keys):
self.assert_(isinstance(key, Key))
self.assert_(key.keyspace == view.record_key.keyspace)
self.assert_(key.column_family == view.record_key.column_family)
view._keys("foo", "bar")
def test_keys_types(self):
"""Ensure that View._keys() returns the correct type & keys."""
view = self.object
keys = view._keys()
self.assert_(hasattr(keys, '__iter__'))
for key in tuple(keys):
self.assert_(isinstance(key, Key))
self.assert_(key.keyspace == view.record_key.keyspace)
self.assert_(key.column_family == view.record_key.column_family)
view._keys("foo", "bar")
def __base_view_test(self, view, ncols, chunk_size):
"""Base test for View._keys iteration."""
cols = map(ColumnOrSuperColumn,
[Column(name=x, value=x)
for x in range(ncols)])
def get_slice(instance, keyspace, key, parent, predicate, level):
try:
start = int(predicate.slice_range.start)
except ValueError:
start = 0
count = int(predicate.slice_range.count)
return cols[start:start + count]
view.chunk_size = chunk_size
MockClient.get_slice = get_slice
keys = tuple(view._keys())
self.assert_(len(keys) == len(cols),
"Got %s keys instead of of %s" %
(len(keys), len(cols)))
self.assert_(len(set(keys)) == len(keys),
"Duplicates present in output")
def __base_view_test(self, view, ncols, chunk_size):
"""Base test for View._keys iteration."""
cols = map(ColumnOrSuperColumn,
[Column(name=x, value=x)
for x in range(ncols)])
def get_slice(instance, keyspace, key, parent, predicate, level):
try:
start = int(predicate.slice_range.start)
except ValueError:
start = 0
count = int(predicate.slice_range.count)
return cols[start:start + count]
view.chunk_size = chunk_size
MockClient.get_slice = get_slice
keys = tuple(view._keys())
self.assert_(len(keys) == len(cols),
"Got %s keys instead of of %s" %
(len(keys), len(cols)))
self.assert_(len(set(keys)) == len(keys),
"Duplicates present in output")
