def test_save_to_database(self):
user = User("*****@*****.**", "abc123", "usertwo")
user.save_to_database()
job = Job("Server")
job.save_to_database()
user_job = UserJob(user, job)
user_job.save_to_database()
def test_remove_by_job_name(self):
job_name = 'manager'
job = Job(job_name)
job.save_to_database()
Job.remove_by_job_name(job_name)
job_name_search = Job.query.get(job.id)
self.assertIsNone(job_name_search)
def test_remove_job_name_by_id(self):
job_name = 'kitchen'
job = Job(job_name)
job.save_to_database()
Job.remove_by_id(job.id)
job_name_search = Job.query.get(job.id)
self.assertIsNone(job_name_search)
def test_create_only_unique_jobs(self):
job_name1 = 'bartender'
job_name2 = 'bartender'
job_name_1 = Job(job_name1)
job_name_1.save_to_database()
job_name_2 = Job(job_name2)
# this should not save to database
result = job_name_2.save_to_database()
self.assertFalse(result)
db.session.delete(job_name_1)
db.session.commit()
def test_remove_job_name_by_id_that_does_not_exist(self):
job_name = 'bank'
job = Job(job_name)
job.save_to_database()
result = Job.remove_by_id(-1)
self.assertFalse(result)
Job.remove_by_id(job.id)
def test_rename_a_job(self):
job_name = 'cook'
job = Job(job_name)
job.save_to_database()
new_name = 'chef'
Job.rename(job.id, new_name)
job_with_new_name = Job.query.get(job.id)
self.assertEqual(job.id, job_with_new_name.id)
self.assertEqual(new_name, job_with_new_name.name)
Job.remove_by_id(job.id)
def test_rename_to_non_unique_name_return_false(self):
job_name = 'bar manager'
job = Job(job_name)
job.save_to_database()
new_name = 'bar manager'
result = Job.rename(job.id, new_name)
self.assertFalse(result)
Job.remove_by_id(job.id)
def test_remove_by_non_existing_job_name_returns_false(self):
result = Job.remove_by_job_name('fake_job')
self.assertFalse(result)
