def test_execute_get_batch(self, mock_get_item, mock_repo_get,
mock_validate_table_is_active,
mock_validate_table_schema):
future = Future()
future.set_result(True)
mock_get_item.return_value = future
context = mock.Mock(tenant='fake_tenant')
table_name = 'fake_table'
request_list = [
models.GetItemRequest(
table_name,
{
'id': models.AttributeValue('N', 1),
'str': models.AttributeValue('S', 'str1'),
},
None,
True
),
models.GetItemRequest(
table_name,
{
'id': models.AttributeValue('N', 1),
'str': models.AttributeValue('S', 'str2'),
},
None,
True
)
]
expected_get = [mock.call(context, req.table_name,
req.key_attribute_map,
req.attributes_to_get, req.consistent)
for req in request_list]
storage_manager = SimpleStorageManager(None, TableInfoRepository())
result, unprocessed_items = storage_manager.execute_get_batch(
context, request_list
)
mock_get_item.has_calls(expected_get)
self.assertEqual(unprocessed_items, [])
def test_execute_get_batch(self, mock_get_item, mock_repo_get,
mock_validate_table_is_active,
mock_validate_table_schema):
future = Future()
future.set_result(True)
mock_get_item.return_value = future
context = mock.Mock(tenant='fake_tenant')
table_name = 'fake_table'
request_list = [
models.GetItemRequest(
table_name, {
'id': models.AttributeValue('N', 1),
'str': models.AttributeValue('S', 'str1'),
}, None, True),
models.GetItemRequest(
table_name, {
'id': models.AttributeValue('N', 1),
'str': models.AttributeValue('S', 'str2'),
}, None, True)
]
expected_get = [
mock.call(context, req.table_name, req.key_attribute_map,
req.attributes_to_get, req.consistent)
for req in request_list
]
storage_manager = SimpleStorageManager(None, TableInfoRepository())
result, unprocessed_items = storage_manager.execute_get_batch(
context, request_list)
mock_get_item.has_calls(expected_get)
self.assertEqual(unprocessed_items, [])
