pyntcli 0.1.92__py3-none-any.whl → 0.1.93__py3-none-any.whl

pyntcli/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.92"
+__version__ = "0.1.93"

pyntcli/commands/burp.py

@@ -147,7 +147,7 @@ def burp_usage():
         .with_line("\t--insecure - Use when target uses self signed certificates")
         .with_line("\t--application-id - Attach the scan to an application, you can find the ID in your applications area at app.pynt.io")
         .with_line("\t--host-ca - Path to the CA file in PEM format to enable SSL certificate verification for pynt when running through a VPN.")
-        .with_line("\t--return-error - 'all-findings' (warnings, or errors), 'errors-only', 'never' (default)")
+        .with_line("\t--severity-level - 'all', 'medium', 'high', 'critical', 'none' (default) ")
         .with_line("\t--verbose - Use to get more detailed information about the run")
     )
 
@@ -171,11 +171,7 @@ class BurpCommand(sub_command.PyntSubCommand):
         burp_cmd.add_argument("--ca-path", type=str, default="")
         burp_cmd.add_argument("--report", type=str, default="")
         burp_cmd.add_argument("--captured-domains", nargs="+", help="", default="")
-        burp_cmd.add_argument(
-            "--return-error",
-            choices=["all-findings", "errors-only", "never"],
-            default="never",
-        )
+        burp_cmd.add_argument("--severity-level", choices=["all", "medium", "high", "critical", "none"], default="none")
         burp_cmd.print_usage = self.print_usage
         burp_cmd.print_help = self.print_usage
         return burp_cmd
@@ -296,7 +292,10 @@ class BurpCommand(sub_command.PyntSubCommand):
             base_container=container,
             use_native=args.use_docker_native)
 
+        proxy_docker.prepare_client()
+        proxy_docker.pre_run_validation(args.port)
         proxy_docker.run()
+
         ui_thread.print_generator(proxy_docker.stdout)
 
         util.wait_for_healthcheck("http://localhost:{}".format(args.port))
@@ -355,3 +354,4 @@ class BurpCommand(sub_command.PyntSubCommand):
                 json_obj = json.loads(json_report)
                 if json_obj:
                     util.check_for_findings_or_warnings(args, json_obj)
+                    util.check_severity(args.severity_level, json_obj)

pyntcli/commands/command.py

@@ -42,7 +42,7 @@ def command_usage():
         .with_line("\t--no-proxy-export - Pynt will not export the proxy settings to the environment")
         .with_line("\t--application-id - Attach the scan to an application, you can find the ID in your applications area at app.pynt.io")
         .with_line("\t--host-ca - Path to the CA file in PEM format to enable SSL certificate verification for pynt when running through a VPN.")
-        .with_line("\t--return-error - 'all-findings' (warnings, or errors), 'errors-only', 'never' (default) ")
+        .with_line("\t--severity-level - 'all', 'medium', 'high', 'critical', 'none' (default) ")
         .with_line("\t--verbose - Use to get more detailed information about the run")
     )
 
@@ -70,7 +70,7 @@ class CommandSubCommand(sub_command.PyntSubCommand):
         proxy_cmd.add_argument("--ca-path", type=str, default="")
         proxy_cmd.add_argument("--self-signed", action="store_true")
         proxy_cmd.add_argument("--report", type=str, default="")
-        proxy_cmd.add_argument("--return-error", choices=["all-findings", "errors-only", "never"], default="never")
+        proxy_cmd.add_argument("--severity-level", choices=["all", "medium", "high", "critical", "none"], default="none")
         proxy_cmd.print_usage = self.print_usage
         proxy_cmd.print_help = self.print_usage
         return proxy_cmd
@@ -183,7 +183,10 @@ class CommandSubCommand(sub_command.PyntSubCommand):
             base_container=container,
             use_native=args.use_docker_native)
 
+        proxy_docker.prepare_client()
+        proxy_docker.pre_run_validation(args.port)
         proxy_docker.run()
+
         ui_thread.print_generator(proxy_docker.stdout)
 
         util.wait_for_healthcheck("http://localhost:{}".format(args.port))
@@ -261,3 +264,4 @@ class CommandSubCommand(sub_command.PyntSubCommand):
                 json_obj = json.loads(json_report)
                 if json_obj:
                     util.check_for_findings_or_warnings(args, json_obj)
+                    util.check_severity(args.severity_level, json_obj)

pyntcli/commands/har.py

