def test_get_images(self, BatchList, Client):
mock_client = Mock()
table = Mock()
table.name = 'the-name-of-the-table'
mock_client.table.return_value = table
dynamo = Mock()
dynamo.batch_get_item.return_value = {
'UnprocessedKeys': {},
'Responses': {
'the-name-of-the-table': {
'Items': [
{
HASH_KEY: 'deadbeef',
'tags': '["cream", "cat", "money"]'},
{
HASH_KEY: 'badcafe',
'tags': '["evil", "nefarious", "cat"]'}],
'ConsumedCapacityUnits': 10.0}}}
mock_client.get_dynamodb.return_value = dynamo
Client.return_value = mock_client
batch_list = Mock()
BatchList.return_value = batch_list
images = list(get_images(['a', 'b']))
BatchList.assert_called_once_with(dynamo)
batch_list.add_batch.assert_called_once_with(
table, ['a', 'b'], attributes_to_get=['tags', HASH_KEY])
dynamo.batch_get_item.assert_called_with(batch_list)
eq_(images, self._standard_fixture_images())
def test_get_get_images__batches_very_large_requests(self, MockClient):
dynamo = Mock()
dynamo.batch_get_item.return_value = MagicMock()
client = Mock()
client.get_dynamodb.return_value = dynamo
MockClient.return_value = client
with patch('catsnap.batch.MAX_ITEMS_TO_REQUEST', 1):
list(get_images(set(['one', 'two'])))
eq_(dynamo.batch_get_item.call_count, 2)
def test_get_images__checks_back_on_unprocessed_keys(self,
BatchList, Client):
mock_client = Mock()
table = Mock()
table.name = 'thetablename'
mock_client.table.return_value = table
dynamo = Mock()
first_getitem = {
'UnprocessedKeys': {
'thetablename': {
'Keys': [
{'HashKeyElement':
'badcafe'}],
'AttributesToGet': ['tags', HASH_KEY]}},
'Responses': {
'thetablename': {
'Items': [
{
HASH_KEY: 'deadbeef',
'tags': '["cream", "cat", "money"]'}],
'ConsumedCapacityUnits': 10.0}}}
second_getitem = {
'UnprocessedKeys': {},
'Responses': {
'thetablename': {
'Items': [
{
HASH_KEY: 'badcafe',
'tags': '["evil", "nefarious", "cat"]'}],
'ConsumedCapacityUnits': 10.0}}}
dynamo.batch_get_item.side_effect = [first_getitem, second_getitem]
mock_client.get_dynamodb.return_value = dynamo
Client.return_value = mock_client
batch_list = Mock()
BatchList.return_value = batch_list
images = list(get_images(['whatever']))
eq_(images, self._standard_fixture_images())
def test_get_images__degenerate_case(self):
eq_(list(get_images([])), [])
