on eecfff2 help sign in
searx/tests/unit/test_query.py
97 lines 3.5 kB history raw
1from searx.query import RawTextQuery
2from searx.testing import SearxTestCase
5class TestQuery(SearxTestCase):
7 def test_simple_query(self):
8 query_text = 'the query'
9 query = RawTextQuery(query_text, [])
11 self.assertEqual(query.getFullQuery(), query_text)
12 self.assertEqual(len(query.query_parts), 1)
13 self.assertEqual(len(query.languages), 0)
14 self.assertFalse(query.specific)
16 def test_language_code(self):
17 language = 'es-ES'
18 query_text = 'the query'
19 full_query = ':' + language + ' ' + query_text
20 query = RawTextQuery(full_query, [])
22 self.assertEqual(query.getFullQuery(), full_query)
23 self.assertEqual(len(query.query_parts), 3)
24 self.assertEqual(len(query.languages), 1)
25 self.assertIn(language, query.languages)
26 self.assertFalse(query.specific)
28 def test_language_name(self):
29 language = 'english'
30 query_text = 'the query'
31 full_query = ':' + language + ' ' + query_text
32 query = RawTextQuery(full_query, [])
34 self.assertEqual(query.getFullQuery(), full_query)
35 self.assertEqual(len(query.query_parts), 3)
36 self.assertIn('en', query.languages)
37 self.assertFalse(query.specific)
39 def test_unlisted_language_code(self):
40 language = 'all'
41 query_text = 'the query'
42 full_query = ':' + language + ' ' + query_text
43 query = RawTextQuery(full_query, [])
45 self.assertEqual(query.getFullQuery(), full_query)
46 self.assertEqual(len(query.query_parts), 3)
47 self.assertIn('all', query.languages)
48 self.assertFalse(query.specific)
50 def test_invalid_language_code(self):
51 language = 'not_a_language'
52 query_text = 'the query'
53 full_query = ':' + language + ' ' + query_text
54 query = RawTextQuery(full_query, [])
56 self.assertEqual(query.getFullQuery(), full_query)
57 self.assertEqual(len(query.query_parts), 1)
58 self.assertEqual(len(query.languages), 0)
59 self.assertFalse(query.specific)
61 def test_timeout_below100(self):
62 query_text = '<3 the query'
63 query = RawTextQuery(query_text, [])
65 self.assertEqual(query.getFullQuery(), query_text)
66 self.assertEqual(len(query.query_parts), 3)
67 self.assertEqual(query.timeout_limit, 3)
68 self.assertFalse(query.specific)
70 def test_timeout_above100(self):
71 query_text = '<350 the query'
72 query = RawTextQuery(query_text, [])
74 self.assertEqual(query.getFullQuery(), query_text)
75 self.assertEqual(len(query.query_parts), 3)
76 self.assertEqual(query.timeout_limit, 0.35)
77 self.assertFalse(query.specific)
79 def test_timeout_above1000(self):
80 query_text = '<3500 the query'
81 query = RawTextQuery(query_text, [])
83 self.assertEqual(query.getFullQuery(), query_text)
84 self.assertEqual(len(query.query_parts), 3)
85 self.assertEqual(query.timeout_limit, 3.5)
86 self.assertFalse(query.specific)
88 def test_timeout_invalid(self):
89 # invalid number: it is not bang but it is part of the query
90 query_text = '<xxx the query'
91 query = RawTextQuery(query_text, [])
93 self.assertEqual(query.getFullQuery(), query_text)
94 self.assertEqual(len(query.query_parts), 1)
95 self.assertEqual(query.query_parts[0], query_text)
96 self.assertEqual(query.timeout_limit, None)
97 self.assertFalse(query.specific)