@@ -30,6 +30,7 @@ def har_usage():
         .with_line(
             "\t--host-ca - Path to the CA file in PEM format to enable SSL certificate verification for pynt when running through a VPN."
         )
+        .with_line("\t--severity-level - 'all', 'medium', 'high', 'critical', 'none' (default) ")
         .with_line("\t--verbose - Use to get more detailed information about the run")
         .with_line("")
     )
@@ -49,6 +50,7 @@ class HarSubCommand(sub_command.PyntSubCommand):
             "--captured-domains", nargs="+", help="", default="", required=True
         )
         har_cmd.add_argument("--reporters", action="store_true")
+        har_cmd.add_argument("--severity-level", choices=["all", "medium", "high", "critical", "none"], default="none")
         har_cmd.print_usage = self.usage
         har_cmd.print_help = self.usage
         return har_cmd
@@ -91,6 +93,8 @@ class HarSubCommand(sub_command.PyntSubCommand):
                 base_container=container,
                 use_native=args.use_docker_native)
 
+            har_docker.prepare_client()
+            har_docker.pre_run_validation(port)
             har_docker.run()
 
             healthcheck = partial(

pyntcli/commands/listen.py

@@ -36,7 +36,7 @@ def listen_usage():
         .with_line("\t--application-id - Attach the scan to an application, you can find the ID in your applications area at app.pynt.io")
         .with_line("\t--insecure - use when target uses self signed certificates")
         .with_line("\t--host-ca - path to the CA file in PEM format to enable SSL certificate verification for pynt when running through a VPN.")
-        .with_line("\t--return-error - 'all-findings' (warnings, or errors), 'errors-only', 'never' (default) ")
+        .with_line("\t--severity-level - 'all', 'medium', 'high', 'critical', 'none' (default) ")
         .with_line("\t--verbose - Use to get more detailed information about the run")
     )
 
@@ -61,7 +61,7 @@ class ListenSubCommand(sub_command.PyntSubCommand):
         listen_cmd.add_argument("--allow-errors", action="store_true")
         listen_cmd.add_argument("--ca-path", type=str, default="")
         listen_cmd.add_argument("--report", type=str, default="")
-        listen_cmd.add_argument("--return-error", choices=["all-findings", "errors-only", "never"], default="never")
+        listen_cmd.add_argument("--severity-level", choices=["all", "medium", "high", "critical", "none"], default="none")
         listen_cmd.print_usage = self.print_usage
         listen_cmd.print_help = self.print_usage
         return listen_cmd
@@ -140,7 +140,11 @@ class ListenSubCommand(sub_command.PyntSubCommand):
             tag="proxy-latest",
             detach=True,
             base_container=container)
+
+        proxy_docker.prepare_client()
+        proxy_docker.pre_run_validation(args.port)
         proxy_docker.run()
+
         ui_thread.print_generator(proxy_docker.stdout)
 
         util.wait_for_healthcheck("http://localhost:{}".format(args.port))
@@ -208,3 +212,4 @@ class ListenSubCommand(sub_command.PyntSubCommand):
                 json_obj = json.loads(json_report)
                 if json_obj:
                     util.check_for_findings_or_warnings(args, json_obj)
+                    util.check_severity(args.severity_level, json_obj)

pyntcli/commands/newman.py

@@ -30,9 +30,7 @@ def newman_usage():
         .with_line(
             "\t--application-id - Attach the scan to an application, you can find the ID in your applications area at app.pynt.io"
         )
-        .with_line(
-            "\t--return-error - 'all-findings' (warnings, or errors), 'errors-only', 'never' (default) "
-        )
+        .with_line("\t--severity-level - 'all', 'medium', 'high', 'critical', 'none' (default) ")
         .with_line("\t--verbose - Use to get more detailed information about the run")
     )
 
@@ -51,12 +49,7 @@ class NewmanSubCommand(sub_command.PyntSubCommand):
         newman_cmd.add_argument(
             "--reporters", action="store_true", default=False, required=False
         )
-        newman_cmd.add_argument(
-            "--return-error",
-            choices=["all-findings", "errors-only", "never"],
-            default="never"
-        )
-
+        newman_cmd.add_argument("--severity-level", choices=["all", "medium", "high", "critical", "none"], default="none")
         newman_cmd.print_usage = self.usage
         newman_cmd.print_help = self.usage
         return newman_cmd
