searxng/tests/unit/test_engine_mariadb_server.py

# SPDX-License-Identifier: AGPL-3.0-or-later
# pylint: disable=missing-module-docstring,disable=missing-class-docstring,invalid-name

from unittest.mock import MagicMock, Mock

from searx.engines import mariadb_server
from tests import SearxTestCase


class MariadbServerTests(SearxTestCase):

    def test_init_no_query_str_raises(self):
        self.assertRaises(ValueError, lambda: mariadb_server.init({}))

    def test_init_non_select_raises(self):
        self.assertRaises(ValueError, lambda: mariadb_server.init({'query_str': 'foobar'}))

    def test_search_returns_results(self):
        test_string = 'FOOBAR'
        cursor_mock = MagicMock()
        with cursor_mock as setup:  # pylint: disable=not-context-manager
            setup.__iter__ = Mock(return_value=iter([{test_string, 1}]))
            setup.description = [[test_string]]
        conn_mock = Mock()
        conn_mock.cursor.return_value = cursor_mock
        mariadb_server._connection = conn_mock  # pylint: disable=protected-access
        results = mariadb_server.search(test_string, {'pageno': 1})
        self.assertEqual(1, len(results))
        self.assertIn(test_string, results[0])
        self.assertEqual(mariadb_server.result_template, results[0]['template'])