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)
def test_create_db(self): script_path = joinpath( dirname(realpath(__file__)).rsplit(os.sep, 1)[0], 'sql/create_db.sql') dbmgr = DBManager(self.db_name) dbmgr.create_db(script_path) select_query = "SELECT count(*) FROM " self.assertEqual(dbmgr.query(select_query + "stream")[0][0], 0) self.assertEqual(dbmgr.query(select_query + "user")[0][0], 0) self.assertEqual(dbmgr.query(select_query + "device")[0][0], 0) self.assertEqual(dbmgr.query(select_query + "like")[0][0], 0) self.assertEqual(dbmgr.query(select_query + "comment")[0][0], 0) self.assertEqual(dbmgr.query(select_query + "job_period")[0][0], 0) self.assertEqual(dbmgr.query(select_query + "photo")[0][0], 0) self.assertEqual(dbmgr.query(select_query + "access")[0][0], 0)