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)
