def test_getTwInfo(self):
expected = (True, [1, 'user', 'key', 'secret', 'access_token', 'access_token_secret'])
try:
with self.conn.cursor() as cursor:
ddl_file = path.dirname(path.abspath(__file__)) + sep + pardir + sep + constants.CREATE_TW_USER_TABLE_DDL
fin = open(ddl_file)
ddl = fin.read()
cursor.execute(ddl)
cursor.execute(
'''INSERT INTO twitter_users
(
user,
consumer_key,
consumer_secret,
access_token,
access_token_secret
)
VALUES(
'user',
'key',
'secret',
'access_token',
'access_token_secret')''')
self.conn.commit()
finally:
if not fin.closed:
fin.close()
actual = dbUtil.getTwInfo(self.conn, 1)
eq_(expected, actual)
def test_getTwInfo_not_exist(self):
expected = False
try:
with self.conn.cursor() as cursor:
ddl_file = path.dirname(path.abspath(__file__)) + sep + pardir + sep + constants.CREATE_TW_USER_TABLE_DDL
fin = open(ddl_file)
ddl = fin.read()
cursor.execute(ddl)
finally:
if not fin.closed:
fin.close()
constants.SELECT_USER_INFO_SQL = 'sql/selectUserInfo_not_exist.sql'
(result, twInfo) = dbUtil.getTwInfo(self.conn, 99)
actual = result
eq_(expected, actual)
