__author__ = 'stubbfel'

import unittest
import feedparser
from utilities.feedfile import FeedFile


class FeedUtilitiesTest(unittest.TestCase):
    def setUp(self):
        self.__ff = FeedFile("testfeeds/rss/feed.xml")

    def test_read_feed_file(self):
        feed = self.__ff.read_feed_file()
        self.assertTrue(isinstance(feed, feedparser.FeedParserDict))

    def test_json_feed_file(self):
        feed = self.__ff.read_feed_file()
        ff2 = FeedFile("testfeed.xml")
        serializer = None
        ff2.write_feedjson_file(feed, serializer)
        feed2 = ff2.read_feed_file()
        self.assertTrue(isinstance(feed2, feedparser.FeedParserDict))
        ff2.check_and_remove_file()
        # self.assertDictEqual(feed, feed2)

    def test_write_feed_file_none_feedparser_object(self):
        feed = "Fail"
        ff2 = FeedFile("testfeed.xml")
        serializer = None
        self.assertRaises(AssertionError, ff2.write_feedjson_file, feed, serializer)
        ff2.check_and_remove_file()

if __name__ == '__main__':
    unittest.main()