def test_GIVEN_one_pending_job_in_the_database_which_has_completed_WHEN_update_dap_client_not_available_THEN_model_run_status_is_not_updated(self):
self.dap_client_factory.get_dap_client = Mock(side_effect=DapClientException("trouble"))
self.create_model_run_ready_for_submit()
model_run_service = ModelRunService()
model_run = model_run_service.get_model_run_being_created_or_default(self.user)
with session_scope(Session) as session:
model_run.change_status(session, constants.MODEL_RUN_STATUS_RUNNING)
self.running_job_client.get_run_model_statuses = Mock(
return_value=[{'id': model_run.id,
'status': constants.MODEL_RUN_STATUS_COMPLETED,
'error_message': ''
}])
self.job_status_updater.update()
model_run = model_run_service.get_model_by_id(self.user, model_run.id)
assert_that(model_run.status.name, is_(constants.MODEL_RUN_STATUS_RUNNING), "model status")
with session_scope() as session:
datasets = session.query(Dataset)\
.filter(Dataset.model_run_id == model_run.id)\
.filter(Dataset.is_input == False)\
.all()
assert_that(len(datasets), is_(0), "Number of output datasets")
datasets = session.query(Dataset)\
.filter(Dataset.model_run_id == model_run.id)\
.filter(Dataset.is_input == True)\
.all()
assert_that(len(datasets), is_(0), "Number of input datasets")
class ModelRunServiceDuplicateTest(TestWithFullModelRun):
def setUp(self):
super(ModelRunServiceDuplicateTest, self).setUp()
self.job_runner_client = JobRunnerClient(config)
self.model_run_service = ModelRunService(job_runner_client=self.job_runner_client)
self.clean_database()
def test_GIVEN_model_doesnot_exist_WHEN_duplicate_THEN_not_found(self):
# Add a user who doesn't have any model runs
with session_scope(Session) as session:
user = User()
user.name = 'Has No Model Runs'
session.add(user)
# Get the users model runs
with self.assertRaises(NoResultFound, msg="Should have thrown a NoResultFound exception"):
model_runs = self.model_run_service.duplicate_run_model(10000, user)
def test_GIVEN_model_belongs_to_someone_else_WHEN_duplicate_THEN_not_found(self):
# Add a user who doesn't have any model runs
other_user = self.login("other_user")
model = self.create_run_model(10, "test", other_user)
with session_scope(Session) as session:
user = User()
user.name = 'Has No Model Runs'
session.add(user)
# Get the users model runs
with self.assertRaises(NoResultFound, msg="Should have thrown a NoResultFound exception"):
self.model_run_service.duplicate_run_model(model.id, user)
def test_GIVEN_model_WHEN_duplicate_THEN_model_is_duplicated(self):
# Add a user who doesn't have any model runs
user = self.login()
model = self.create_run_model(10, "test", user)
model = self.model_run_service.get_model_by_id(user, model.id)
self.job_runner_client.duplicate = Mock()
self.model_run_service.duplicate_run_model(model.id, user)
assert_that(self.job_runner_client.duplicate.called, is_(False), "Job runner not called")
model_run = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
assert_that(model_run.name, is_(model.name + " (Copy)"), "names are the same")
assert_that(model_run.description, is_(model.description), "description are the same")
assert_that(model_run.driving_dataset_id, is_(model.driving_dataset_id), "driving_dataset_id are the same")
assert_that(len(model_run.parameter_values), is_(len(model.parameter_values)), "names are the same")
assert_that(model_run.parameter_values[0].parameter_id, is_(model.parameter_values[0].parameter_id), "first parameter value is for same paramerer")
assert_that(model_run.parameter_values[0].value, is_(model.parameter_values[0].value), "first parameter value is for same value")
def test_GIVEN_model_with_land_cover_actions_WHEN_duplicate_THEN_model_is_duplicated(self):
# Add a user who doesn't have any model runs
user = self.login()
model = self.create_run_model(10, "test", user)
model = self.model_run_service.get_model_by_id(user, model.id)
with session_scope() as session:
region1 = LandCoverRegion()
region2 = LandCoverRegion()
session.add(region1)
session.add(region2)
session.add(LandCoverAction(model_run_id=model.id, order=1, value_id=1, region=region1))
session.add(LandCoverAction(model_run_id=model.id, order=2, value_id=2, region=region2))
self.job_runner_client.duplicate = Mock()
self.model_run_service.duplicate_run_model(model.id, user)
assert_that(self.job_runner_client.duplicate.called, is_(False), "Job runner not called")
model_run = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
with session_scope() as session:
regions = session.query(LandCoverAction).filter(LandCoverAction.model_run_id == model_run.id).all()
assert_that(len(regions), is_(2), "regions")
assert_that(regions[0].value_id, is_(1), "value id")
assert_that(regions[0].order, is_(1), "order id")
assert_that(regions[0].region_id, is_(region1.id), "region id")
assert_that(regions[1].value_id, is_(2), "value id")
assert_that(regions[1].order, is_(2), "order id")
assert_that(regions[1].region_id, is_(region2.id), "region id")
def test_GIVEN_model_belongs_to_someone_else_and_is_published_WHEN_duplicate_THEN_duplicate_model(self):
self.job_runner_client.delete = Mock()
# Add a user who doesn't have any model runs
other_user = self.login("other_user")
model = self.create_run_model(10, "test", other_user, constants.MODEL_RUN_STATUS_PUBLISHED)
user = self.login(access_level=constants.USER_ACCESS_LEVEL_ADMIN)
self.model_run_service.duplicate_run_model(model.id, user)
model_run = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
assert_that(model_run.name, is_(model.name), "model name")
def test_GIVEN_model_belongs_to_someone_else_and_is_public_WHEN_duplicate_THEN_duplicate_model(self):
self.job_runner_client.delete = Mock()
# Add a user who doesn't have any model runs
other_user = self.login("other_user")
model = self.create_run_model(10, "test", other_user, constants.MODEL_RUN_STATUS_PUBLIC)
user = self.login()
self.model_run_service.duplicate_run_model(model.id, user)
model_run = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
assert_that(model_run.name, is_(model.name), "model name")
def test_GIVEN_model_WHEN_duplicate_and_name_is_same_as_owner_THEN_model_is_duplicated_name_is_changed(self):
# Add a user who doesn't have any model runs
user = self.login()
model = self.create_run_model(10, "test", user)
self.create_run_model(10, "test (Copy)", user)
model = self.model_run_service.get_model_by_id(user, model.id)
self.job_runner_client.duplicate = Mock()
self.model_run_service.duplicate_run_model(model.id, user)
assert_that(self.job_runner_client.duplicate.called, is_(False), "Job runner not called")
model_run = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
assert_that(model_run.name, is_(model.name + " (Copy 2)"), "names are the same")
def test_GIVEN_model_run_already_being_created_WHEN_duplicate_THEN_model_is_duplicated_and_created_is_overwritten(self):
# Add a user who doesn't have any model runs
user = self.login()
model_being_created = self.create_run_model(10, "test", user, status=constants.MODEL_RUN_STATUS_CREATED)
model = self.create_run_model(10, "test", user)
model = self.model_run_service.get_model_by_id(user, model.id)
self.job_runner_client.duplicate = Mock()
self.job_runner_client.delete = Mock()
self.model_run_service.duplicate_run_model(model.id, user)
assert_that(self.job_runner_client.duplicate.called, is_(False), "Job runner not called")
model_run = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
assert_that(model_run.name, is_(model.name + " (Copy)"), "names are the same")
with self.assertRaises(NoResultFound):
self.model_run_service.get_model_by_id(user, model_being_created.id)
def test_GIVEN_uploaded_driving_data_model_run_WHEN_duplicate_THEN_uploaded_driving_data_is_duplicated(self):
# Add a user who doesn't have any model runs
user = self.login()
model = self.create_run_model(10, "test", user)
model = self.model_run_service.get_model_by_id(user, model.id)
id_for_user_upload_driving_dataset = DatasetService().get_id_for_user_upload_driving_dataset()
with session_scope() as session:
model = session.query(ModelRun).filter(ModelRun.id == model.id).one()
model.driving_dataset_id = id_for_user_upload_driving_dataset
self.job_runner_client.duplicate_uploaded_driving_data = Mock()
self.job_runner_client.delete = Mock()
self.model_run_service.duplicate_run_model(model.id, user)
assert_that(self.job_runner_client.duplicate_uploaded_driving_data.called, is_(True), "Job runner duplicate called")
model_run = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
assert_that(model_run.name, is_(model.name + " (Copy)"), "names are the same")
def test_GIVEN_uploaded_driving_data_model_run_WHEN_duplicate_and_job_runner_duplicate_fails_THEN_model_is_not_duplicated_exception_is_thrown(self):
# Add a user who doesn't have any model runs
error_message = "error"
user = self.login()
model = self.create_run_model(10, "test", user)
model = self.model_run_service.get_model_by_id(user, model.id)
id_for_user_upload_driving_dataset = DatasetService().get_id_for_user_upload_driving_dataset()
with session_scope() as session:
model = session.query(ModelRun).filter(ModelRun.id == model.id).one()
model.driving_dataset_id = id_for_user_upload_driving_dataset
self.job_runner_client.duplicate_uploaded_driving_data = Mock(side_effect=ServiceException(error_message))
self.job_runner_client.delete = Mock()
with self.assertRaises(ServiceException):
self.model_run_service.duplicate_run_model(model.id, user)
with self.assertRaises(NoResultFound):
self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
class ModelRunServiceDeleteTest(TestWithFullModelRun):
def setUp(self):
super(ModelRunServiceDeleteTest, self).setUp()
self.job_runner_client = JobRunnerClient(config)
self.model_run_service = ModelRunService(job_runner_client=self.job_runner_client)
self.clean_database()
def test_GIVEN_model_doesnot_exist_WHEN_delete_THEN_not_found(self):
# Add a user who doesn't have any model runs
with session_scope(Session) as session:
user = User()
user.name = 'Has No Model Runs'
session.add(user)
# Get the users model runs
with self.assertRaises(NoResultFound, msg="Should have thrown a NoResultFound exception"):
model_runs = self.model_run_service.delete_run_model(10000, user)
def test_GIVEN_model_belongs_to_someone_else_WHEN_delete_THEN_not_found(self):
# Add a user who doesn't have any model runs
other_user = self.login("other_user")
model = self.create_run_model(10, "test", other_user)
with session_scope(Session) as session:
user = User()
user.name = 'Has No Model Runs'
session.add(user)
# Get the users model runs
with self.assertRaises(NoResultFound, msg="Should have thrown a NoResultFound exception"):
self.model_run_service.delete_run_model(model.id, user)
def test_GIVEN_model_WHEN_delete_THEN_model_delete_job_runner_called(self):
# Add a user who doesn't have any model runs
user = self.login()
model = self.create_run_model(10, "test", user)
self.job_runner_client.delete = Mock()
self.model_run_service.delete_run_model(model.id, user)
assert_that(self.job_runner_client.delete.called, is_(True), "Job runner called")
with self.assertRaises(NoResultFound):
self.model_run_service.get_model_by_id(user, model.id)
def test_GIVEN_model_belongs_to_someone_else_and_user_is_an_admin_WHEN_delete_THEN_delete_model(self):
# Add a user who doesn't have any model runs
self.job_runner_client.delete = Mock()
other_user = self.login("other_user")
model = self.create_run_model(10, "test", other_user)
user = self.login(access_level=constants.USER_ACCESS_LEVEL_ADMIN)
self.model_run_service.delete_run_model(model.id, user)
with self.assertRaises(NoResultFound):
self.model_run_service.get_model_by_id(user, model.id)
def test_GIVEN_model_belongs_to_someone_else_and_is_published_and_user_is_an_admin_WHEN_delete_THEN_delete_model(self):
self.job_runner_client.delete = Mock()
# Add a user who doesn't have any model runs
other_user = self.login("other_user")
model = self.create_run_model(10, "test", other_user, constants.MODEL_RUN_STATUS_PUBLISHED)
user = self.login(access_level=constants.USER_ACCESS_LEVEL_ADMIN)
self.model_run_service.delete_run_model(model.id, user)
with self.assertRaises(NoResultFound):
self.model_run_service.get_model_by_id(user, model.id)
def test_GIVEN_model_belongs_to_someone_else_and_is_public_and_user_is_an_admin_WHEN_delete_THEN_delete_model(self):
self.job_runner_client.delete = Mock()
# Add a user who doesn't have any model runs
other_user = self.login("other_user")
model = self.create_run_model(10, "test", other_user, constants.MODEL_RUN_STATUS_PUBLIC)
user = self.login(access_level=constants.USER_ACCESS_LEVEL_ADMIN)
self.model_run_service.delete_run_model(model.id, user)
with self.assertRaises(NoResultFound):
self.model_run_service.get_model_by_id(user, model.id)
def test_GIVEN_model_and_is_public_and_user_is_not_an_admin_WHEN_delete_THEN_raise(self):
self.job_runner_client.delete = Mock()
# Add a user who doesn't have any model runs
user = self.login("user")
model = self.create_run_model(10, "test", user, constants.MODEL_RUN_STATUS_PUBLIC)
assert_that(calling(self.model_run_service.delete_run_model).with_args(model.id, user),
raises(ModelPublished))
def test_GIVEN_full_model_WHEN_delete_THEN_model_is_deleted(self):
user = self.login('')
self.create_model_run_ready_for_submit()
with session_scope(Session) as session:
dataset_count = session.query(Dataset).count()
model = self.model_run_service.get_models_for_user(user)[0]
dataset = Dataset()
dataset.model_run_id = model.id
session.add(dataset)
region_count = session.query(LandCoverAction).count()
session.add(LandCoverAction(model_run_id=model.id))
model_not_to_delete = self.create_run_model(0, "test", user, constants.MODEL_RUN_STATUS_PUBLISHED)
self.job_runner_client.delete = Mock()
self.model_run_service.delete_run_model(model.id, user)
with session_scope(Session) as session:
count = session.query(ModelRun).filter(ModelRun.id == model.id).count()
assert_that(count, is_(0), 'Count(Model)')
with session_scope(Session) as session:
count = session.query(ModelRun).filter(ModelRun.id == model_not_to_delete.id).count()
assert_that(count, is_(1), 'Count(Model)')
with session_scope(Session) as session:
count = session.query(Dataset).count()
assert_that(count, is_(dataset_count), 'Count(Datasets)is same as before creating model')
with session_scope(Session) as session:
count = session.query(LandCoverAction).count()
assert_that(count, is_(region_count), 'Count(LandCoverAction)is same as before creating model')
def test_GIVEN_full_model_WHEN_delete_THEN_parameter_values_are_deleted(self):
user = self.login('')
with session_scope(Session) as session:
pv_count = session.query(ParameterValue).count()
self.create_model_run_ready_for_submit()
with session_scope(Session) as session:
model = self.model_run_service.get_models_for_user(user)[0]
assert_that(session.query(ParameterValue).count(), greater_than(pv_count), "Creating model run adds parameters")
self.job_runner_client.delete = Mock()
self.model_run_service.delete_run_model(model.id, user)
with session_scope(Session) as session:
assert_that(session.query(ParameterValue).count(), is_(pv_count), "Parameter values is same as before model was created")
class ModelRunServiceTest(TestWithFullModelRun):
def setUp(self):
super(ModelRunServiceTest, self).setUp()
self.job_runner_client = JobRunnerClient(config)
self.model_run_service = ModelRunService(job_runner_client=self.job_runner_client)
self.clean_database()
def test_GIVEN_no_model_runs_WHEN_get_model_runs_for_user_THEN_empty_list(self):
# Add a user who doesn't have any model runs
with session_scope(Session) as session:
user = User()
user.name = 'Has No Model Runs'
session.add(user)
# Get the users model runs
model_runs = self.model_run_service.get_models_for_user(user)
assert_that(model_runs, is_([]))
def test_GIVEN_non_existent_user_WHEN_get_model_runs_for_user_THEN_returns_empty_list(self):
user = User()
user.id = -100
user.name = "Doesn't Exist"
model_runs = self.model_run_service.get_models_for_user(user)
assert_that(model_runs, is_([]))
def test_GIVEN_two_users_with_one_model_each_WHEN_get_model_runs_for_user_THEN_returns_correct_model(self):
# Add two users and give them each one model run
with session_scope(Session) as session:
# First add two users
user1 = User()
user1.name = 'user1'
user2 = User()
user2.name = 'user2'
session.add_all([user1, user2])
session.commit()
# Give them each a model
model_run1 = ModelRun()
model_run1.name = "MR1"
model_run1.user_id = user1.id
model_run2 = ModelRun()
model_run2.name = "MR2"
model_run2.user_id = user2.id
session.add_all([model_run1, model_run2])
# Get the users model runs
model_runs = self.model_run_service.get_models_for_user(user1)
assert_that(len(model_runs), is_(1))
assert_that(model_runs[0].name, is_("MR1"))
def test_GIVEN_user_has_published_model_run_WHEN_get_model_runs_for_user_THEN_published_model_is_returned(self):
# Add one user and give them a published and unpublished model run
with session_scope(Session) as session:
# First add user
user1 = User()
user1.name = 'user1'
session.add(user1)
session.commit()
# Give a model
model_run1 = ModelRun()
model_run1.name = "MR1"
model_run1.user_id = user1.id
model_run1.status = self._status(constants.MODEL_RUN_STATUS_CREATED)
model_run2 = ModelRun()
model_run2.name = "MR2"
model_run2.status = self._status(constants.MODEL_RUN_STATUS_PUBLISHED)
model_run2.user_id = user1.id
model_run2.status = self._status(constants.MODEL_RUN_STATUS_COMPLETED)
session.add_all([model_run1, model_run2])
# Get the users model runs
model_runs = self.model_run_service.get_models_for_user(user1)
assert_that(len(model_runs), is_(2))
def test_GIVEN_user_has_public_model_run_WHEN_get_model_runs_for_user_THEN_public_model_is_returned(self):
# Add one user and give them a public and unpublished model run
with session_scope(Session) as session:
# First add user
user1 = User()
user1.name = 'user1'
session.add(user1)
session.commit()
# Give a model
model_run1 = ModelRun()
model_run1.name = "MR1"
model_run1.user_id = user1.id
model_run1.status = self._status(constants.MODEL_RUN_STATUS_CREATED)
model_run2 = ModelRun()
model_run2.name = "MR2"
model_run2.status = self._status(constants.MODEL_RUN_STATUS_PUBLIC)
model_run2.user_id = user1.id
session.add_all([model_run1, model_run2])
# Get the users model runs
model_runs = self.model_run_service.get_models_for_user(user1)
assert_that(len(model_runs), is_(2))
def test_GIVEN_user_has_published_model_run_WHEN_get_published_model_runs_THEN_only_published_model_run_returned(self):
# Add one user and give them a published and unpublished model run
with session_scope(Session) as session:
# First add user
user1 = User()
user1.name = 'user1'
session.add(user1)
session.commit()
# Give them each a model
model_run1 = ModelRun()
model_run1.name = "MR1"
model_run1.user_id = user1.id
model_run2 = ModelRun()
model_run2.name = "MR2"
model_run2.status = self._status(constants.MODEL_RUN_STATUS_PUBLISHED)
model_run2.user_id = user1.id
session.add_all([model_run1, model_run2])
# Get the published model runs
model_runs = self.model_run_service.get_published_models()
assert_that(len(model_runs), is_(1))
assert_that(model_runs[0].name, is_("MR2"))
def test_GIVEN_user_has_public_model_run_WHEN_get_published_model_runs_THEN_only_public_model_run_returned(self):
# Add one user and give them a published and unpublished model run
with session_scope(Session) as session:
# First add user
user1 = User()
user1.name = 'user1'
session.add(user1)
session.commit()
# Give them each a model
model_run1 = ModelRun()
model_run1.name = "MR1"
model_run1.user_id = user1.id
model_run2 = ModelRun()
model_run2.name = "MR2"
model_run2.status = self._status(constants.MODEL_RUN_STATUS_PUBLIC)
model_run2.user_id = user1.id
session.add_all([model_run1, model_run2])
# Get the published model runs
model_runs = self.model_run_service.get_published_models()
assert_that(len(model_runs), is_(1))
assert_that(model_runs[0].name, is_("MR2"))
def test_GIVEN_no_published_or_public_runs_WHEN_get_published_model_runs_THEN_empty_list(self):
model_runs = self.model_run_service.get_published_models()
assert_that(model_runs, is_([]))
def test_WHEN_get_user_model_runs_THEN_list_ordered_by_date_created_newest_first(self):
with session_scope(Session) as session:
# First add user
user1 = User()
user1.name = 'user1'
session.add(user1)
session.commit()
# Create three published models not in datetime order
model_run1 = ModelRun()
model_run1.user_id = user1.id
model_run1.name = "MR1"
model_run1.date_created = datetime(2013, 12, 7, 17, 15, 30)
session.add(model_run1)
session.commit()
model_run2 = ModelRun()
model_run2.user_id = user1.id
model_run2.name = "MR2"
model_run2.date_created = datetime(2014, 6, 9, 12, 30, 24)
session.add(model_run2)
session.commit()
model_run3 = ModelRun()
model_run3.user_id = user1.id
model_run3.name = "MR3"
model_run3.date_created = datetime(2014, 6, 9, 11, 39, 30)
session.add_all([model_run1, model_run2, model_run3])
model_runs = self.model_run_service.get_models_for_user(user1)
assert_that(model_runs[0].name, is_("MR2"))
assert_that(model_runs[1].name, is_("MR3"))
assert_that(model_runs[2].name, is_("MR1"))
def test_WHEN_get_published_model_runs_THEN_list_ordered_by_date_created_newest_first(self):
with session_scope(Session) as session:
# Create three published models not in datetime order
model_run1 = ModelRun()
model_run1.name = "MR1"
model_run1.date_created = datetime(2013, 12, 7, 17, 15, 30)
model_run1.status = self._status(constants.MODEL_RUN_STATUS_PUBLISHED)
session.add(model_run1)
session.commit()
model_run2 = ModelRun()
model_run2.name = "MR2"
model_run2.date_created = datetime(2014, 6, 9, 12, 30, 24)
model_run2.status = self._status(constants.MODEL_RUN_STATUS_PUBLIC)
session.add(model_run2)
session.commit()
model_run3 = ModelRun()
model_run3.name = "MR3"
model_run3.status = self._status(constants.MODEL_RUN_STATUS_PUBLISHED)
model_run3.date_created = datetime(2014, 6, 9, 11, 39, 30)
session.add(model_run3)
session.commit()
model_runs = self.model_run_service.get_published_models()
assert_that(model_runs[0].name, is_("MR2"))
assert_that(model_runs[1].name, is_("MR3"))
assert_that(model_runs[2].name, is_("MR1"))
def test_WHEN_get_code_versions_THEN_returns_list_including_default_code_version(self):
models = self.model_run_service.get_code_versions()
assert_that(len(models), is_not(0), "There should be at least one code version on the list")
assert_that([x.name for x in models], has_item(config['default_code_version']), "The list of code versions")
def test_GIVEN_valid_code_version_id_WHEN_get_code_version_THEN_code_version_returned(self):
expectedModel = self.model_run_service.get_code_versions()[0]
model = self.model_run_service.get_code_version_by_id(expectedModel.id)
assert_that(model.id, is_(expectedModel.id), "Id")
assert_that(model.name, is_(expectedModel.name), "Name")
def test_GIVEN_non_existent_code_version_id_WHEN_get_code_version_THEN_raises_NoResultFound_exception(self):
with self.assertRaises(NoResultFound, msg="Should have thrown a NoResultFound exception"):
model = self.model_run_service.get_code_version_by_id(-100)
def test_GIVEN_user_has_model_run_WHEN_get_model_run_by_id_THEN_model_run_returned(self):
# Add a user and give them a model
with session_scope(Session) as session:
# First add user
user = User()
user.name = 'user1'
session.add(user)
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_PUBLIC)
session.add(model_run)
# Get the users model runs
model_run_returned = self.model_run_service.get_model_by_id(user, model_run.id)
assert_that(model_run_returned.name, is_("MR1"))
def test_GIVEN_model_run_has_parameters_WHEN_get_model_run_by_id_THEN_model_run_has_parameter_values_loaded(self):
# Add a user and give them a model
user = self.login()
model_run = self.create_run_model(0, "MR1", user)
with session_scope(Session) as session:
# Add parameter value
parameter_value = ParameterValue()
parameter_value.parameter_id = 1
parameter_value.set_value_from_python(123)
parameter = Parameter()
parameter.name = "Param"
parameter.parameter_values = [parameter_value]
# Give them a model
model_run = session.query(ModelRun).filter(ModelRun.id == model_run.id).one()
model_run.parameter_values = [parameter_value]
session.add(model_run)
with session_scope(Session) as session:
model_run_id = session.query(ModelRun).filter(ModelRun.name == model_run.name).one().id
# Get the users model runs
model_run_returned = self.model_run_service.get_model_by_id(user, model_run_id)
pv = model_run_returned.parameter_values[0]
assert_that(pv.value, is_('123'))
assert_that(pv.parameter.name, is_("Param"))
def test_GIVEN_model_run_id_belongs_to_another_user_WHEN_get_model_run_by_id_THEN_NoResultFound_exception(self):
# Add two users and give one a model
with session_scope(Session) as session:
user = User()
user.name = 'user1'
user2 = User()
user2.name = 'user2'
session.add_all([user, user2])
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_COMPLETED)
session.add(model_run)
# Get the users model runs
with self.assertRaises(NoResultFound, msg="Should have thrown a NoResultFound exception"):
self.model_run_service.get_model_by_id(user2, model_run.id)
def test_GIVEN_no_defining_model_run_WHEN_get_defining_model_run_THEN_error_returned(self):
user = self.login()
with self.assertRaises(NoResultFound, msg="Should have thrown a NoResultFound exception"):self.model_run_service.get_parameters_for_model_being_created(user)
def test_GIVEN_incomplete_model_WHEN_publish_model_THEN_ServiceException_raised(self):
# Add a user and give them a model
with session_scope(Session) as session:
user = User()
user.name = 'user1'
session.add(user)
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_FAILED)
session.add(model_run)
# Get the users model runs
with self.assertRaises(ServiceException, msg="Should have raised a ServiceException"):
self.model_run_service.publish_model(user, model_run.id)
def test_GIVEN_model_belongs_to_another_user_WHEN_publish_model_THEN_ServiceException_raised(self):
# Add two users and give one a model
with session_scope(Session) as session:
user = User()
user.name = 'user1'
user2 = User()
user2.name = 'user2'
session.add_all([user, user2])
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_COMPLETED)
session.add(model_run)
# Get the users model runs
with self.assertRaises(ServiceException, msg="Should have raised a ServiceException"):
self.model_run_service.publish_model(user2, model_run.id)
def test_GIVEN_nonexistent_model_id_WHEN_publish_model_THEN_ServiceException_raised(self):
# Add a user
with session_scope(Session) as session:
user = User()
user.name = 'user1'
session.add(user)
with self.assertRaises(ServiceException, msg="Should have raised a ServiceException"):
self.model_run_service.publish_model(user, -100)
def test_GIVEN_user_has_completed_model_WHEN_publish_model_THEN_model_published(self):
# Add a user and give them a model
with session_scope(Session) as session:
user = User()
user.name = 'user1'
session.add(user)
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_COMPLETED)
session.add(model_run)
# Get the users model runs
self.model_run_service.publish_model(user, model_run.id)
with session_scope(Session) as session:
updated_model_run = session.query(ModelRun).join(ModelRunStatus).filter(ModelRun.id == model_run.id).one()
assert_that(updated_model_run.status.name, is_(constants.MODEL_RUN_STATUS_PUBLISHED))
def test_GIVEN_complete_model_WHEN_make_public_model_THEN_ServiceException_raised(self):
# Add a user and give them a model
with session_scope(Session) as session:
user = User()
user.name = 'user1'
session.add(user)
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_COMPLETED)
session.add(model_run)
# Get the users model runs
with self.assertRaises(ServiceException, msg="Should have raised a ServiceException"):
self.model_run_service.make_public_model(user, model_run.id)
def test_GIVEN_model_belongs_to_another_user_WHEN_make_public_model_THEN_ServiceException_raised(self):
# Add two users and give one a model
with session_scope(Session) as session:
user = User()
user.name = 'user1'
user2 = User()
user2.name = 'user2'
session.add_all([user, user2])
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_PUBLISHED)
session.add(model_run)
# Get the users model runs
with self.assertRaises(ServiceException, msg="Should have raised a ServiceException"):
self.model_run_service.make_public_model(user2, model_run.id)
def test_GIVEN_nonexistent_model_id_WHEN_make_public_model_THEN_ServiceException_raised(self):
# Add a user
with session_scope(Session) as session:
user = User()
user.name = 'user1'
session.add(user)
with self.assertRaises(ServiceException, msg="Should have raised a ServiceException"):
self.model_run_service.make_public_model(user, -100)
def test_GIVEN_user_has_published_model_WHEN_make_public_model_THEN_model_public(self):
# Add a user and give them a model
with session_scope(Session) as session:
user = User()
user.name = 'user1'
session.add(user)
session.commit()
# Give them a model
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_PUBLISHED)
session.add(model_run)
# Get the users model runs
self.model_run_service.make_public_model(user, model_run.id)
with session_scope(Session) as session:
updated_model_run = session.query(ModelRun).join(ModelRunStatus).filter(ModelRun.id == model_run.id).one()
assert_that(updated_model_run.status.name, is_(constants.MODEL_RUN_STATUS_PUBLIC))
def test_GIVEN_default_science_configurations_THEN_all_configurations_returned(self):
results = self.model_run_service.get_scientific_configurations()
assert_that(len(results), greater_than(0), "at least one default configuration")
assert_that(results[0]['id'], not_none(), "Configuration has an id")
assert_that(results[0]['name'], not_none(), "Configuration has a name")
assert_that(results[0]['description'], not_none(), "Configuration has a description")
def test_GIVEN_model_with_parameters_WHEN_get_parameter_value_THEN_parameter_value_returned(self):
with session_scope(Session) as session:
user = User()
user.name = 'user1'
session.add(user)
session.commit()
param = session.query(Parameter).first()
param_id = param.id
param_name = param.name
param_namelist = param.namelist.name
model_run = ModelRun()
model_run.name = "MR1"
model_run.user_id = user.id
model_run.status = self._status(constants.MODEL_RUN_STATUS_CREATED)
session.add(model_run)
session.commit()
parameter1 = ParameterValue()
parameter1.parameter_id = param_id
parameter1.set_value_from_python("param1 value")
parameter1.model_run_id = model_run.id
session.add(parameter1)
model_run_returned = self.model_run_service.get_model_being_created_with_non_default_parameter_values(user)
param_value_returned = model_run_returned.get_python_parameter_value([param_namelist, param_name])
assert_that(param_value_returned, is_("param1 value"))
def test_GIVEN_no_run_model_WHEN_create_run_model_with_science_config_THEN_model_created_with_parameter_values_copied(self):
user = self.login()
expected_name = "model run name"
expected_description = "some slightly long winded description"
self.model_run_service.update_model_run(
user,
expected_name,
constants.DEFAULT_SCIENCE_CONFIGURATION,
expected_description)
parameter_values_count = self.count_parameter_values_in_model_being_created(user)
assert_that(parameter_values_count, is_not(0), "parameter values have been set")
def test_GIVEN_run_model_WHEN_create_run_model_with_same_science_config_THEN_model_updated_with_new_parameter_values_copied(self):
user = self.login()
expected_name = "model run name"
expected_description = "some slightly long winded description"
self.model_run_service.update_model_run(
user,
expected_name,
constants.DEFAULT_SCIENCE_CONFIGURATION,
expected_description)
expected_parameters_count = self.count_parameter_values_in_model_being_created(user)
self.model_run_service.update_model_run(
user,
expected_name,
constants.DEFAULT_SCIENCE_CONFIGURATION,
expected_description)
parameter_values_count = self.count_parameter_values_in_model_being_created(user)
assert_that(parameter_values_count, is_(expected_parameters_count), "parameter values have been set and old ones removed")
def test_GIVEN_run_model_with_time_extent_WHEN_create_run_model_with_same_science_config_THEN_model_updated_with_new_parameter_values_copied_and_has_time_extent(self):
user = self.login()
expected_name = "model run name"
expected_description = "some slightly long winded description"
self.model_run_service.update_model_run(
user,
expected_name,
constants.DEFAULT_SCIENCE_CONFIGURATION,
expected_description)
self.model_run_service.save_parameter(constants.JULES_PARAM_LATLON_REGION, True, user)
expected_parameters_count = self.count_parameter_values_in_model_being_created(user)
self.model_run_service.update_model_run(
user,
expected_name,
constants.DEFAULT_SCIENCE_CONFIGURATION,
expected_description)
parameter_values_count = self.count_parameter_values_in_model_being_created(user)
assert_that(parameter_values_count, is_(expected_parameters_count), "parameter values have been set and old ones removed")
def count_parameter_values_in_model_being_created(self, user):
"""
Count the number of parameter values that this model has
"""
with session_scope(Session) as session:
parameter_values = session \
.query(ParameterValue) \
.join(ModelRun) \
.join(ModelRunStatus) \
.filter(ModelRunStatus.name == constants.MODEL_RUN_STATUS_CREATED) \
.filter(ModelRun.user == user) \
.count()
return parameter_values
