tests.py

import unittest
import reddit, pages

class TestMainWindow(unittest.TestCase):
    
    def setUp(self):
        self.main_window = reddit.MainWindow()
        self.main_window.listings = []
        
    def testLoadStories(self):
        """tests type of loaded stories"""
        self.main_window._MainWindow__load_stories()
        self.assertIsInstance(self.main_window.listings, list)
        
    def testSetStatus(self):
        """should create a status of the word 'firefly'"""
        self.main_window.set_status("firefly")
        self.assertEquals(self.main_window.footer_content.text, "firefly")
    
    def testRefresh(self):
        """should populate listings"""
        self.main_window.refresh()
        self.assertNotEquals(self.main_window.listings, [])
    
    def testSubreddit(self):
        """should switch subreddits"""
        self.main_window.refresh()
        before = self.main_window.listings
        self.main_window.set_subreddit("gaming")
        after = self.main_window.listings
        self.assertNotEquals(before, after)
        self.assertEquals(self.main_window._MainWindow__subreddit, "gaming")

        
class TestStory(unittest.TestCase):

    def setUp(self):
        self.main_window = reddit.MainWindow()

    def testFormatLineSize(self):
        """format_lines should adjust line width"""
        self.main_window.listings[0].original_widget.story.object['title'] = "123456789 " * 10
        lines = self.main_window.listings[0].original_widget.story.format_lines(70)
        self.assertLessEqual( len(lines[0]), 70, lines[0])
        lines = self.main_window.listings[0].original_widget.story.format_lines(40)
        self.assertLessEqual( len(lines[0]), 40, lines[0])
        self.assertLessEqual( len(lines[1]), 40, lines[1])
        
class TestNavigation(unittest.TestCase):
    def testNavigationCreation(self):
        """Navigation.__init__() should set attributes properly"""
        nav = pages.Navigation("abc", 0, ["start"])
        self.assertEquals(nav.next, "abc")
        self.assertEquals(nav.count, 0)
        

class TestDownloadStories(unittest.TestCase):
    
    def testBadSubredditError(self):
        """should raise a BadSubredditError"""
        handler = pages.RedditHandler()
        self.assertRaises(pages.BadSubredditError, handler.download_stories, "qwer345g63")
        self.assertRaises(pages.BadSubredditError, handler.download_stories, "qwer3 45g 63")
        self.assertRaises(pages.BadSubredditError, handler.download_stories, "78b@$@#$@#   @ 42 4 7cs")

    def testReturnList(self):
        """should return a list"""
        handler = pages.RedditHandler()
        self.assertIsInstance(handler.download_stories(None)[0], list)

    def testReturnStories(self):
        """should return a list of stories"""
        handler = pages.RedditHandler()
        for s in handler.download_stories(None, None, None)[0]:
            self.assertIsInstance(s, pages.Story)

class TestLogin(unittest.TestCase):

    def testInvalidLogin(self):
        """login attempt should fail"""
        handler = pages.RedditHandler()
        self.assertFalse(handler.login("gsdfgsdfg", "sdfgsfgsgf"))
             
        
if __name__ == "__main__":
    unittest.main()