def test_put_note(self):
with app.app_context():
try:
data = {'new_note_name': 'new_note_name'}
resp = self.app.put('/api/notes/1',
data=data,
content_type='multipart/form-data')
data = json.loads(resp.data)
self.assertEqual(data['result'], True)
self.assertEqual(resp.status_code, 200)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_select_clauses_by_note_id(self):
with app.app_context():
try:
result = sqlalchemy_orm.get_item_from_db(
Clauses, {'note_id': 1})
self.assertIsNotNone(result)
self.assertIsInstance(result, list)
self.assertNotEqual(len(result), 0)
self.assertEqual(result[0]['note_id'], 1)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_update_note_name(self):
with app.app_context():
try:
sqlalchemy_orm.update_data_to_db(Notes, {'note_id' : 1}, {'note_name' : '새로운 노트 이름'})
result = sqlalchemy_orm.get_item_from_db(Notes, {'note_id' : 1})
result = result[0]
self.assertEqual(result['note_name'], '새로운 노트 이름')
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_update_clauses(self):
with app.app_context():
try:
sqlalchemy_orm.update_data_to_db(Clauses, {'note_id' : 1, 'clause_id' : 2}, {'title': '바뀐 제목', 'contents' : "바뀐 내용"})
result = sqlalchemy_orm.get_item_from_db(Clauses, {'note_id' : 1, 'clause_id':2})
result = result[0]
self.assertEqual(result['title'], '바뀐 제목')
self.assertEqual(result['contents'], '바뀐 내용')
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_delete_db(self):
with app.app_context():
try:
db = sqlalchemy_orm.drop_db(False)
table_names = db.engine.table_names()
self.assertEqual(table_names, [])
self.assertEqual(len(table_names), 0)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_update_clausePoints(self):
with app.app_context():
try:
sqlalchemy_orm.update_data_to_db(ClausePoints, {'clause_id' : 3}, {'imp' : 1, 'und' : 2})
result = sqlalchemy_orm.get_item_from_db(ClausePoints, {'clause_id' : 3})
result = result[0]
self.assertEqual(result['imp'], 1)
self.assertEqual(result['und'], 2)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_create_db(self):
with app.app_context():
try:
db = sqlalchemy_orm.create_db(False)
table_names = db.engine.table_names()
self.assertIn('notes', table_names)
self.assertIn('clauses', table_names)
self.assertIn('clause_points', table_names)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_delete_multiple_note(self):
with app.app_context():
try:
sqlalchemy_orm.delete_data_from_db(Notes, {'note_id': [1, 2]})
# Check if result is empty
note_result = sqlalchemy_orm.get_item_from_db(
Notes, {'note_id': [1, 2]})
self.assertEqual(note_result, [])
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_delete_clause(self):
with app.app_context():
try:
sqlalchemy_orm.delete_data_from_db(Clauses, {'clause_id': 11})
clause_result = sqlalchemy_orm.get_item_from_db(
Clauses, {'clause_id': 11})
clause_point_result = sqlalchemy_orm.get_item_from_db(
ClausePoints, {'clause_id': 11})
# Check if result is empty
self.assertEqual(clause_result, [])
self.assertEqual(clause_point_result, [])
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_delete_clause(self):
with app.app_context():
try:
resp = self.app.delete('/api/clauses/2')
data = json.loads(resp.data)
self.assertEqual(resp.status_code, 200)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_post_clause(self):
with app.app_context():
try:
data = {
'note_id': 1,
'title': '특별한 날',
'contents': '특별한 날에 대한 설명'
}
resp = self.app.post('/api/clauses',
data=data,
content_type='multipart/form-data')
data = json.loads(resp.data)
self.assertIsNotNone(data['clause_id'])
self.assertEqual(resp.status_code, 200)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_insert_clausePoints_db(self):
with app.app_context():
try:
clausePoints = sqlalchemy_orm.ClausePoints
# column = ['clause_id', 'note_id', 'imp', 'und']
clausepoints_id = sqlalchemy_orm.insert_data_to_db(
"ClausePoints", clausePoints(1, 1))
self.assertIsNotNone(clausepoints_id)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_select_all_notes(self):
with app.app_context():
try:
result = None
result = sqlalchemy_orm.get_all_data_from_db(Notes)
self.assertIsNotNone(result)
self.assertIsInstance(result, list)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_insert_note_db(self):
with app.app_context():
try:
Notes = sqlalchemy_orm.Notes
note_id = sqlalchemy_orm.insert_data_to_db(
'Notes', Notes('test_title_1'))
self.assertIsNotNone(note_id)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_insert_clause_db(self):
with app.app_context():
try:
Clauses = sqlalchemy_orm.Clauses
clauses_id = sqlalchemy_orm.insert_data_to_db(
'Clauses', Clauses(1, "test_clause_1", "test_content_1"))
self.assertIsNotNone(clauses_id)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_post_note(self):
with app.app_context():
try:
data = {
'studi_material':
open('../studi/uploads/studi_test_file.csv', 'rb')
}
resp = self.app.post('/api/notes',
data=data,
content_type='multipart/form-data')
data = json.loads(resp.data)
self.assertEqual(data['result'], True)
self.assertEqual(resp.status_code, 200)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_get_notes(self):
with app.app_context():
try:
resp = self.app.get('/api/notes')
data = json.loads(resp.data)
self.assertIsInstance(data['notes'], list)
self.assertNotEqual(len(data['notes']), 0)
self.assertEqual(resp.status_code, 200)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_put_clausepoint(self):
with app.app_context():
try:
data = {'imp': 1, 'und': 1}
resp = self.app.put('/api/clausePoints/1',
data=data,
content_type='multipart/form-data')
data = json.loads(resp.data)
self.assertEqual(resp.status_code, 200)
self.assertIsNotNone(data['clause_point'])
self.assertNotEqual(len(data['clause_point']), 0)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_get_clausepoints(self):
with app.app_context():
try:
# update for testing
sqlalchemy_orm.update_data_to_db(sqlalchemy_orm.ClausePoints,
{'clause_id': 7}, {
'imp': 0,
'und': 1
})
sqlalchemy_orm.update_data_to_db(sqlalchemy_orm.ClausePoints,
{'clause_id': 6}, {
'imp': 1,
'und': 1
})
sqlalchemy_orm.update_data_to_db(sqlalchemy_orm.ClausePoints,
{'clause_id': 5}, {
'imp': 1,
'und': 1
})
params = {'imp': 1, 'und': 1}
resp = self.app.get('/api/notes/2/clausePoints',
query_string=params)
data = json.loads(resp.data)
self.assertEqual(resp.status_code, 200)
self.assertIsNotNone(data['clause_points'])
self.assertNotEqual(len(data['clause_points']), 0)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
def test_insert_csv_to_db(self):
with app.app_context():
if os.path.exists('../studi/uploads/studi_test_files.csv'):
try:
note_id = upload.insert_csv_to_db()
self.assertIsNotNone(note_id)
except sqlalchemy.exc.SQLAlchemyError:
error_message = util.traceback_custom_error()
error = custom_error.SQLAlchemyError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except AssertionError:
error_message = util.traceback_custom_error()
error = custom_error.AssertionError(error_message)
error.set_logger(logger)
error.to_dict()
raise
except:
error_message = util.traceback_custom_error()
error = custom_error.UnExpectedError(error_message)
error.set_logger(logger)
error.to_dict()
raise
