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, [])