Пример #1
0
    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'])
Пример #2
0
 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)
Пример #3
0
 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)
Пример #4
0
    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
Пример #5
0
    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
Пример #6
0
    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