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)
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)