def test_add_job_period(self): FB.__init__ = lambda x : None dbmgr = DBManager(self.db_name) dbmgr.connect() fb = FB() fb._db = dbmgr fb.add_job_period() self.assertEqual(dbmgr.query("SELECT count(*) FROM job_period")[0][0], 20) fb.add_job_period() self.assertEqual(dbmgr.query("SELECT count(*) FROM job_period")[0][0], 20)
def test_add_job_period(self): FB.__init__ = lambda x: None dbmgr = DBManager(self.db_name) dbmgr.connect() fb = FB() fb._db = dbmgr fb.add_job_period() self.assertEqual( dbmgr.query("SELECT count(*) FROM job_period")[0][0], 20) fb.add_job_period() self.assertEqual( dbmgr.query("SELECT count(*) FROM job_period")[0][0], 20)