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")