def test_unique(self):
# A job can not be created without a jobtype, create one first
jobtype = JobType()
jobtype.name = "foo"
jobtype.description = "this is a job type"
jobtype.classname = "Foobar"
jobtype.code = dedent("""
class Foobar(JobType):
pass""").encode("utf-8")
db.session.add(jobtype)
job = Job()
job.job_type = jobtype
software = Software()
software.software = "foo"
requirementA = JobSoftwareRequirement()
requirementB = JobSoftwareRequirement()
requirementA.job = job
requirementA.software = software
requirementB.job = job
requirementB.software = software
db.session.add_all([job, requirementA, requirementB])
with self.assertRaises(DatabaseError):
db.session.commit()
def test_null(self):
with self.assertRaises(DatabaseError):
model = JobSoftwareRequirement()
db.session.add(model)
db.session.commit()
db.session.remove()
with self.assertRaises(DatabaseError):
software = Software()
software.software = "foo"
requirement = JobSoftwareRequirement()
requirement.software = software
db.session.add(requirement)
db.session.commit()
def test_insert(self):
# A job can not be created without a jobtype, create one first
jobtype = JobType()
jobtype.name = "foo"
jobtype.description = "this is a job type"
jobtype_version = JobTypeVersion()
jobtype_version.jobtype = jobtype
jobtype_version.version = 1
jobtype_version.classname = "Foobar"
jobtype_version.code = ("""
class Foobar(JobType):
pass""").encode("utf-8")
db.session.add(jobtype_version)
queue = JobQueue()
queue.name = "FooQueue"
job = Job()
job.title = "Test Job"
job.jobtype_version = jobtype_version
job.queue = queue
# Software requirement needs a software first
software = Software()
software.software = "foo"
requirement = JobSoftwareRequirement()
requirement.job = job
requirement.software = software
db.session.add(job)
db.session.commit()
job_id = job.id
requirement_id = requirement.id
requirement2 = JobSoftwareRequirement.query.\
filter_by(id=requirement_id).first()
self.assertEqual(requirement.job.id, job_id)
self.assertEqual(requirement2.software.software, "foo")
self.assertEqual(requirement2.min_version, None)
self.assertEqual(requirement2.max_version, None)
