Revert "squash (#169)" (#172)

This reverts commit 3ffef3b.
lacework · Mar 22, 2024 · a70aa89 · a70aa89
1 parent 3ffef3b
commit a70aa89
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 23 deletions.
diff --git a/.github/workflows/python-test.yml b/.github/workflows/python-test.yml
@@ -56,7 +56,7 @@ jobs:
           LW_API_SECRET: ${{ secrets.LW_API_SECRET }}
           LW_BASE_DOMAIN: ${{ secrets.LW_BASE_DOMAIN }}
       - name: Report Status
-        if: github.ref_name == 'main'
+        if: contains(github.ref, "main")
         uses: ravsamhq/notify-slack-action@v2
         with:
           status: ${{ job.status }}

diff --git a/laceworksdk/api/v2/vulnerabilities.py b/laceworksdk/api/v2/vulnerabilities.py
@@ -42,7 +42,6 @@ def __init__(self, session):
         self.containers = self.ContainerVulnerabilitiesAPI(session, self._base_path)
         self.hosts = self.HostVulnerabilitiesAPI(session, self._base_path)
         self.packages = self.SoftwarePackagesAPI(session, self._base_path)
-        self.imageSummary = self.ImageSummaryVulnerabilitiesAPI(session, self._base_path)
 
     class ContainerVulnerabilitiesAPI(SearchEndpoint):
         """A class used to represent the Container Vulnerabilities API endpoint."""
@@ -91,12 +90,6 @@ def status(self, request_id):
             )
 
             return response.json()
-
-    class ImageSummaryVulnerabilitiesAPI(SearchEndpoint):
-        """A class used to represent the ImageSummary Vulnerabilities API endpoint."""
-
-        RESOURCE = "ImageSummary"
-
 
     class HostVulnerabilitiesAPI(SearchEndpoint):
         """A class used to represent the Host Vulnerabilities API endpoint."""

diff --git a/tests/api/v2/test_vulnerabilities.py b/tests/api/v2/test_vulnerabilities.py
@@ -23,8 +23,7 @@ class TestVulnerabilitesEndpoint(SearchEndpoint):
     OBJECT_TYPE = VulnerabilitiesAPI
     OBJECT_MAP = {
         "containers": VulnerabilitiesAPI.ContainerVulnerabilitiesAPI,
-        "hosts": VulnerabilitiesAPI.HostVulnerabilitiesAPI,
-        "imageSummary": VulnerabilitiesAPI.ImageSummaryVulnerabilitiesAPI
+        "hosts": VulnerabilitiesAPI.HostVulnerabilitiesAPI
     }
 
     def test_vulnerabilities_containers_api_scan(self, api_object, request):
@@ -68,16 +67,3 @@ def test_vulnerabilities_packages_api_scan(api, api_object):
             "pkgVer": "1.1.1-1ubuntu2.1~20.04"
         }])
         assert "data" in response.keys()
-
-    def test_vulnerabilities_image_summary_search(api, api_object):
-        json = {
-            "timeFilter": {
-                "startTime": "2024-03-18T00:00:00Z",
-                "endTime":   "2024-03-19T08:00:00Z"
-            },
-            "filters" : [
-                {"field": "ndvContainers", "expression": "gt", "value": 0}
-            ]
-        } 
-        response = api_object.imageSummary.search(json)
-        assert "data" in next(response, None)