def test_batch_load(self): records, keys = [], [] for x in range(10): record = Record() record.key = Key('eggs', 'bacon') record['number'] = x record['square'] = x * x records.append(record) keys.append(record.key) backing = {} for record in records: backing[record.key.key] = [ColumnOrSuperColumn(col) for col in record._columns.values()] mock_client = MockClient("Eggs", []) mock_client.multiget_slice = \ lambda keys, parent, pred, clvl: backing sets.itr.get_pool = lambda ks: mock_client out_records = self.object._batch_load(Record, keys) for record in out_records: self.assert_(isinstance(record, Record)) self.assert_(record.key in keys) orig = records[records.index(record)] self.assert_(orig['number'] == record['number']) self.assert_(orig['square'] == record['square'])
def test_append(self): view = self.object MockClient.insert = \ lambda conn, keyspace, key, path, value, time, x: True rec = Record() rec.key = Key(keyspace="eggs", column_family="bacon", key="tomato") self.object.append(rec)
def _get_records(self, count, **kwargs): """Return records for testing.""" records = [] for n in range(count): record = Record() if kwargs: record.key = Key(**kwargs) records.append(record) return records
def setUp(self): """Prepare the test environment.""" num_records = 15 keys = [Key(keyspace="spam", column_family="bacon") for x in range(num_records)] records = [] for key in keys: r = Record() r.key = key records.append(r) self.records = records
def setUp(self): """Prepare the test environment.""" num_records = 15 keys = [Key(keyspace='spam', column_family='bacon') for x in range(num_records)] records = [] for key in keys: r = Record() r.key = key records.append(r) self.records = records