def test_get_results_serializer_filters(self, mock_process, mock_add_repo, mock_ctrl): """ Get results, ensure that filters are translated. """ m_serial = mock_ctrl.get_model_serializer_for_type.return_value content_search = ContentUnitSearch() mock_query = {'filters': {'mock': 'filters'}} mock_search = mock.MagicMock(return_value=['result_1', 'result_2']) serialized_results = content_search.get_results( mock_query, mock_search, {'include_repos': True}, type_id='mock_type') m_serial.translate_filters.assert_called_once_with( m_serial.model, {'mock': 'filters'}) self.assertEqual(m_serial.translate_filters.call_count, 1) mock_process.assert_has_calls([ mock.call('result_1', 'mock_type'), mock.call('result_2', 'mock_type') ]) self.assertEqual( serialized_results, [mock_process.return_value, mock_process.return_value]) mock_add_repo.assert_called_once_with( [mock_process(), mock_process()], 'mock_type')
def test_get_results_without_repos(self, mock_process, mock_add_repo): """ Get results without the optional `include_repos`. """ content_search = ContentUnitSearch() mock_query = mock.MagicMock() mock_search = mock.MagicMock(return_value=['result_1', 'result_2']) serialized_results = content_search.get_results(mock_query, mock_search, {}, type_id='mock_type') mock_process.assert_has_calls([mock.call('result_1', 'mock_type'), mock.call('result_2', 'mock_type')]) self.assertEqual(serialized_results, [mock_process.return_value, mock_process.return_value]) self.assertEqual(mock_add_repo.call_count, 0)
def test_get_results_without_repos(self, mock_process, mock_add_repo): """ Get results without the optional `include_repos`. """ content_search = ContentUnitSearch() mock_query = mock.MagicMock() mock_search = mock.MagicMock(return_value=['result_1', 'result_2']) serialized_results = content_search.get_results(mock_query, mock_search, {}, type_id='mock_type') mock_process.assert_has_calls([mock.call('result_1', 'mock_type'), mock.call('result_2', 'mock_type')]) self.assertEqual(serialized_results, [mock_process.return_value, mock_process.return_value]) self.assertEqual(mock_add_repo.call_count, 0)
def test_get_results_serializer_no_filters(self, mock_process, mock_add_repo, mock_ctrl): """ Get results, ensure that if the query does not have filters, they are not translated. """ m_serializer = mock_ctrl.get_model_serializer_for_type.return_value content_search = ContentUnitSearch() mock_query = {} mock_search = mock.MagicMock(return_value=['result_1', 'result_2']) serialized_results = content_search.get_results( mock_query, mock_search, {'include_repos': True}, type_id='mock_type' ) self.assertEqual(m_serializer.translate_filters.call_count, 0) mock_process.assert_has_calls([mock.call('result_1', 'mock_type'), mock.call('result_2', 'mock_type')]) self.assertEqual(serialized_results, [mock_process.return_value, mock_process.return_value]) mock_add_repo.assert_called_once_with([mock_process(), mock_process()], 'mock_type')