def setUp(self):
conn = pymysql.Connect(self.HOSTNAME, self.USER)
conn.query('create database %s;' % self.DB_NAME)
conn.close()
uri = 'mysql+pymysql://%s@%s/%s' % (self.USER, self.HOSTNAME, self.DB_NAME)
conf = DummyConfig(uri)
session.initialize(conf)
with session.Session() as sess:
with sess.begin():
tables.create_user(sess,
id=DUMMY_ID,
name=DUMMY_NAME)
with sess.begin():
tables.create_project(sess,
id=DUMMY_ID,
owner_id=DUMMY_ID,
name=DUMMY_NAME,
repository_uri='',
repository_type='')
with sess.begin():
tables.create_file(sess,
id=DUMMY_ID,
project_id=DUMMY_ID,
filepath=DUMMY_FILEPATH)
def test_file(self):
u'''
ファイルテーブルの定義テスト
'''
fpath = 'aaaa'
with session.Session() as sess:
with sess.begin():
f = tables.create_file(sess,
project_id=DUMMY_ID,
filepath=fpath)
results = tables.search_file(sess, project_id=DUMMY_ID,
filepath=fpath)
self.assertEqual(len(results), 1)
# foreign key constraint
with self.assertRaises(exceptions.IntegrityError):
with session.Session() as sess:
with sess.begin():
f = tables.create_file(sess,
project_id=DUMMY_ID*100,
filepath=fpath)
# unique constraint
with self.assertRaises(exceptions.IntegrityError):
with session.Session() as sess:
with sess.begin():
f = tables.create_file(sess,
project_id=DUMMY_ID,
filepath=fpath)