mirror of
https://github.com/netzbegruenung/green-spider.git
synced 2024-05-02 09:04:51 +02:00
Add unit test for reduce_urls function
This commit is contained in:
parent
16ca2699d0
commit
7dce339801
18
test.py
18
test.py
|
@ -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()
|
||||
|
|
Loading…
Reference in a new issue