regscale-cli 6.25.1.0__py3-none-any.whl → 6.27.0.0__py3-none-any.whl

tests/regscale/integrations/public/test_fedramp.py

@@ -849,3 +849,304 @@ class TestFedRamp(CLITestFixture):
 
         assert asset.get("ipAddress") is None
         assert asset.get("iPv6Address") is None
+
+    @staticmethod
+    def test_ssp_parser_vertical_table_single_header():
+        """Test vertical table detection with single header (original behavior)."""
+        from regscale.integrations.public.fedramp.docx_parser import SSPDocParser
+        from unittest.mock import MagicMock
+
+        xml_content = b"""
+        <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+            <w:tbl>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>System Owner Information</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Name</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>John Doe</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Title</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>System Owner</w:t></w:r></w:p></w:tc>
+                </w:tr>
+            </w:tbl>
+        </w:document>
+        """
+
+        parser = SSPDocParser.__new__(SSPDocParser)
+        parser.doc = MagicMock()
+        parser.doc.element.body = []  # No preceding text needed for this test
+
+        tables = parser.parse_xml_for_tables(xml_content)
+
+        assert len(tables) == 1
+        table_data = tables[0]["table_data"]
+        assert len(table_data) == 2
+        assert table_data[0] == {"Name": "John Doe"}
+        assert table_data[1] == {"Title": "System Owner"}
+
+    @staticmethod
+    def test_ssp_parser_vertical_table_two_headers_empty_second():
+        """Test vertical table detection with two headers where second is empty (new fix)."""
+        from regscale.integrations.public.fedramp.docx_parser import SSPDocParser
+        from unittest.mock import MagicMock
+
+        xml_content = b"""
+        <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+            <w:tbl>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>System Information</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t></w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>CSP Name:</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Public Safety Platform (PSP)</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>CSO Name:</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Mark43, Inc.</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Service Model:</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Software-as-a-Service (SaaS)</w:t></w:r></w:p></w:tc>
+                </w:tr>
+            </w:tbl>
+        </w:document>
+        """
+
+        parser = SSPDocParser.__new__(SSPDocParser)
+        parser.doc = MagicMock()
+        parser.doc.element.body = []  # No preceding text needed for this test
+
+        tables = parser.parse_xml_for_tables(xml_content)
+
+        assert len(tables) == 1
+        table_data = tables[0]["table_data"]
+        assert len(table_data) == 3
+        assert table_data[0] == {"CSP Name:": "Public Safety Platform (PSP)"}
+        assert table_data[1] == {"CSO Name:": "Mark43, Inc."}
+        assert table_data[2] == {"Service Model:": "Software-as-a-Service (SaaS)"}
+
+    @staticmethod
+    def test_ssp_parser_vertical_table_two_headers_whitespace_second():
+        """Test vertical table detection with two headers where second is only whitespace."""
+        from regscale.integrations.public.fedramp.docx_parser import SSPDocParser
+        from unittest.mock import MagicMock
+
+        xml_content = b"""
+        <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+            <w:tbl>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>System Information</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>   </w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>FedRAMP Package ID:</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>FR2235965777</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>FIPS PUB 199 Level:</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>High</w:t></w:r></w:p></w:tc>
+                </w:tr>
+            </w:tbl>
+        </w:document>
+        """
+
+        parser = SSPDocParser.__new__(SSPDocParser)
+        parser.doc = MagicMock()
+        parser.doc.element.body = []  # No preceding text needed for this test
+
+        tables = parser.parse_xml_for_tables(xml_content)
+
+        assert len(tables) == 1
+        table_data = tables[0]["table_data"]
+        assert len(table_data) == 2
+        assert table_data[0] == {"FedRAMP Package ID:": "FR2235965777"}
+        assert table_data[1] == {"FIPS PUB 199 Level:": "High"}
+
+    @staticmethod
+    def test_ssp_parser_horizontal_table_not_detected_as_vertical():
+        """Test that horizontal tables with meaningful headers are not detected as vertical."""
+        from regscale.integrations.public.fedramp.docx_parser import SSPDocParser
+        from unittest.mock import MagicMock
+
+        xml_content = b"""
+        <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+            <w:tbl>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Date</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Description</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Version</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>05/14/2024</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Initial version</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>1.0</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>11/01/2024</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Updated version</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>2.0</w:t></w:r></w:p></w:tc>
+                </w:tr>
+            </w:tbl>
+        </w:document>
+        """
+
+        parser = SSPDocParser.__new__(SSPDocParser)
+        parser.doc = MagicMock()
+        parser.doc.element.body = []  # No preceding text needed for this test
+
+        tables = parser.parse_xml_for_tables(xml_content)
+
+        assert len(tables) == 1
+        table_data = tables[0]["table_data"]
+        assert len(table_data) == 2
+        # Should be horizontal format with named columns
+        assert table_data[0] == {"Date": "05/14/2024", "Description": "Initial version", "Version": "1.0"}
+        assert table_data[1] == {"Date": "11/01/2024", "Description": "Updated version", "Version": "2.0"}
+
+    @staticmethod
+    def test_ssp_parser_two_column_table_non_vertical():
+        """Test that 2-column tables with meaningful second header are not detected as vertical."""
+        from regscale.integrations.public.fedramp.docx_parser import SSPDocParser
+        from unittest.mock import MagicMock
+
+        xml_content = b"""
+        <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+            <w:tbl>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Security Objective</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Level</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Confidentiality</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>High</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Integrity</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>High</w:t></w:r></w:p></w:tc>
+                </w:tr>
+            </w:tbl>
+        </w:document>
+        """
+
+        parser = SSPDocParser.__new__(SSPDocParser)
+        parser.doc = MagicMock()
+        parser.doc.element.body = []  # No preceding text needed for this test
+
+        tables = parser.parse_xml_for_tables(xml_content)
+
+        assert len(tables) == 1
+        table_data = tables[0]["table_data"]
+        assert len(table_data) == 2
+        # Should be horizontal format since second header is not empty
+        assert table_data[0] == {"Security Objective": "Confidentiality", "Level": "High"}
+        assert table_data[1] == {"Security Objective": "Integrity", "Level": "High"}
+
+    @staticmethod
+    def test_ssp_parser_vertical_table_first_header_not_in_list():
+        """Test that tables with first header not in vertical_tables list are not detected as vertical."""
+        from regscale.integrations.public.fedramp.docx_parser import SSPDocParser
+        from unittest.mock import MagicMock
+
+        xml_content = b"""
+        <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+            <w:tbl>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Random Table Header</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t></w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Field One:</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Value One</w:t></w:r></w:p></w:tc>
+                </w:tr>
+                <w:tr>
+                    <w:tc><w:p><w:r><w:t>Field Two:</w:t></w:r></w:p></w:tc>
+                    <w:tc><w:p><w:r><w:t>Value Two</w:t></w:r></w:p></w:tc>
+                </w:tr>
+            </w:tbl>
+        </w:document>
+        """
+
+        parser = SSPDocParser.__new__(SSPDocParser)
+        parser.doc = MagicMock()
+        parser.doc.element.body = []  # No preceding text needed for this test
+
+        tables = parser.parse_xml_for_tables(xml_content)
+
+        assert len(tables) == 1
+        table_data = tables[0]["table_data"]
+        assert len(table_data) == 2
+        # Should be horizontal format since "Random Table Header" is not in vertical_tables list
+        assert table_data[0] == {"Random Table Header": "Field One:", "": "Value One"}
+        assert table_data[1] == {"Random Table Header": "Field Two:", "": "Value Two"}
+
+    @staticmethod
+    def test_ssp_parser_all_vertical_table_types():
+        """Test all table types that should be detected as vertical."""
+        from regscale.integrations.public.fedramp.docx_parser import SSPDocParser
+        from unittest.mock import MagicMock
+
+        vertical_table_headers = [
+            "Identification of Organization that Prepared this Document",
+            "Identification of Cloud Service Provider",
+            "System Owner Information",
+            "System Information",
+            "System Component Information",
+            "ISSO (or Equivalent) Point of Contact",
+        ]
+
+        for header in vertical_table_headers:
+            # Test with single header
+            xml_content = f"""
+            <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+                <w:tbl>
+                    <w:tr>
+                        <w:tc><w:p><w:r><w:t>{header}</w:t></w:r></w:p></w:tc>
+                    </w:tr>
+                    <w:tr>
+                        <w:tc><w:p><w:r><w:t>Field:</w:t></w:r></w:p></w:tc>
+                        <w:tc><w:p><w:r><w:t>Value</w:t></w:r></w:p></w:tc>
+                    </w:tr>
+                </w:tbl>
+            </w:document>
+            """.encode()
+
+            parser = SSPDocParser.__new__(SSPDocParser)
+            parser.doc = MagicMock()
+            parser.doc.element.body = []  # No preceding text needed for this test
+
+            tables = parser.parse_xml_for_tables(xml_content)
+
+            assert len(tables) == 1
+            table_data = tables[0]["table_data"]
+            assert len(table_data) == 1
+            assert table_data[0] == {"Field:": "Value"}
+
+            # Test with two headers (empty second)
+            xml_content_two_headers = f"""
+            <w:document xmlns:w="http://schemas.openxmlformats.org/wordprocessingml/2006/main">
+                <w:tbl>
+                    <w:tr>
+                        <w:tc><w:p><w:r><w:t>{header}</w:t></w:r></w:p></w:tc>
+                        <w:tc><w:p><w:r><w:t></w:t></w:r></w:p></w:tc>
+                    </w:tr>
+                    <w:tr>
+                        <w:tc><w:p><w:r><w:t>Field:</w:t></w:r></w:p></w:tc>
+                        <w:tc><w:p><w:r><w:t>Value</w:t></w:r></w:p></w:tc>
+                    </w:tr>
+                </w:tbl>
+            </w:document>
+            """.encode()
+
+            parser2 = SSPDocParser.__new__(SSPDocParser)
+            parser2.doc = MagicMock()
+            parser2.doc.element.body = []  # No preceding text needed for this test
+
+            tables2 = parser2.parse_xml_for_tables(xml_content_two_headers)
+
+            assert len(tables2) == 1
+            table_data2 = tables2[0]["table_data"]
+            assert len(table_data2) == 1
+            assert table_data2[0] == {"Field:": "Value"}