Kaynağa Gözat

Correct a few UT

Cqoicebordel 10 yıl önce
ebeveyn
işleme
56ad00bc32

+ 8
- 3
searx/tests/engines/test_blekko_images.py Dosyayı Görüntüle

@@ -12,9 +12,14 @@ class TestBlekkoImagesEngine(SearxTestCase):
12 12
         dicto['pageno'] = 0
13 13
         dicto['safesearch'] = 1
14 14
         params = blekko_images.request(query, dicto)
15
-        self.assertTrue('url' in params)
16
-        self.assertTrue(query in params['url'])
17
-        self.assertTrue('blekko.com' in params['url'])
15
+        self.assertIn('url', params)
16
+        self.assertIn(query, params['url'])
17
+        self.assertIn('blekko.com', params['url'])
18
+        self.assertIn('page', params['url'])
19
+
20
+        dicto['pageno'] = 1
21
+        params = blekko_images.request(query, dicto)
22
+        self.assertNotIn('page', params['url'])
18 23
 
19 24
     def test_response(self):
20 25
         self.assertRaises(AttributeError, blekko_images.response, None)

+ 8
- 3
searx/tests/engines/test_google_images.py Dosyayı Görüntüle

@@ -11,9 +11,14 @@ class TestGoogleImagesEngine(SearxTestCase):
11 11
         dicto = defaultdict(dict)
12 12
         dicto['pageno'] = 1
13 13
         params = google_images.request(query, dicto)
14
-        self.assertTrue('url' in params)
15
-        self.assertTrue(query in params['url'])
16
-        self.assertTrue('googleapis.com' in params['url'])
14
+        self.assertIn('url', params)
15
+        self.assertIn(query, params['url'])
16
+        self.assertIn('googleapis.com', params['url'])
17
+        self.assertIn('safe=on', params['url'])
18
+
19
+        dicto['safesearch'] = 0
20
+        params = google_images.request(query, dicto)
21
+        self.assertIn('safe=off', params['url'])
17 22
 
18 23
     def test_response(self):
19 24
         self.assertRaises(AttributeError, google_images.response, None)

+ 2
- 12
searx/tests/engines/test_yahoo.py Dosyayı Görüntüle

@@ -75,12 +75,6 @@ class TestYahooEngine(SearxTestCase):
75 75
     <li>
76 76
         <div class="dd algo lst Sr">
77 77
             <div class="compTitle">
78
-                <h3 class="title"><a class=" td-u" href="http://r.search.yahoo.com/_ylt=AwrBT7zgEudUW.wAe2ZXNyoA;
79
-                     _ylu=X3oDMTBybGY3bmpvBGNvbG8DYmYxBHBvcwMyBHZ0aWQDBHNlYwNzcg--/RV=2\/RE=1424458593/RO=10
80
-                     /RU=https%3a%2f%2fthis.is.the.second.url%2f/RK=0/RS=jIctjj_cBH1Efj88GCgHKp3__Qk-"
81
-                     target="_blank" data-bid="54e712e136926">
82
-                     This is the second <b><b>title</b></b></a>
83
-                </h3>
84 78
             </div>
85 79
             <div class="compText aAbs">
86 80
                 <p class="lh-18">This is the second content</p>
@@ -102,16 +96,12 @@ class TestYahooEngine(SearxTestCase):
102 96
         """
103 97
         response = mock.Mock(text=html)
104 98
         results = yahoo.response(response)
105
-        print results
106 99
         self.assertEqual(type(results), list)
107
-        self.assertEqual(len(results), 3)
100
+        self.assertEqual(len(results), 2)
108 101
         self.assertEqual(results[0]['title'], 'This is the title')
109 102
         self.assertEqual(results[0]['url'], 'https://this.is.the.url/')
110 103
         self.assertEqual(results[0]['content'], 'This is the content')
111
-        self.assertEqual(results[1]['title'], 'This is the second title')
112
-        self.assertEqual(results[1]['url'], 'https://this.is.the.second.url/')
113
-        self.assertEqual(results[1]['content'], 'This is the second content')
114
-        self.assertEqual(results[2]['suggestion'], 'This is the suggestion')
104
+        self.assertEqual(results[1]['suggestion'], 'This is the suggestion')
115 105
 
116 106
         html = """
117 107
 <ol class="reg mb-15 searchCenterMiddle">