"""Lambda test module.""" from unittest.mock import patch import pytest from src.logic import review_queue_item @patch('src.logic.product.gql_connector.fetch_review_queue_item_data') def test_get_review_queue_item(mock_fetch_item, mock_review_queue_item): """Test get_review_queue_item.""" mock_fetch_item.return_value = mock_review_queue_item result = review_queue_item.get_review_queue_item(111) assert result == { 'product_id': 1, 'moved_by_user_name': 'foo baz', 'escalation_type': 'some target', 'target_group': 'some group', 'move_note': 'sample note xyz' } @patch('src.logic.product.gql_connector.fetch_review_queue_item_data') def test_get_review_queue_item_no_target(mock_fetch_item, mock_item_no_target): """Test get_review_queue_item no target.""" mock_fetch_item.return_value = mock_item_no_target result = review_queue_item.get_review_queue_item(111) assert result == { 'product_id': 1, 'moved_by_user_name': 'foo baz', 'escalation_type': None, 'target_group': None, 'move_note': 'sample note xyz' } @patch('src.logic.product.gql_connector.fetch_review_queue_item_data') def test_get_review_queue_item_error(mock_fetch_item): """Test get_review_queue_item error.""" mock_fetch_item.side_effect = Exception('sorry, there was a gql error') with pytest.raises(Exception): review_queue_item.get_review_queue_item(111)