eregs/regulations-core

import pytest
from django.test import TestCase, override_settings
from django.test.client import Client
from mock import patch

pytest.importorskip('pyelasticsearch')  # noqa
from regcore_read.views.es_search import transform_results


@override_settings(SEARCH_HANDLER='regcore_read.views.es_search.search')
class ViewsESSearchTest(TestCase):
    def test_search_missing_q(self):
        response = Client().get('/search?non_q=test')
        self.assertEqual(400, response.status_code)

    @patch('regcore_read.views.es_search.ElasticSearch')
    def test_search_success(self, es):
        es.return_value.search.return_value = {'hits': {'hits': [],
                                                        'total': 0}}
        response = Client().get('/search?q=test')
        self.assertEqual(200, response.status_code)
        self.assertTrue(es.called)
        self.assertTrue(es.return_value.search.called)

    @patch('regcore_read.views.es_search.ElasticSearch')
    def test_search_version(self, es):
        es.return_value.search.return_value = {'hits': {'hits': [],
                                                        'total': 0}}
        response = Client().get('/search?q=test&version=12345678')
        self.assertEqual(200, response.status_code)
        self.assertTrue(es.called)
        self.assertTrue(es.return_value.search.called)
        self.assertIn('12345678', str(es.return_value.search.call_args))

    @patch('regcore_read.views.es_search.ElasticSearch')
    def test_search_version_regulation(self, es):
        es.return_value.search.return_value = {'hits': {'hits': [],
                                                        'total': 0}}
        response = Client().get('/search?q=test&version=678&regulation=123')
        self.assertEqual(200, response.status_code)
        self.assertTrue(es.called)
        self.assertTrue(es.return_value.search.called)
        self.assertIn('678', str(es.return_value.search.call_args))
        self.assertIn('123', str(es.return_value.search.call_args))

    @patch('regcore_read.views.es_search.ElasticSearch')
    def test_search_paging(self, es):
        es.return_value.search.return_value = {'hits': {'hits': [],
                                                        'total': 0}}
        response = Client().get('/search?q=test&page=5')
        self.assertEqual(200, response.status_code)
        self.assertTrue(es.called)
        self.assertTrue(es.return_value.search.called)
        query = es.return_value.search.call_args[0][0]
        self.assertEqual(50, query['size'])
        self.assertEqual(250, query['from'])

    @patch('regcore_read.views.es_search.ESLayers')
    def test_transform_results(self, eslayers):
        # combine keyterms and terms into a single layer
        eslayers.return_value.get.return_value = {
            '2': [{'key_term': 'k2'}], '3': [{'key_term': 'k3'}],
            '6': [{'key_term': 'k6'}], '7': [{'key_term': 'k7'}],
            'referenced': {
                'lab1': {'reference': '1', 'term': 'd1'},
                'lab2': {'reference': '3', 'term': 'd3'},
                'lab3': {'reference': '5', 'term': 'd5'},
                'lab4': {'reference': '7', 'term': 'd7'}
            }
        }
        results = transform_results([
            {'regulation': 'r', 'version': 'v', 'label_string': '0'},
            {'regulation': 'rr', 'version': 'v', 'label_string': '1'},
            {'regulation': 'r', 'version': 'vv', 'label_string': '2'},
            {'regulation': 'r', 'version': 'v', 'label_string': '3'},
            {'regulation': 'rr', 'version': 'vv', 'label_string': '4',
             'title': 't4'},
            {'regulation': 'r', 'version': 'vv', 'label_string': '5',
             'title': 't5'},
            {'regulation': 'rr', 'version': 'v', 'label_string': '6',
             'title': 't6'},
            {'regulation': 'r', 'version': 'v', 'label_string': '7',
             'title': 't7'}])

        self.assertNotIn('title', results[0])
        self.assertEqual('d1', results[1]['title'])
        self.assertEqual('k2', results[2]['title'])
        self.assertEqual('k3', results[3]['title'])
        self.assertEqual('t4', results[4]['title'])
        self.assertEqual('t5', results[5]['title'])
        self.assertEqual('t6', results[6]['title'])
        self.assertEqual('t7', results[7]['title'])