def test_get_date(self):
today = datetime.datetime.today()
for file in TEST_FILE_LIST:
file = "%s/%s" % (TEST_DATA_DIR, file)
r = RSSReader(file)
readDate = r.get_date()
self.assertEqual(today.year, readDate.year)
self.assertEqual(today.month, readDate.month)
self.assertEqual(today.day, readDate.day)
def test_rss_get(self):
for file in TEST_FILE_LIST:
file = "%s/%s" % (TEST_DATA_DIR, file)
r = RSSReader(file)
self.assertEqual(len(r.feed["entries"]), r.get_num_entries())
self.assertEqual(r.url, file)
d = r.get_entries()
for entry in r.feed["entries"]:
rawKey = entry["title"]
self.assertTrue(d.has_key(rawKey))
self.assertTrue(entry["link"] in d[rawKey])
self.assertTrue(r.has_entries())
r.feed["entries"] = {}
self.assertFalse(r.has_entries())
del r.feed["entries"]
self.assertFalse(r.has_entries())
def test_feed(self):
for file in TEST_FILE_LIST:
file = "%s/%s" % (TEST_DATA_DIR, file)
r = RSSReader(file)
self.assertEqual(r.get_url(), file)
def test_empty_data(self):
r = RSSReader(EMPTY_FILE)
self.assertFalse(r.has_entries())
errCount = 0
try:
r.get_entries()
except Exception as ans:
errCount += 1
self.assertTrue(isinstance(ans, KeyNotFoundException))
r.feed["entries"] =[]
try:
r.get_entries()
except Exception as ans:
errCount += 1
self.assertTrue(isinstance(ans, KeyNotFoundException))
r.feed["entries"].append({})
try:
r.get_entries()
except Exception as ans:
errCount += 1
self.assertTrue(isinstance(ans, KeyNotFoundException))
r.feed["entries"][0]["title"]="hi"
try:
r.get_entries()
except Exception as ans:
errCount += 1
self.assertTrue(isinstance(ans, KeyNotFoundException))
r.feed["entries"][0]["link"]="hi2"
try:
r.get_entries()
except Exception as ans:
errCount += 1
self.assertTrue(isinstance(ans, KeyNotFoundException))
self.assertEqual(errCount, 4)
def test_num_entries(self):
for file in TEST_FILE_LIST:
f = "%s/%s" % (TEST_DATA_DIR, file)
r = RSSReader(f)
self.assertEqual(NUM_ENTRIES[file], r.get_num_entries())
