def test_slice_iterator(self):
"""Test slice_iterator."""
key = Key(keyspace="eggs", column_family="bacon", key="tomato")
slice_iterator = iterators.slice_iterator(key, ConsistencyLevel.ONE)
self.assert_(isinstance(slice_iterator, types.GeneratorType))
for col in slice_iterator:
self.assert_(isinstance(col, ttypes.Column))
def load(self, key, consistency=None):
"""Load this record from primary key"""
if not isinstance(key, Key):
key = self.make_key(key)
self._clean()
consistency = consistency or self.consistency
return self._inject(key, iterators.slice_iterator(key, consistency))
def load(self, key, consistency=None):
"""Load this record from primary key"""
if not isinstance(key, Key):
key = self.make_key(key)
self._clean()
consistency = consistency or self.consistency
return self._inject(key, iterators.slice_iterator(key, consistency))
def load(self, key, consistency=None):
"""Load this record from primary key"""
if not isinstance(key, Key):
key = self.make_key(key)
self._clean()
consistency = consistency or self.consistency
columns = iterators.slice_iterator(key, consistency)
self._inject(key, dict([(column.name, column) for column in columns]))
return self
def test_slice_iterator_supercolumns(self):
"""Test slice_iterator with supercolumns."""
key = Key(keyspace="eggs", column_family="bacon", key="tomato",
super_column="spam")
cols = [bla.column for bla in self.client.get_slice()]
scol = ttypes.SuperColumn(name="spam", columns=cols)
corsc = ttypes.ColumnOrSuperColumn(super_column=scol)
self.client.get_slice = lambda *args: [corsc]
slice_iterator = iterators.slice_iterator(key, ConsistencyLevel.ONE)
self.assert_(isinstance(slice_iterator, types.GeneratorType))
for col in slice_iterator:
self.assert_(isinstance(col, ttypes.SuperColumn))
