def test_save_employee(app):
with app.app_context():
employee = EmployeeModel(first_name='test',
second_name='test2',
username='******')
employee.save_employee()
employee_db = EmployeeModel.find_by_id(employee.id)
assert employee_db is not None
def post(self):
"""Add new employee"""
data = parser_required.parse_args()
username = data['username']
if EmployeeModel.find_by_username(username) is not None:
api.abort(400, f'Employee {username} already exist')
employee = EmployeeModel(**data)
try:
employee.save_employee()
except:
api.abort(500, 'Failed to add new employee')
return employee
