def test_with_fixture(self): actual_summonses = \ undertest.get_summonses_from_webpage( pkg_resources.resource_string(__name__, 'fixture.html')) self.assertEqual(5, len(actual_summonses)) self.assertEqual({arrow.get("%s 2011" % month, 'MMMM-YYYY') for month in ("August", "September", "October", "November", "December")}, set(actual_summonses.keys()))
def test_get_summonses_from_webpage_without_url(self): actual_summons = undertest.get_summonses_from_webpage(TestWebpageParsing.webpage) expected_summons = { arrow.get("August 2012", 'MMMM-YYYY'): "../../random_url1.shtml", arrow.get("December 2011", 'MMMM-YYYY'): "random_url3.shtml" } self.assertEqual(expected_summons, actual_summons)
def test_get_summonses_from_webpage_with_url(self): actual_summons = undertest.get_summonses_from_webpage(TestWebpageParsing.webpage, url="http://my.com/absolute/long/path.html") expected_summons = { arrow.get("August 2012", 'MMMM-YYYY'): "http://my.com/random_url1.shtml", arrow.get("December 2011", 'MMMM-YYYY'): "http://my.com/absolute/long/random_url3.shtml" } self.assertEqual(expected_summons, actual_summons)