@@ -113,7 +106,11 @@ class NewmanSubCommand(sub_command.PyntSubCommand):
                 detach=True,
                 base_container=container,
                 use_native=args.use_docker_native)
+
+            newman_docker.prepare_client()
+            newman_docker.pre_run_validation(port)
             newman_docker.run()
+
             healthcheck = partial(
                 util.wait_for_healthcheck, "http://localhost:{}".format(port)
             )

pyntcli/commands/postman.py

@@ -102,8 +102,10 @@ class PostmanSubCommand(sub_command.PyntSubCommand):
                                                       detach=True,
                                                       base_container=container,
                                                       use_native=args.use_docker_native)
-
+        postman_docker.prepare_client()
+        postman_docker.pre_run_validation(args.port)
         postman_docker.run()
+
         ui_thread.print_generator(postman_docker.stdout)
 
         util.wait_for_healthcheck("http://localhost:{}".format(args.port))

pyntcli/commands/util.py

@@ -68,6 +68,10 @@ class SomeFindingsOrWarningsException(Exception):
     pass
 
 
+class SeverityException(Exception):
+    pass
+
+
 @contextmanager
 def create_default_file_mounts(args):
     html_report_path = os.path.join(tempfile.gettempdir(), "results.html")
@@ -93,8 +97,10 @@ def create_default_file_mounts(args):
         report.PyntReporter(json_report_path).print_summary()
 
     check_for_findings_or_warnings(args, json.load(open(json_report_path)))
+    check_severity(args.severity_level, json.load(open(json_report_path)))
 
 
+# Deprecate - keep it for backward customers that use it
 def check_for_findings_or_warnings(args, json_report):
     security_tests = json_report.get("securityTests", {})
     findings = security_tests.get("Findings", 0)
@@ -105,3 +111,30 @@ def check_for_findings_or_warnings(args, json_report):
 
     if "return_error" in args and args.return_error == "all-findings" and warnings != 0:
         raise SomeFindingsOrWarningsException()
+
+
+def check_severity(severity_flag, json_report):
+    severity_levels = ['medium', 'high', 'critical']
+
+    if severity_flag is None or severity_flag.lower() == "none":
+        return
+
+    risk_data = json_report.get("securityTests", {})
+
+    # Normalize all te keys to lower to reduce the risk that user input is diff then what we have internally
+    risks_counter = {key.lower(): value for key, value in risk_data['RisksCounter'].items()}
+
+    severity_filter = severity_flag.lower()
+
+    if severity_filter == "all":
+        severities_to_check = severity_levels
+    else:
+        if severity_filter not in severity_levels:
+            raise ValueError(f"Invalid user filter for severity level: {severity_filter}")
+
+        flag_index = severity_levels.index(severity_filter)
+        severities_to_check = severity_levels[flag_index:]
+
+    for severity in severities_to_check:
+        if risks_counter.get(severity, 0) > 0:
+            raise SeverityException(f"Severity '{severity}' has a count greater than 0.")

pyntcli/main.py

@@ -12,7 +12,7 @@ from requests.exceptions import SSLError
 from requests.exceptions import ProxyError
 from pyntcli.transport.pynt_requests import InvalidPathException, InvalidCertFormat
 from pyntcli.commands.util import HtmlReportNotCreatedException
-from pyntcli.commands.util import SomeFindingsOrWarningsException
+from pyntcli.commands.util import SomeFindingsOrWarningsException, SeverityException
 from pyntcli.commands.postman import PyntWebSocketException
 from pyntcli import __version__
 
@@ -131,6 +131,8 @@ def main():
         analytics.emit(analytics.ERROR, {"error": "port in use. e: {}".format(e)})
     except SomeFindingsOrWarningsException as e:
         exit(1)
+    except SeverityException as e:
+        exit(1)
     except Exception as e:
         analytics.emit(analytics.ERROR, {"error": "{}".format(e)})
         pynt_errors.unexpected_error(e)

pyntcli/pynt_docker/pynt_container.py

@@ -2,6 +2,7 @@ import platform
 import subprocess
 import docker
 from docker.errors import DockerException, APIError, ImageNotFound
+from requests.exceptions import ReadTimeout
 from docker.types import Mount
 import os
 import json
@@ -17,6 +18,9 @@ from pyntcli.store import CredStore
 from pyntcli.auth.login import PYNT_ID, PYNT_SAAS, PYNT_BUCKET_NAME, PYNT_PARAM1, PYNT_PARAM2
 
 PYNT_DOCKER_IMAGE = "ghcr.io/pynt-io/pynt"
