adding a fixture for unit test
@@ -229,3 +229,34 @@ def composite_resource_with_mi_aggregation(composite_resource):
# set file to model instance aggregation type
ModelInstanceLogicalFile.set_file_type(res, user, res_file.id)
yield res, user
+
+@pytest.mark.django_db
+@pytest.fixture(scope="function")
+def composite_resource_with_mi_mp_aggregation(composite_resource):
+ res, user = composite_resource
+ # create model instance aggregation
+ file_path = 'pytest/assets/generic_file.txt'
+ upload_folder = None
+ file_to_upload = UploadedFile(file=open(file_path, 'rb'),
+ name=os.path.basename(file_path))
+
+ res_file = add_file_to_resource(
+ res, file_to_upload, folder=upload_folder, check_target_folder=True
+ )
+
+ # set file to model instance aggregation type
+ ModelInstanceLogicalFile.set_file_type(res, user, res_file.id)
+
+ # create model program aggregation
+ file_path = 'pytest/assets/logan.vrt'
+ upload_folder = None
+ file_to_upload = UploadedFile(file=open(file_path, 'rb'),
+ name=os.path.basename(file_path))
+
+ res_file = add_file_to_resource(
+ res, file_to_upload, folder=upload_folder, check_target_folder=True
+ )
+
+ # set file to model program aggregation type
+ ModelProgramLogicalFile.set_file_type(res, user, res_file.id)
+ yield res, user
