def test_get_valid_ref_by_instance_name(self):
# Valid values
assert get_valid_ref(model=Experiment,
instance=self.instance_mock,
entity_args=[self.project_experiment.name
])[0] == self.project_experiment.id
assert get_valid_ref(model=Job,
instance=self.instance_mock,
entity_args=[self.project_job.name
])[0] == self.project_job.id
# Non valid project and user
assert get_valid_ref(model=Experiment,
instance=self.wrong_instance_mock,
entity_args=[self.project_experiment.name
]).count() == 0
assert get_valid_ref(model=Job,
instance=self.wrong_instance_mock,
entity_args=[self.project_job.name]).count() == 0
# Non valid values
assert get_valid_ref(model=Experiment,
instance=self.instance_mock,
entity_args=[self.experiment.name]).count() == 0
assert get_valid_ref(model=Job,
instance=self.instance_mock,
entity_args=[self.job.name]).count() == 0
def test_get_valid_ref_by_id(self):
assert get_valid_ref(
model=Experiment,
entity_id=self.experiment.id)[0] == self.experiment.id
assert get_valid_ref(model=Experiment,
entity_id=self.project_experiment.id
)[0] == self.project_experiment.id
assert get_valid_ref(model=Job,
entity_id=self.job.id)[0] == self.job.id
assert get_valid_ref(
model=Job, entity_id=self.project_job.id)[0] == self.project_job.id
assert get_valid_ref(model=Experiment, entity_id=-1).count() == 0
assert get_valid_ref(model=Job, entity_id=-1).count() == 0
def test_get_valid_ref_by_instance_and_project_name_user(self):
# Valid values
assert get_valid_ref(model=Experiment,
instance=self.instance_mock,
entity_args=[
self.user.username, self.project.name,
self.project_experiment.name
])[0] == self.project_experiment.id
assert get_valid_ref(model=Job,
instance=self.instance_mock,
entity_args=[
self.user.username, self.project.name,
self.project_job.name
])[0] == self.project_job.id
# Non valid project and user and all information should pass this time
assert get_valid_ref(model=Experiment,
instance=self.wrong_instance_mock,
entity_args=[
self.user.username, self.project.name,
self.project_experiment.name
]).count() == 1
assert get_valid_ref(model=Job,
instance=self.wrong_instance_mock,
entity_args=[
self.user.username, self.project.name,
self.project_job.name
]).count() == 1
# Non valid values
assert get_valid_ref(model=Experiment,
instance=self.instance_mock,
entity_args=[
self.user.username, self.project.name,
self.experiment.name
]).count() == 0
assert get_valid_ref(
model=Job,
instance=self.instance_mock,
entity_args=[self.user.username, self.project.name,
self.job.name]).count() == 0
