def test_insert_failure(self, mock_warning, mock_BucketLoader, mock_uuid4,
mock_dumps):
db = mock.Mock(
**{
'lrange.return_value': ['record1', 'record2'],
'linsert.return_value': -1,
})
limit = mock.Mock(bucket_class='bucket_class')
compactor.compact_bucket(db, 'bucket_key', limit)
db.assert_has_calls([
mock.call.lrange('bucket_key', 0, -1),
mock.call.linsert('bucket_key', 'after', 'last_record',
dict(bucket='bucket', uuid='bucket_uuid')),
])
self.assertEqual(len(db.method_calls), 2)
mock_BucketLoader.assert_called_once_with('bucket_class',
db,
limit,
'bucket_key',
['record1', 'record2'],
stop_summarize=True)
mock_warning.assert_called_once_with(
"Bucket compaction on bucket_key failed; will retry")
def test_normal(self, mock_warning, mock_BucketLoader, mock_uuid4,
mock_dumps):
db = mock.Mock(
**{
'lrange.return_value': ['record1', 'record2'],
'linsert.return_value': 23,
})
limit = mock.Mock(bucket_class='bucket_class')
compactor.compact_bucket(db, 'bucket_key', limit)
db.assert_has_calls([
mock.call.lrange('bucket_key', 0, -1),
mock.call.linsert('bucket_key', 'after', 'last_record',
dict(bucket='bucket', uuid='bucket_uuid')),
mock.call.ltrim('bucket_key', 18, -1),
])
self.assertEqual(len(db.method_calls), 3)
mock_BucketLoader.assert_called_once_with('bucket_class',
db,
limit,
'bucket_key',
['record1', 'record2'],
stop_summarize=True)
self.assertFalse(mock_warning.called)
def test_insert_failure(self, mock_warning, mock_BucketLoader, mock_uuid4,
mock_dumps):
db = mock.Mock(**{
'lrange.return_value': ['record1', 'record2'],
'linsert.return_value': -1,
})
limit = mock.Mock(bucket_class='bucket_class')
compactor.compact_bucket(db, 'bucket_key', limit)
db.assert_has_calls([
mock.call.lrange('bucket_key', 0, -1),
mock.call.linsert('bucket_key', 'after', 'last_record',
dict(bucket='bucket', uuid='bucket_uuid')),
])
self.assertEqual(len(db.method_calls), 2)
mock_BucketLoader.assert_called_once_with(
'bucket_class', db, limit, 'bucket_key', ['record1', 'record2'],
stop_summarize=True)
mock_warning.assert_called_once_with(
"Bucket compaction on bucket_key failed; will retry")
def test_normal(self, mock_warning, mock_BucketLoader, mock_uuid4,
mock_dumps):
db = mock.Mock(**{
'lrange.return_value': ['record1', 'record2'],
'linsert.return_value': 23,
})
limit = mock.Mock(bucket_class='bucket_class')
compactor.compact_bucket(db, 'bucket_key', limit)
db.assert_has_calls([
mock.call.lrange('bucket_key', 0, -1),
mock.call.linsert('bucket_key', 'after', 'last_record',
dict(bucket='bucket', uuid='bucket_uuid')),
mock.call.ltrim('bucket_key', 18, -1),
])
self.assertEqual(len(db.method_calls), 3)
mock_BucketLoader.assert_called_once_with(
'bucket_class', db, limit, 'bucket_key', ['record1', 'record2'],
stop_summarize=True)
self.assertFalse(mock_warning.called)
