feat: adds fedramp-transform command implementation

setup.cfg

@@ -28,6 +28,7 @@ include_package_data = True
 install_requires =
     compliance-trestle>=3.3.0
     saxonche>=12.4.1.0
+    python-docx>=1.1.0
 
 [options.packages.find]
 include = trestle_fedramp*

tests/conftest.py

@@ -15,16 +15,27 @@
 # limitations under the License.
 """Common fixtures."""
 
+import argparse
 import os
 import pathlib
 import sys
-from typing import Iterator
+from typing import Iterator, Tuple
+
+from docx import Document  # type: ignore
+from docx.document import Document as DocxDocument  # type: ignore
 
 import pytest
 from pytest import MonkeyPatch
 
+from tests import test_utils
+
 from trestle.cli import Trestle
 from trestle.common.err import TrestleError
+from trestle.core.commands.import_ import ImportCmd
+
+from trestle_fedramp import const
+from trestle_fedramp.core.baselines import BaselineLevel
+from trestle_fedramp.core.ssp_reader import FedrampControlDict, FedrampSSPData
 
 
 @pytest.fixture(scope='function')
@@ -46,3 +57,100 @@ def tmp_trestle_dir(tmp_path: pathlib.Path, monkeypatch: MonkeyPatch) -> Iterato
         yield tmp_path
     finally:
         os.chdir(pytest_cwd)
+
+
+@pytest.fixture(scope='function')
+def tmp_trestle_dir_with_ssp(tmp_path: pathlib.Path, monkeypatch: MonkeyPatch) -> Iterator[Tuple[pathlib.Path, str]]:
+    """Create initialized trestle workspace and import the model into it."""
+    pytest_cwd = pathlib.Path.cwd()
+    model_name = 'ssp'
+    file_path = test_utils.JSON_TEST_DATA_PATH / test_utils.TEST_SSP_JSON
+    os.chdir(tmp_path)
+    testargs = ['trestle', 'init']
+    monkeypatch.setattr(sys, 'argv', testargs)
+    try:
+        Trestle().run()
+        i = ImportCmd()
+        args = argparse.Namespace(
+            trestle_root=tmp_path, file=str(file_path), output=model_name, verbose=1, regenerate=False
+        )
+        assert i._run(args) == 0
+    except Exception as e:
+        raise TrestleError(f'Error creating trestle workspace with ssp: {e}')
+    else:
+        yield (tmp_path, model_name)
+    finally:
+        os.chdir(pytest_cwd)
+
+
+@pytest.fixture(scope='function')
+def docx_document() -> Iterator[DocxDocument]:
+    """Return a docx document."""
+    template = pathlib.Path(BaselineLevel.get_template('high')).resolve()
+    with open(template, 'rb') as file:
+        document: DocxDocument = Document(file)
+    yield document
+
+
+@pytest.fixture(scope='function')
+def test_ssp_control_dict() -> Iterator[FedrampControlDict]:
+    """Return a dictionary of control data."""
+    control_dict: FedrampControlDict = {
+        'AC-1': FedrampSSPData(
+            control_implementation_description={
+                'a': 'Part a example for AC-1', 'b': 'Part b example for AC-1', 'c': 'Part c example for AC-1'
+            },
+            parameters={
+                'AC-1(a)': 'AC-1(a) parameter example',
+                'AC-1(a)(1)': 'AC-1(a)(1) parameter example',
+                'AC-1(b)': 'AC-1(b) parameter example',
+                'AC-1(c)(1)-1': 'AC-1(c)(1)-1 parameter example',
+                'AC-1(c)(1)-2': 'AC-1(c)(1)-2 parameter example',
+                'AC-1(c)(2)-1': 'AC-1(c)(2)-1 parameter example',
+                'AC-1(c)(2)-2': 'AC-1(c)(2)-2 parameter example',
+            },
+            control_origination=None,
+            implementation_status=None,
+            responsible_roles=None
+        ),
+        'AC-2(1)': FedrampSSPData(
+            control_implementation_description={'': 'Overall description for AC-2(1)'},
+            parameters={
+                'AC-2(2)-1': 'AC-2(2)-1 parameter example',
+                'AC-2(2)-2': 'AC-2(2)-2 parameter example',
+            },
+            control_origination=[const.FEDRAMP_SP_SYSTEM],
+            implementation_status=const.FEDRAMP_PLANNED,
+            responsible_roles=['Customer']
+        ),
+        'AC-20': FedrampSSPData(
+            control_implementation_description={
+                'a': 'Part a example for AC-20',
+                'b': 'Part b example for AC-20',
+            },
+            parameters={
+                'AC-20(a)': 'AC-20(a) parameter example',
+                'AC-20(b)': 'AC-20(b) parameter example',
+            },
+            control_origination=[const.FEDRAMP_SP_CORPORATE, const.FEDRAMP_INHERITED],
+            implementation_status=const.FEDRAMP_PARTIAL,
+            responsible_roles=['Admin', 'Customer']
+        ),
+        'CM-6': FedrampSSPData(
+            control_implementation_description={
+                'a': 'Part a example for CM-6',
+                'b': 'Part b example for CM-6',
+                'c': 'Part c example for CM-6',
+                'd': 'Part d example for CM-6',
+            },
+            parameters={
+                'CM-6(a)': 'CM-6(a) parameter example',
+                'CM-6(c)-1': 'CM-6(c)-1 parameter example',
+                'CM-6(c)-2': 'CM-6(c)-2 parameter example',
+            },
+            control_origination=[const.FEDRAMP_SHARED],
+            implementation_status=const.FEDRAMP_IMPLEMENTED,
+            responsible_roles=['Provider']
+        )
+    }
+    yield control_dict