| mock_askleah_class = <AsyncMock name='read_entry' id='1868710771824'>
|
|
|
| @pytest.mark.asyncio
|
| @patch('src.utils.common.common_utils.AskLeah.read_entry')
|
| async def test_get_ask_leah_response(mock_askleah_class):
|
| # Arrange
|
| tenant_db = "dummy_tenant_db"
|
| request_id = "dummy_request_id"
|
| file_guid = "dummy_file_guid"
|
| session_id = "dummy_session_id"
|
| user_id = "dummy_user_id"
|
|
|
| mock_askleah_class.return_value = [
|
| {"query": "some query", "ask_leah_response": "some response"}
|
| ]
|
|
|
| # Act
|
| result = await get_ask_leah_response(tenant_db, request_id, file_guid, session_id, user_id)
|
|
|
| # Assert
|
| mock_askleah_class.assert_called_once()
|
| > assert result == [{"query": "some query", "response": "some response"}]
|
| E AssertionError: assert [{'query': 'some query', 'ask_leah_response': 'some response'}] == [{'query': 'some query', 'response': 'some response'}]
|
| E
|
| E At index 0 diff: [0m{ [33m' [39;49;00m [33mquery [39;49;00m [33m' [39;49;00m: [33m' [39;49;00m [33msome query [39;49;00m [33m' [39;49;00m, [33m' [39;49;00m [33mask_leah_response [39;49;00m [33m' [39;49;00m: [33m' [39;49;00m [33msome response [39;49;00m [33m' [39;49;00m} [90m [39;49;00m != [0m{ [33m' [39;49;00m [33mquery [39;49;00m [33m' [39;49;00m: [33m' [39;49;00m [33msome query [39;49;00m [33m' [39;49;00m, [33m' [39;49;00m [33mresponse [39;49;00m [33m' [39;49;00m: [33m' [39;49;00m [33msome response [39;49;00m [33m' [39;49;00m} [90m [39;49;00m
|
| E
|
| E Full diff:
|
| E [0m [90m [39;49;00m [ [90m [39;49;00m
|
| E [90m [39;49;00m { [90m [39;49;00m
|
| E [92m+ 'ask_leah_response': 'some response', [39;49;00m [90m [39;49;00m
|
| E [90m [39;49;00m 'query': 'some query', [90m [39;49;00m
|
| E [91m- 'response': 'some response', [39;49;00m [90m [39;49;00m
|
| E [90m [39;49;00m }, [90m [39;49;00m
|
| E [90m [39;49;00m ] [90m [39;49;00m
|
|
|
| test_common_utils.py:405: AssertionError
|