class TestInfibeamCrawler(unittest.TestCase):
def setUp(self):
self.crawler = InfibeamCrawler({ "url" : "http://localhost/page" })
def tearDown(self):
self.crawler = None
def test_crawl_page(self):
self.crawler.crawl_page = Mock(return_value="Mock Page!")
self.assertEquals("Mock Page!", self.crawler.crawl_page())