+IMAGE_TAGS = ["postman-latest", "newman-latest", "har-latest", "proxy-latest"]
+
+PYNT_CONTAINER_INTERNAL_PORT = "5001"
 
 
 def create_mount(src, destination, mount_type="bind"):
@@ -37,7 +41,8 @@ class ImageUnavailableException(Exception):
 
 class PortInUseException(Exception):
     def __init__(self, port=""):
-        self.message = ui_thread.print(ui_thread.PrinterText("Port: {} already in use, please use a different one".format(port), ui_thread.PrinterText.WARNING))
+        self.message = ui_thread.print(
+            ui_thread.PrinterText(f"Port: {port} already in use, please use a different one", ui_thread.PrinterText.WARNING))
         super().__init__(self.message)
 
 
@@ -87,8 +92,6 @@ def get_container_with_arguments(args: argparse.Namespace, *port_args: PyntDocke
     docker_arguments = []
     ports = {}
     for p in port_args:
-        if container_utils.is_port_in_use(p.dest):
-            raise PortInUseException(p.dest)
         if "desktop" in get_docker_type().lower():
             ports[str(p.src)] = int(p.dest)
         else:
@@ -189,9 +192,11 @@ class PyntContainerNative:
 
         return len(result.stdout.splitlines()) > 1
 
+    def prepare_client(self):
+        pass
+
     def run(self):
         self.running = True
-        self.kill_other_instances()
 
         self.get_image()
         args = self.base_container.docker_arguments if self.base_container.docker_arguments else None
@@ -247,18 +252,22 @@ class PyntContainerNative:
 
         self.stdout = logs_stdout
 
-    def kill_other_instances(self):
+    def kill_other_instances(self, report_to_user=True):
+        ui_thread.print_verbose("Killing other pynt containers if such exist")
         try:
-            ui_thread.print_verbose("Killing other pynt containers if such exist")
-            command = ["docker", "ps", "-q", "-f", f"ancestor={self.image_name}:{self.tag}"]
-            containers_output = subprocess.check_output(command, text=True)
-            if not containers_output:
-                return
-
-            container_ids = containers_output.splitlines()
-            for container_id in container_ids:
-                command = ["docker", "kill", container_id]
-                subprocess.run(command)
+            for tag in IMAGE_TAGS:
+                command = ["docker", "ps", "-q", "-f", f"ancestor={self.image_name}:{tag}"]
+                containers_output = subprocess.check_output(command, text=True)
+                if not containers_output:
+                    continue
+
+                container_ids = containers_output.splitlines()
+                for container_id in container_ids:
+                    command = ["docker", "kill", container_id]
+                    subprocess.run(command)
+                    if report_to_user:
+                        ui_thread.print(
+                            ui_thread.PrinterText("Another Pynt container was running, killed it", ui_thread.PrinterText))
 
         except subprocess.CalledProcessError:
             analytics.emit(analytics.ERROR, {"error": "Unable to kill other pynt containers"})
@@ -305,7 +314,7 @@ class PyntContainerNative:
     def stop(self):
         if not self.running:
             return
-        self.kill_other_instances()
+        self.kill_other_instances(report_to_user=False)
         self.running = False
 
     def adapt_run_command(self, docker_command=[]):
@@ -333,6 +342,7 @@ class PyntContainerSDK:
         self.container_name = ""
         self.stdout = None
         self.running = False
+        self.system = platform.system().lower()
 
     def _initialize(self):
         self.docker_client = docker.from_env()
@@ -352,15 +362,16 @@ class PyntContainerSDK:
 
         return l[0].status == "running"
 
+    def prepare_client(self):
+        if not self.docker_client:
+            self._initialize()
+
     def run(self):
         if not self.docker_client:
             self._initialize()
 
         self.running = True
 
-        ui_thread.print_verbose("Killing other pynt containers if such exist")
-        self.kill_other_instances()
-
         image = self.get_image()
         ui_thread.print(ui_thread.PrinterText("Docker pull done", ui_thread.PrinterText.INFO))
 
@@ -383,10 +394,25 @@ class PyntContainerSDK:
         self.container_name = c.name
         self.stdout = c.logs(stream=True)
 
-    def kill_other_instances(self):
+    def kill_other_instances(self, report_to_user=True):
         for c in self.docker_client.containers.list():
             if len(c.image.tags) and _container_image_from_tag(c.image.tags[0]) == self.image_name:
                 c.kill()
+                self.wait_for_container_end(c)
+                if report_to_user:
+                    ui_thread.print(ui_thread.PrinterText("Another Pynt container was running, killed it", ui_thread.PrinterText))
+
+    def wait_for_container_end(self, container):
+        # only windows kill is require a wait for the container to stop, otherwise the port stays in use
+        if self.system != "windows":
+            return
+        try:
+            container.wait(timeout=10)
+        except ReadTimeout:  # container is still running
+            ui_thread.print(
+                ui_thread.PrinterText("Timeout reached while waiting for container to stop", ui_thread.PrinterText))
+        except APIError:  # container is already stopped
+            pass
 
     def pull_image(self):
         try:
@@ -410,7 +436,7 @@ class PyntContainerSDK:
     def stop(self):
         if not self.running:
             return
-        self.kill_other_instances()
+        self.kill_other_instances(report_to_user=False)
         self.docker_client.close()
         self.docker_client = None
         self.running = False
@@ -452,6 +478,15 @@ class PyntContainer:
         self.stdout = self.client_implementation.stdout
         PyntContainerRegistry.instance().register_container(self)
 
+    def pre_run_validation(self, port):
+        self.kill_other_instances()
+
+        if container_utils.is_port_in_use(int(port)):
+            raise PortInUseException(port)
+
+    def prepare_client(self):
+        self.client_implementation.prepare_client()
+
     def running(self):
         return self.client_implementation.running
 

{pyntcli-0.1.92.dist-info → pyntcli-0.1.93.dist-info}/METADATA

@@ -1,16 +1,16 @@
 Metadata-Version: 2.1
 Name: pyntcli
-Version: 0.1.92
+Version: 0.1.93
 Summary: Command line utility to handle all of Pynt's different integrations
 Author-email: Pynt-io <support@pynt.io>
 Project-URL: Homepage, https://pynt.io
 Requires-Python: >=3.7
 Requires-Dist: docker
 Requires-Dist: rich
-Requires-Dist: requests ==2.31.0
+Requires-Dist: requests==2.31.0
 Requires-Dist: pem
-Requires-Dist: certifi >=2017.4.17
-Requires-Dist: logzio-python-handler >=4.1.0
+Requires-Dist: certifi>=2017.4.17
+Requires-Dist: logzio-python-handler>=4.1.0
 Requires-Dist: websocket-client
 Requires-Dist: xmltodict
 

{pyntcli-0.1.92.dist-info → pyntcli-0.1.93.dist-info}/RECORD

@@ -1,27 +1,30 @@
-pyntcli/__init__.py,sha256=jibR5k16CvIIIRqIusW7mrU6CMM0wet7LwtTuFLXbmU,23
-pyntcli/main.py,sha256=ucC9d4wlhUpvA92OXRIZ55iMdFXEDiAOVvP_wtRu7zs,6432
+ignoreTests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
+ignoreTests/auth/login.py,sha256=KFlzWhXBAuwdi7GXf16gCB3ya94LQG2wjcSChE149rQ,3798
+ignoreTests/store/cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
+pyntcli/__init__.py,sha256=9ZJPYemtF-vnUzZtNNDjmi0UnkkMXYvPkfRASPJ5iWc,23
+pyntcli/main.py,sha256=HORWY5AEK5xmEQybUZBW9ZEDIga8vSnqDfmAR3TY9F0,6502
 pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/analytics/send.py,sha256=9TRAEoPbv4rWOZfcNaGanrRJAFvNs39P-uKSl49GcQE,3679
 pyntcli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/auth/login.py,sha256=TljsRXbEkNI1YUrKm5mlTw4YiecYScYUsit8Z8vstss,5228
 pyntcli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pyntcli/commands/burp.py,sha256=NfvMPlw8yISaDzvSuF96MXGVTPKRS7yR84TNDqFiblU,12274
-pyntcli/commands/command.py,sha256=mYeNer6WeswYLPSZhvNC04S5FOdMqH-Jk8a4iPRJtpc,10973
-pyntcli/commands/har.py,sha256=IZGd900hz0fFM3iZTG3NNXB2NiuAJGeuv-I49eXrf-o,4158
+pyntcli/commands/burp.py,sha256=TYUspSLhubtKEL4ShUOvT7Fnq3MFsFo-HyrPXiZboZ4,12380
+pyntcli/commands/command.py,sha256=cMVrlxMZgx0jFrlpMTKPEYfsSv2RhN6XuPrRqqqYB6k,11125
+pyntcli/commands/har.py,sha256=yUuLvxlAoq2dQotJe8PV02sZD-WgYSefzGc6bhdroSo,4464
 pyntcli/commands/id_command.py,sha256=UBEgMIpm4vauTCsKyixltiGUolNg_OfHEJvJ_i5BpJY,943
-pyntcli/commands/listen.py,sha256=0wY9itDQyxSeHQA4GCGF67Xkb4ym48uP17WCPJMcLBE,8910
-pyntcli/commands/newman.py,sha256=ocg85L7HopXRYkhLc75QUtHH9ey48Z6Yxy1stgVW1aE,5283
-pyntcli/commands/postman.py,sha256=T1qdFHXX_kTW9p8PPbY9P7ynK4UcGoG3FfEMrRetwco,5163
+pyntcli/commands/listen.py,sha256=b_ES6VRmPFkIkCCQ6MuP6KOeUlf5vT1e7AafnDqJVak,9063
+pyntcli/commands/newman.py,sha256=o3kobgUFoC3jw2XmbYEsQNdYpGO5suFA92lxHw71dzw,5301
+pyntcli/commands/postman.py,sha256=VNARuyCDaDYlrzRMmJk4oM4YTCExOQUDziOS8NpU-KA,5256
 pyntcli/commands/pynt_cmd.py,sha256=X39hiDq6blGh2sGpwVeUXOgnDlhHruGkOoR5aUBzw_k,2982
 pyntcli/commands/root.py,sha256=-2aAiOKc_yXAlymxGove6su1mcvYsk4kCNaEawDKwy8,3677
 pyntcli/commands/static_file_extensions.py,sha256=PZJb02BI-64tbU-j3rdCNsXzTh7gkIDGxGKbKNw3h5k,1995
 pyntcli/commands/sub_command.py,sha256=GF3-rE_qk2L4jGPFqHLm9SdGINmu3EakhjJTFyWjRms,374
-pyntcli/commands/util.py,sha256=4p4rzNqL8nUJJfj1uy_JOHsFBEvVT6-k-V9GbAqHVyM,3218
+pyntcli/commands/util.py,sha256=csZHQ2Xbdh-_KX-yIVrnaeNsT0NbuS-ej6kND3CxD_w,4414
 pyntcli/log/__init__.py,sha256=cOGwOYzMoshEbZiiasBGkj6wF0SBu3Jdpl-AuakDesw,19
 pyntcli/log/log.py,sha256=cWCdWmUaAwePwdhYDcgNMEG9d9RM34sGahxBCYEdv2Y,1069
 pyntcli/pynt_docker/__init__.py,sha256=PQIOVxc7XXtMLfEX7ojgwf_Z3mmTllO3ZvzUZTPOxQY,30
 pyntcli/pynt_docker/container_utils.py,sha256=_Onn7loInzyJAG2-Uk6CGpsuRyelmUFHOvtJj4Uzi9A,175
-pyntcli/pynt_docker/pynt_container.py,sha256=sqWKi3HUIn3AR9HgDYQtofCkaEQiOT3hG1nYQHf8RD8,17500
+pyntcli/pynt_docker/pynt_container.py,sha256=R2iq4Gbbkxg3OEJfJwUp-vroHf9-pJ6HWFqvWucPsGA,18965
 pyntcli/store/__init__.py,sha256=xuS9OB21F6B1sUx5XPGxz_6WpG6-KTMbuq50RrZS5OY,29
 pyntcli/store/json_connector.py,sha256=UGs3uORw3iyn0YJ8kzab-veEZToA6d-ByXYuqEleWsA,560
 pyntcli/store/store.py,sha256=4YqmcHRltbbSWewKQoOZH8QdyMFs6i3L8o2bXY_YvSw,1909
@@ -33,11 +36,9 @@ pyntcli/ui/progress.py,sha256=RrnO_jJNunoyupylakmWmHOEPw3lh99OHpKBzL6OBiE,1008
 pyntcli/ui/pynt_errors.py,sha256=00UprD4tFViREv7kuXGQ99PAKGTpXYixxi3Ndeoeiew,689
 pyntcli/ui/report.py,sha256=W-icPSZrGLOubXgam0LpOvHLl_aZg9Zx9qIkL8Ym5PE,1930
 pyntcli/ui/ui_thread.py,sha256=XUBgLpYQjVhrilU-ofw7VSXgTiwneSdTxm61EvC3x4Q,5091
-tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
-tests/auth/test_login.py,sha256=KFlzWhXBAuwdi7GXf16gCB3ya94LQG2wjcSChE149rQ,3798
-tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
-pyntcli-0.1.92.dist-info/METADATA,sha256=MS97IJ6CIx7fidgjFBCYP8gKop_deMK3JGRBfXCsj5Y,463
-pyntcli-0.1.92.dist-info/WHEEL,sha256=Wyh-_nZ0DJYolHNn1_hMa4lM7uDedD_RGVwbmTjyItk,91
-pyntcli-0.1.92.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
-pyntcli-0.1.92.dist-info/top_level.txt,sha256=u9MDStwVHB7UG8PUcODeWCul_NvzL2EzoLvSlgwLHFs,30
-pyntcli-0.1.92.dist-info/RECORD,,
+tests/test_utils.py,sha256=t5fTQUk1U_Js6iMxcGYGqt4C-crzOJ0CqCKtLkRtUi0,2050
+pyntcli-0.1.93.dist-info/METADATA,sha256=VRflEz4VuO9RUwlmI0nWXOqHaTWdkwCvdcPq7cixewY,460
+pyntcli-0.1.93.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
+pyntcli-0.1.93.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
+pyntcli-0.1.93.dist-info/top_level.txt,sha256=64XSgBzSpgwjYjEKHZE7q3JH2a816zEeyZBXfJi3AKI,42
+pyntcli-0.1.93.dist-info/RECORD,,

{pyntcli-0.1.92.dist-info → pyntcli-0.1.93.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (71.1.0)
+Generator: setuptools (73.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{pyntcli-0.1.92.dist-info → pyntcli-0.1.93.dist-info}/top_level.txt

@@ -1,4 +1,5 @@
 automation
 dist
+ignoreTests
 pyntcli
 tests

tests/test_utils.py

@@ -0,0 +1,75 @@
+import pytest
+from pyntcli.commands.util import check_severity, SeverityException
+
+
+@pytest.fixture
+def risk_data():
+    return {
+        "securityTests": {
+            'DidNotRun': 0,
+            'Duration': 0,
+            'Findings': 5,
+            'Passed': 27,
+            'RisksCounter': {'Critical': 2, 'High': 0, 'Low': 3, 'Medium': 3},
+            'Warnings': 1
+        }
+    }
+
+
+def test_exception_on_medium_severity(risk_data):
+    with pytest.raises(SeverityException):
+        check_severity('MEDIUM', risk_data)
+
+
+def test_exception_on_high_severity(risk_data):
+    with pytest.raises(SeverityException):
+        check_severity('HIGH', risk_data)
+
+
+def test_exception_on_critical_severity(risk_data):
+    with pytest.raises(SeverityException):
+        check_severity('CRITICAL', risk_data)
+
+
+def test_no_exception_on_higher_than_all_severity():
+    data_with_low_risk_only = {
+        "securityTests": {
+            'DidNotRun': 0,
+            'Duration': 0,
+            'Findings': 5,
+            'Passed': 27,
+            'RisksCounter': {'Critical': 0, 'High': 0, 'Low': 3, 'Medium': 0},
+            'Warnings': 1
+        }
+    }
+    try:
+        check_severity('HIGH', data_with_low_risk_only)
+    except SeverityException:
+        pytest.fail("SeverityException was raised unexpectedly!")
+
+
+def test_invalid_severity_flag(risk_data):
+    with pytest.raises(ValueError):
+        check_severity('invalid', risk_data)
+
+
+def test_exception_on_all_flag(risk_data):
+    with pytest.raises(SeverityException):
+        check_severity('ALL', risk_data)
+
+
+def test_no_exception_on_all_flag_when_empty():
+    data_with_no_risks = {
+        "securityTests": {
+            'DidNotRun': 0,
+            'Duration': 0,
+            'Findings': 5,
+            'Passed': 27,
+            'RisksCounter': {'Critical': 0, 'High': 0, 'Low': 0, 'Medium': 0},
+            'Warnings': 1
+        }
+    }
+    try:
+        check_severity('ALL', data_with_no_risks)
+    except SeverityException:
+        pytest.fail("SeverityException was raised unexpectedly!")