def create_trestle_project_with_model(
top_dir: pathlib.Path, model_obj: OscalBaseModel, model_name: str, monkeypatch: MonkeyPatch
) -> pathlib.Path:
"""Create initialized trestle project and import the model into it."""
cur_dir = pathlib.Path.cwd()
# create subdirectory for trestle project
trestle_root = top_dir / 'my_trestle'
trestle_root.mkdir()
os.chdir(trestle_root)
try:
testargs = ['trestle', 'init']
monkeypatch.setattr(sys, 'argv', testargs)
assert Trestle().run() == 0
# place model object in top directory outside trestle project
# so it can be imported
tmp_model_path = top_dir / (model_name + '.json')
model_obj.oscal_write(tmp_model_path)
i = ImportCmd()
args = argparse.Namespace(
trestle_root=trestle_root, file=str(tmp_model_path), output=model_name, verbose=0, regenerate=False
)
assert i._run(args) == 0
except Exception as e:
raise TrestleError(f'Error creating trestle project with model: {e}')
finally:
os.chdir(cur_dir)
return trestle_root
def prepare_trestle_project_dir(
repo_dir: pathlib.Path, content_type: FileContentType, model_obj: OscalBaseModel, models_dir_name: str
):
"""Prepare a temp directory with an example OSCAL model."""
ensure_trestle_config_dir(repo_dir)
model_alias = str_utils.classname_to_alias(model_obj.__class__.__name__, AliasMode.JSON)
file_ext = FileContentType.to_file_extension(content_type)
models_full_path = repo_dir / models_dir_name / 'my_test_model'
model_def_file = models_full_path / f'{model_alias}{file_ext}'
models_full_path.mkdir(exist_ok=True, parents=True)
model_obj.oscal_write(model_def_file)
return models_full_path, model_def_file
def prepare_trestle_project_dir(
tmp_dir, content_type: FileContentType, model_obj: OscalBaseModel, models_dir_name: str
):
"""Prepare a temp directory with an example OSCAL model."""
ensure_trestle_config_dir(tmp_dir)
model_alias = utils.classname_to_alias(model_obj.__class__.__name__, 'json')
file_ext = FileContentType.to_file_extension(content_type)
models_full_path = tmp_dir / models_dir_name / 'my_test_model'
model_def_file = models_full_path / f'{model_alias}{file_ext}'
fs.ensure_directory(models_full_path)
model_obj.oscal_write(model_def_file)
return models_full_path, model_def_file