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