NB-Public/green-spider
14
0
Fork 1
mirror of https://github.com/netzbegruenung/green-spider.git synced 2024-04-30 16:14:51 +02:00
Code Issues 2 Projects Releases Wiki Activity

Add unit test for reduce_urls function

Browse source
This commit is contained in:
Marian Steinbach 2018-05-03 13:55:34 +02:00
parent 16ca2699d0
commit 7dce339801
1 changed files with 15 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
test.py
View file

@ -3,15 +3,27 @@ import spider
class TestSpider(unittest.TestCase):
def test_derive_test_hostnames1(self):
def test_derive_test_hostnames(self):
# case 1
hn = spider.derive_test_hostnames('www.my-domain.de')
expected = ['my-domain.de', 'www.my-domain.de']
self.assertEqual(hn, expected)
def test_derive_test_hostnames2(self):
# case 2
hn = spider.derive_test_hostnames('domain.de')
expected = ['domain.de', 'www.domain.de']
self.assertEqual(hn, expected)
def test_reduce_urls(self):
# This is our testdata
testdata = [
{'url': 'one', 'error': None, 'redirects_to': None},
{'url': 'two', 'error': 'Yes', 'redirects_to': None},
{'url': 'three', 'error': None, 'redirects_to': 'five'},
]
expected_result = ['five', 'one']
result = spider.reduce_urls(testdata)
self.assertEqual(result, expected_result)
if __name__ == '__main__':
unittest.main()