def test_task_CRUD(self):
origin_task = Task()
origin_task.name = 'origin task'
origin_task.status = False
# create
db.session.add(origin_task)
db.session.commit()
# read
tasks = Task.query.filter_by(name=origin_task.name)
task = tasks.first()
self.assertEqual(1, tasks.count())
self.assertEqual(origin_task.status, task.status)
# update
task.status = True
db.session.commit()
tasks = Task.query.filter_by(name=origin_task.name)
task = tasks.first()
self.assertEqual(True, task.status)
# delete
db.session.delete(task)
db.session.commit()
tasks = Task.query.filter_by(name=task.name)
self.assertEqual(0, tasks.count())
def setUp(self):
db.create_all()
# example task
t = Task()
t.name = 'test added task'
db.session.add(t)
db.session.commit()
self.task = Task.query.filter_by(name=t.name).first()
def post(self):
args = parser.parse_args()
task = Task()
task.name = args['name']
try:
db.session.add(task)
db.session.commit()
except exc.SQLAlchemyError:
db.session.rollback()
app.logger.error(self)
return '', 201
def exampleData():
init_db()
tasks = list()
for idx in range(1, 100):
task = Task()
task.name = 'taskname{}'.format(idx)
tasks.append(task)
for task in tasks:
db.session.add(task)
db.session.commit()