Example #1
0
 def test_save_source(self):
     '''Testcase method to test Source if object is saved'''
     source2 = Source(
         'abc-1', 'ABC News',
         'ABC News is all here to enable you go through the world news updates.'
     )
     source2.save_source(self)
     self.assertEqual(len(Source.all_sources), 1)
Example #2
0
class SourceTest(unittest.TestCase):
    """
    test case for source class
    """
    def setUp(self):
        """
        runs before each testcase
        """
        self.new_source = Source('abc-news', 'ABC News', 'Trusted source for breaking news', 'us', 'https://abcnews.go.com')

    def test_instance(self):
        self.assertTrue(isinstance(self.new_source, Source))

    def test_save_source(self):
        self.new_source.save_source()
        self.assertEqual(len(Source.all_sources), 1)