searxng/tests/unit/engines/test_framalibre.py

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


class TestFramalibreEngine(SearxTestCase):

    def test_request(self):
        query = 'test_query'
        dicto = defaultdict(dict)
        dicto['pageno'] = 0
        params = framalibre.request(query, dicto)
        self.assertTrue('url' in params)
        self.assertTrue(query in params['url'])
        self.assertTrue('framalibre.org' in params['url'])

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

        response = mock.Mock(text='{}')
        self.assertEqual(framalibre.response(response), [])

        response = mock.Mock(text='{"data": []}')
        self.assertEqual(framalibre.response(response), [])

        html = u"""
        <div class="nodes-list-row">
          <div id="node-431"
              class="node node-logiciel-annuaires node-promoted node-teaser node-teaser node-sheet clearfix nodes-list"
              about="/content/gogs" typeof="sioc:Item foaf:Document">
            <header class="media">
              <div class="media-left">
                <div class="field field-name-field-logo field-type-image field-label-hidden">
                  <div class="field-items">
                    <div class="field-item even">
                      <a href="/content/gogs">
                        <img class="media-object img-responsive" typeof="foaf:Image"
 src="https://framalibre.org/sites/default/files/styles/teaser_logo/public/leslogos/gogs-lg.png?itok=rrCxKKBy"
 width="70" height="70" alt="" />
                      </a>
                    </div>
                  </div>
                </div>
              </div>
              <div class="media-body">
                <h3 class="node-title"><a href="/content/gogs">Gogs</a></h3>
                <span property="dc:title" content="Gogs" class="rdf-meta element-hidden"></span>
                <div class="field field-name-field-annuaires field-type-taxonomy-term-reference field-label-hidden">
                  <div class="field-items">
                    <div class="field-item even">
                      <a href="/annuaires/cloudwebapps"
 typeof="skos:Concept" property="rdfs:label skos:prefLabel"
 datatype="" class="label label-primary">Cloud/webApps</a>
                    </div>
                  </div>
                </div>
              </div>
            </header>
            <div class="content">
              <div class="field field-name-field-votre-appr-ciation field-type-fivestar field-label-hidden">
                <div class="field-items">
                  <div class="field-item even">
                  </div>
                </div>
              </div>
              <div class="field field-name-body field-type-text-with-summary field-label-hidden">
                <div class="field-items">
                  <div class="field-item even" property="content:encoded">
                    <p>Gogs est une interface web basée sur git et une bonne alternative à GitHub.</p>
                  </div>
                </div>
              </div>
            </div>
            <footer>
              <a href="/content/gogs" class="read-more btn btn-default btn-sm">Voir la notice</a>
              <div class="field field-name-field-lien-officiel field-type-link-field field-label-hidden">
                <div class="field-items">
                  <div class="field-item even">
                    <a href="https://gogs.io/" target="_blank" title="Voir le site officiel">
                      <span class="glyphicon glyphicon-globe"></span>
                      <span class="sr-only">Lien officiel</span>
                    </a>
                  </div>
                </div>
              </div>
            </footer>
          </div>
        </div>
        """
        response = mock.Mock(text=html)
        results = framalibre.response(response)
        self.assertEqual(type(results), list)
        self.assertEqual(len(results), 1)
        self.assertEqual(results[0]['title'], 'Gogs')
        self.assertEqual(results[0]['url'],
                         'https://framalibre.org/content/gogs')
        self.assertEqual(results[0]['content'],
                         u"Gogs est une interface web basée sur git et une bonne alternative à GitHub.")