| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112 |
- # -*- coding: utf-8 -*-
- from collections import defaultdict
- import mock
- from searx.engines import btdigg
- from searx.testing import SearxTestCase
- class TestBtdiggEngine(SearxTestCase):
- def test_request(self):
- query = 'test_query'
- dicto = defaultdict(dict)
- dicto['pageno'] = 0
- params = btdigg.request(query, dicto)
- self.assertIn('url', params)
- self.assertIn(query, params['url'])
- self.assertIn('btdig.com', params['url'])
- def test_response(self):
- self.assertRaises(AttributeError, btdigg.response, None)
- self.assertRaises(AttributeError, btdigg.response, [])
- self.assertRaises(AttributeError, btdigg.response, '')
- self.assertRaises(AttributeError, btdigg.response, '[]')
- response = mock.Mock(text='<html></html>')
- self.assertEqual(btdigg.response(response), [])
- html = u"""
- <div class="one_result" style="display:table-row;background-color:#e8e8e8">
- <div style="display:table-cell;color:rgb(0, 0, 0)">
- <div style="display:table">
- <div style="display:table-row">
- <div class="torrent_name" style="display:table-cell">
- <a style="color:rgb(0, 0, 204);text-decoration:underline;font-size:150%"
- href="http://btdig.com/a72f35b7ee3a10928f02bb799e40ae5db701ed1c/pdf?q=pdf&p=1&order=0"
- >3.9GBdeLibrosByHuasoFromHell(3de4)</a>
- </div>
- </div>
- </div>
- <div style="display:table">
- <div style="display:table-row">
- <div style="display:table-cell">
- <span class="torrent_files" style="color:#666;padding-left:10px">4217</span> files <span
- class="torrent_size" style="color:#666;padding-left:10px">1 GB</span><span
- class="torrent_age" style="color:rgb(0, 102, 0);padding-left:10px;margin: 0px 4px"
- >found 3 years ago</span>
- </div>
- </div>
- </div>
- <div style="display:table;width:100%;padding:10px">
- <div style="display:table-row">
- <div class="torrent_magnet" style="display:table-cell">
- <div class="fa fa-magnet" style="color:#cc0000">
- <a href="magnet:?xt=urn:btih:a72f35b7ee3a10928f02bb799e40ae5db701ed1c&dn=3.9GBdeLibrosBy..."
- title="Download via magnet-link"> magnet:?xt=urn:btih:a72f35b7ee...</a>
- </div>
- </div>
- <div style="display:table-cell;color:rgb(0, 0, 0);text-align:right">
- <span style="color:rgb(136, 136, 136);margin: 0px 0px 0px 4px"></span><span
- style="color:rgb(0, 102, 0);margin: 0px 4px">found 3 years ago</span>
- </div>
- </div>
- </div>
- <div class="torrent_excerpt" style="display:table;padding:10px;white-space:nowrap">
- <div class="fa fa-folder-open" style="padding-left:0em"> 3.9GBdeLibrosByHuasoFromHell(3de4)</div><br/>
- <div class="fa fa-folder-open" style="padding-left:1em"> Libros H-Z</div><br/>
- <div class="fa fa-folder-open" style="padding-left:2em"> H</div><br/><div class="fa fa-file-archive-o"
- style="padding-left:3em"> H.H. Hollis - El truco de la espada-<b
- style="color:red; background-color:yellow">pdf</b>.zip</div><span
- style="color:#666;padding-left:10px">17 KB</span><br/>
- <div class="fa fa-file-archive-o" style="padding-left:3em"> Hagakure - El Libro del Samurai-<b
- style="color:red; background-color:yellow">pdf</b>.zip</div><span
- style="color:#666;padding-left:10px">95 KB</span><br/>
- <div class="fa fa-folder-open" style="padding-left:3em"> Hamsun, Knut (1859-1952)</div><br/>
- <div class="fa fa-file-archive-o" style="padding-left:4em"> Hamsun, Knut - Hambre-<b
- style="color:red; background-color:yellow">pdf</b>.zip</div><span
- style="color:#666;padding-left:10px">786 KB</span><br/>
- <div class="fa fa-plus-circle"><a
- href="http://btdig.com/a72f35b7ee3a10928f02bb799e40ae5db701ed1c/pdf?q=pdf&p=1&order=0"
- > 4214 hidden files<span style="color:#666;padding-left:10px">1 GB</span></a></div>
- </div>
- </div>
- </div>
- """
- response = mock.Mock(text=html.encode('utf-8'))
- results = btdigg.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 1)
- self.assertEqual(results[0]['title'], '3.9GBdeLibrosByHuasoFromHell(3de4)')
- self.assertEqual(results[0]['url'],
- 'http://btdig.com/a72f35b7ee3a10928f02bb799e40ae5db701ed1c/pdf?q=pdf&p=1&order=0')
- self.assertEqual(results[0]['content'],
- '3.9GBdeLibrosByHuasoFromHell(3de4) | ' +
- 'Libros H-Z | ' +
- 'H H.H. Hollis - El truco de la espada-pdf.zip17 KB | ' +
- 'Hagakure - El Libro del Samurai-pdf.zip95 KB | ' +
- 'Hamsun, Knut (1859-1952) | Hamsun, Knut - Hambre-pdf.zip786 KB | ' +
- '4214 hidden files1 GB')
- self.assertEqual(results[0]['filesize'], 1 * 1024 * 1024 * 1024)
- self.assertEqual(results[0]['files'], 4217)
- self.assertEqual(results[0]['magnetlink'],
- 'magnet:?xt=urn:btih:a72f35b7ee3a10928f02bb799e40ae5db701ed1c&dn=3.9GBdeLibrosBy...')
- html = """
- <div style="display:table-row;background-color:#e8e8e8">
- </div>
- """
- response = mock.Mock(text=html.encode('utf-8'))
- results = btdigg.response(response)
- self.assertEqual(type(results), list)
- self.assertEqual(len(results), 0)
|
