searx/tests/unit/engines/pubmed.py

# -*- coding: utf-8 -*-
from collections import defaultdict
import mock
from searx.engines import pubmed
from searx.testing import SearxTestCase


class TestPubmedEngine(SearxTestCase):

    def test_request(self):
        query = 'test_query'
        dicto = defaultdict(dict)
        dicto['pageno'] = 1
        params = pubmed.request(query, dicto)
        self.assertIn('url', params)
        self.assertIn('eutils.ncbi.nlm.nih.gov/', params['url'])
        self.assertIn('term', params['url'])

    def test_response(self):
        self.assertRaises(AttributeError, pubmed.response, None)
        self.assertRaises(AttributeError, pubmed.response, [])
        self.assertRaises(AttributeError, pubmed.response, '')
        self.assertRaises(AttributeError, pubmed.response, '[]')

        response = mock.Mock(text='<PubmedArticleSet></PubmedArticleSet>')
        self.assertEqual(pubmed.response(response), [])

        xml_mock = """<eSearchResult><Count>1</Count><RetMax>1</RetMax><RetStart>0</RetStart><IdList>
<Id>1</Id>
</IdList></eSearchResult>
"""

        response = mock.Mock(text=xml_mock.encode('utf-8'))
        results = pubmed.response(response)
        self.assertEqual(type(results), list)
        self.assertEqual(len(results), 1)
        self.assertEqual(results[0]['content'], 'No abstract is available for this publication.')