34 lines
814 B
Python
34 lines
814 B
Python
import unittest
|
|
from pprint import pprint
|
|
|
|
from spider.spider import check_and_rate_site
|
|
|
|
class TestSpider(unittest.TestCase):
|
|
|
|
"""
|
|
Simply calls the spider.check_and_rate_site function
|
|
with httpbin.org URLs. We don't assert a lot here,
|
|
but at least make sure that most of our code is executed
|
|
in tests.
|
|
"""
|
|
|
|
def test_html(self):
|
|
"""Loads a simple HTML web page"""
|
|
|
|
entry = {
|
|
"url": "https://httpbin.org/html",
|
|
"type": "type",
|
|
"state": "state",
|
|
"level": "level",
|
|
"district": "district",
|
|
"city": "city",
|
|
}
|
|
|
|
url = "https://httpbin.org/html"
|
|
result = check_and_rate_site(entry)
|
|
|
|
self.assertEqual(result["input_url"], url)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|