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

pyntcli/__init__.py

@@ -1 +1,20 @@
-__version__ = "0.1.93"
+__version__ = "0.1.94"
+
+try:
+    from logzio import sender
+    import logging
+
+
+    def get_silent_logger():
+        logger = logging.getLogger("silent_logger")
+        logger.addHandler(logging.NullHandler())
+        return logger
+
+
+    def patched_get_stdout_logger(debug):
+        return get_silent_logger()
+
+
+    sender.get_stdout_logger = patched_get_stdout_logger
+except Exception as e:
+    pass

pyntcli/commands/pynt_cmd.py

@@ -1,15 +1,22 @@
 import argparse
+import requests
 from typing import Dict, List
+from datetime import datetime, timedelta
 from pyntcli import __version__ as cli_version
 from pyntcli.analytics import send as analytics
 from pyntcli.transport import pynt_requests
 from pyntcli.ui import ui_thread
+from pyntcli.ui import prompt
+from pyntcli.store import CredStore, StateStore
+import pyntcli.log.log as log
 
 from requests.exceptions import SSLError, HTTPError
 
 
 from . import command, listen, postman, root, sub_command, id_command, newman, har, burp
 
+logger = log.get_logger()
+
 avail_sub_commands = [
     postman.PostmanSubCommand("postman"),
     id_command.PyntShowIdCommand("pynt-id"),
@@ -20,11 +27,17 @@ avail_sub_commands = [
     burp.BurpCommand("burp"),
 ]
 
+commands_without_app_id = ["postman", "pynt-id"]
+
 
 class PyntCommandException(Exception):
     pass
 
 
+class UserAbortedException(Exception):
+    pass
+
+
 class BadArgumentsException(PyntCommandException):
     pass
 
@@ -48,18 +61,22 @@ def check_is_latest_version(current_version):
             ui_thread.print(ui_thread.PrinterText("""Pynt CLI new version is available, upgrade now with:
 python3 -m pip install --upgrade pyntcli""", ui_thread.PrinterText.WARNING))
     except SSLError:
-        ui_thread.print(ui_thread.PrinterText("""Error: Unable to check if Pynt CLI version is up-to-date due to VPN/proxy. Run Pynt with --insecure to fix.""", ui_thread.PrinterText.WARNING))
+        ui_thread.print(ui_thread.PrinterText(
+            """Error: Unable to check if Pynt CLI version is up-to-date due to VPN/proxy. Run Pynt with --insecure to fix.""", ui_thread.PrinterText.WARNING))
     except HTTPError:
-        ui_thread.print("""Unable to check if Pynt CLI version is up-to-date""")
+        ui_thread.print(
+            """Unable to check if Pynt CLI version is up-to-date""")
     except Exception as e:
-        ui_thread.print(ui_thread.PrinterText("""We could not check for updates.""", ui_thread.PrinterText.WARNING))
+        ui_thread.print(ui_thread.PrinterText(
+            """We could not check for updates.""", ui_thread.PrinterText.WARNING))
         pass
 
 
 class PyntCommand:
     def __init__(self) -> None:
         self.base: root.BaseCommand = root.BaseCommand()
-        self.sub_commands: Dict[str, sub_command.PyntSubCommand] = {sc.get_name(): sc for sc in avail_sub_commands}
+        self.sub_commands: Dict[str, sub_command.PyntSubCommand] = {
+            sc.get_name(): sc for sc in avail_sub_commands}
         self._start_command()
 
     def _start_command(self):
@@ -77,15 +94,93 @@ class PyntCommand:
         command = getattr(args, "command")
         if not command in self.sub_commands:
             raise NoSuchCommandException()
-
+        check_is_latest_version(cli_version)
+        analytics.emit(analytics.CLI_START)
         if "host_ca" in args and args.host_ca:
             pynt_requests.add_host_ca(args.host_ca)
 
         if "insecure" in args and args.insecure:
             pynt_requests.disable_tls_termination()
 
-        check_is_latest_version(cli_version)
-        analytics.emit(analytics.CLI_START)
-
         self.base.run_cmd(args)
+
+        # Some args are recommended/required for business-plan users.
+        # These can be verified only after running 'self.base.run_cmd()' where the login occurs
+        is_business_plan_user = self._is_business_plan_user()
+        try:
+            self._post_login_args_validation(
+                args, command, is_business_plan_user)
+        except UserAbortedException as e:
+            ui_thread.print("Aborting...")
+            return
+
         self.sub_commands[command].run_cmd(args)
+
+    def _is_business_plan_user(self):
+        """ 
+        A workaround for checking whether it's a free tier user or not
+        by try accessing api.pynt.io/v1/config (it's a free tier if the user can't access it)
+
+        Returns:
+            bool: True if it's a business-plan user
+        """
+        saas_url = "https://api.pynt.io/v1/config"
+        headers = {}
+        with CredStore() as store:
+            access_keys = store.get_access_token()
+            token_type = store.get_token_type()
+            headers["Authorization"] = f"{token_type} {access_keys}"
+        try:
+            response = requests.get(saas_url, headers=headers, timeout=10)
+
+            if response.status_code == requests.codes.ok:
+                return True
+            else:
+                return False
+        except requests.exceptions.RequestException as e:
+            # Catch all other request exceptions
+            logger.debug(f"An error occurred when accessing '{saas_url}': {e}")
+            return False
+
+    def _post_login_args_validation(self, args: argparse.Namespace, command: str, is_business_plan_user: bool):
+        # Confirm not using application-id flag if applicable
+        if getattr(args, "application_id") or command in commands_without_app_id:
+            return
+
+        # For a business user, it's recommended to always provide the application-id.
+        if is_business_plan_user:
+            if getattr(args, "yes") or self._is_missing_app_id_confirmed():
+                return
+            else:
+                raise UserAbortedException()
+
+    def _is_missing_app_id_confirmed(self) -> bool:
+        """
+        Ask for the user's confirmation to continue if he/she wasn't asked in the last week.
+
+        Returns:
+            bool: True if the user confirms or he/she has confirmed once in the last week. Otherwise, returns False
+        """
+        with StateStore() as state_store:
+            current_time = datetime.now()
+            prompts_history = state_store.get_prompts_history()
+            last_confirmed = prompts_history.get(
+                "missing_app_id", {}).get("last_confirmation", "")
+            if last_confirmed:
+                # Calculate the time delta
+                parsed_datetime = datetime.strptime(
+                    last_confirmed, "%Y-%m-%d %H:%M:%S")
+                difference = current_time - parsed_datetime
+                if difference < timedelta(days=7):
+                    return True
+
+            if prompt.confirmation_prompt_with_timeout("Application ID is missing. Use the '--application-id' flag to provide it.\n" +
+                                          "Without an Application ID, the scan will not be associated with your application.\n" +
+                                          "The Application ID can be fetched from https://app.pynt.io/dashboard/applications.\n" +
+                                          "Do you want to continue without associating the scan?", default="yes", timeout=15):
+                prompts_history["missing_app_id"] = {"last_confirmation": current_time.strftime("%Y-%m-%d %H:%M:%S")}
+                state_store.put_prompts_history(
+                    prompts_history)
+                return True
+
+        return False

pyntcli/commands/root.py

@@ -63,17 +63,27 @@ class BaseCommand:
             action="store_true",
             help="use when target uses self signed certificates",
         )
-        parser.add_argument("--dev-flags", type=str, default="", help=argparse.SUPPRESS)
+        parser.add_argument("--dev-flags", type=str,
+                            default="", help=argparse.SUPPRESS)
         parser.add_argument("--host-ca", type=str, default="")
         parser.add_argument("--transport-config", type=str, default="")
-        parser.add_argument("--application-id", type=str, default="", required=False)
+        parser.add_argument("--application-id", type=str,
+                            default="", required=False)
         parser.add_argument("--proxy", type=str, default="", required=False)
+        parser.add_argument(
+            "--yes",
+            "-y",
+            default=False,
+            required=False,
+            action="store_true",
+            help="Automatically answer yes to all prompts and run non-interactively ",
+        )
         parser.add_argument(
             "--verbose",
             default=False,
             required=False,
             action="store_true",
-            help="use to get more detailed execution information",
+            help="Use to get more detailed execution information",
         )
         parser.add_argument(
             "--use-docker-native",

pyntcli/store/__init__.py

@@ -1 +1 @@
-from .store import CredStore
+from .store import CredStore, StateStore

pyntcli/store/store.py

@@ -22,7 +22,8 @@ class Store():
 
         if not os.path.exists(self.file_location):
             with open(self.file_location, "w") as f:
-                self.connector = self._connector_type(self._connector_type.default_value())
+                self.connector = self._connector_type(
+                    self._connector_type.default_value())
                 return
 
         with open(self.file_location, "r+") as f:
@@ -39,6 +40,9 @@ class Store():
     def get_path(self):
         return self.file_location
 
+    def _get_default_store_dir(self):
+        return os.path.join(os.path.expanduser("~"), ".pynt")
+
     def __enter__(self):
         self._get_file_data()
         return self
@@ -50,14 +54,31 @@ class Store():
 
 class CredStore(Store):
     def __init__(self) -> None:
-        dir = ".pynt"
-        super().__init__(file_location=os.path.join(os.path.expanduser("~"), dir, "creds.json"),
+        pynt_dir = super()._get_default_store_dir()
+        super().__init__(file_location=os.path.join(pynt_dir, "creds.json"),
                          connector_type=JsonStoreConnector)
 
     def get_access_token(self):
         return self.get("token")["access_token"]
 
+    def get_token_type(self):
+        return self.get("token")["token_type"]
+
     def get_tokens(self):
         all_tokens = self.get("token")
-        token_to_json_string = '{"token":' + str(all_tokens).replace("\'", "\"") + "}"
+        token_to_json_string = '{"token":' + \
+            str(all_tokens).replace("\'", "\"") + "}"
         return token_to_json_string
+
+
+class StateStore(Store):
+    def __init__(self) -> None:
+        pynt_dir = super()._get_default_store_dir()
+        super().__init__(file_location=os.path.join(pynt_dir, "state.json"),
+                         connector_type=JsonStoreConnector)
+
+    def get_prompts_history(self):
+        return self.get("prompts_history") or {}
+
+    def put_prompts_history(self, value):
+        return self.put("prompts_history", value)

pyntcli/ui/prompt.py

@@ -0,0 +1,53 @@
+import signal
+from pyntcli.ui import ui_thread
+
+
+class TimeoutExpired(Exception):
+    pass
+
+
+def _timeout_handler(signum, frame):
+    raise TimeoutExpired
+
+
+def confirmation_prompt_with_timeout(question, default, timeout=10):
+    """Prompt the user with a confirmation question (Yes/No) and return True for 'yes' or False for 'no'.
+
+    Args:
+        question (str): The question to present to the user.
+        default (str): The default answer if the user just presses Enter. It should be 'yes' or 'no'.
+        timeout (int): The timeout in seconds. If the user doesn't respond within the timeout, the default answer will be selected.
+
+    Returns:
+        bool: True if the answer is 'yes', False if the answer is 'no'.
+    """
+    valid = {"yes": True, "y": True, "no": False, "n": False}
+
+    if default is None:
+        prompt = " [y/n] "
+    elif default == "yes":
+        prompt = " [Y/n] "
+    elif default == "no":
+        prompt = " [y/N] "
+    else:
+        raise ValueError(f"Invalid default answer: '{default}'")
+
+    signal.signal(signal.SIGALRM, _timeout_handler)
+    signal.alarm(timeout)
+
+    try:
+        while True:
+            choice = input(question + prompt).strip().lower()
+
+            if choice == "" and default:  # Only 'Enter' with default will continue
+                return valid[default]
+            elif choice in valid:
+                return valid[choice]
+            else:
+                ui_thread.print("Please respond with 'yes' or 'no' (or 'y' or 'n').")
+    except TimeoutExpired:
+        ui_thread.print(ui_thread.PrinterText(f"\nNo response within {timeout} seconds, defaulting to {'Yes' if default == 'yes' else 'No'}.\n" +
+                        "You can use '--yes' flag for auto-confirm", ui_thread.PrinterText.WARNING))
+        return valid[default]
+    finally:
+        signal.alarm(0)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyntcli
-Version: 0.1.93
+Version: 0.1.94
 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

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

@@ -1,7 +1,7 @@
 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/__init__.py,sha256=5NPBuxeQo_sfTOMuw7XfyXg8HYg3BMvDd69Cm_HznpU,402
 pyntcli/main.py,sha256=HORWY5AEK5xmEQybUZBW9ZEDIga8vSnqDfmAR3TY9F0,6502
 pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/analytics/send.py,sha256=9TRAEoPbv4rWOZfcNaGanrRJAFvNs39P-uKSl49GcQE,3679
@@ -15,8 +15,8 @@ pyntcli/commands/id_command.py,sha256=UBEgMIpm4vauTCsKyixltiGUolNg_OfHEJvJ_i5BpJ
 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/pynt_cmd.py,sha256=T7jee0yw67Zth3lTkSDInCLnbu_IhpNqb7GqnTiTceQ,7012
+pyntcli/commands/root.py,sha256=4yeGlTeL3Xu2xlSbMYhhaDwUnP29sF2Y2sxq8E1P3lo,3986
 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=csZHQ2Xbdh-_KX-yIVrnaeNsT0NbuS-ej6kND3CxD_w,4414
@@ -25,20 +25,22 @@ 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=R2iq4Gbbkxg3OEJfJwUp-vroHf9-pJ6HWFqvWucPsGA,18965
-pyntcli/store/__init__.py,sha256=xuS9OB21F6B1sUx5XPGxz_6WpG6-KTMbuq50RrZS5OY,29
+pyntcli/store/__init__.py,sha256=1fP8cEAQCF_myja3gnhHH9FEqtBiOJ-2aBmUXSKBdFA,41
 pyntcli/store/json_connector.py,sha256=UGs3uORw3iyn0YJ8kzab-veEZToA6d-ByXYuqEleWsA,560
-pyntcli/store/store.py,sha256=4YqmcHRltbbSWewKQoOZH8QdyMFs6i3L8o2bXY_YvSw,1909
+pyntcli/store/store.py,sha256=ZLSe0WAjHDp8cSt4BBFDkPGRux4cgOo5UfF7V4naM7U,2559
 pyntcli/store/store_connector.py,sha256=w4LzcpRZesUZL1f63RmLlWEFRtJ6Y6rcS6PkkGtO4MA,357
 pyntcli/transport/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/transport/pynt_requests.py,sha256=C7OPvcKkRTcxSYuyiWKE59KgA9sRX0d6fm1wnopAmPo,1719
 pyntcli/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/ui/progress.py,sha256=RrnO_jJNunoyupylakmWmHOEPw3lh99OHpKBzL6OBiE,1008
+pyntcli/ui/prompt.py,sha256=6URUWCGpKUPhluWTYVaXvCQuABjiFfJSE3rfRJMvAS0,1838
 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/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,,
+tests/commands/test_pynt_cmd.py,sha256=BjGFCFACcSziLrNA6_27t6TjSmvdu54wx9njwLpRSJY,8379
+pyntcli-0.1.94.dist-info/METADATA,sha256=m__wWpEYNkwzQMqrCy14feFm2kT8jxPRamGVZSk1cJ4,460
+pyntcli-0.1.94.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+pyntcli-0.1.94.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
+pyntcli-0.1.94.dist-info/top_level.txt,sha256=64XSgBzSpgwjYjEKHZE7q3JH2a816zEeyZBXfJi3AKI,42
+pyntcli-0.1.94.dist-info/RECORD,,

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

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

tests/commands/test_pynt_cmd.py

@@ -0,0 +1,178 @@
+import unittest
+import argparse
+from datetime import datetime, timedelta
+from unittest.mock import patch, MagicMock
+from pyntcli.commands.pynt_cmd import PyntCommand, UserAbortedException
+import requests
+
+
+class TestIsBusinessPlanUser(unittest.TestCase):
+
+    @patch('pyntcli.commands.pynt_cmd.requests.get')
+    @patch('pyntcli.commands.pynt_cmd.CredStore')
+    def test_is_business_plan_user_success(self, mock_cred_store, mock_requests_get):
+        # Mock the CredStore context manager
+        mock_store_instance = MagicMock()
+        mock_store_instance.get_access_token.return_value = 'fake_token'
+        mock_store_instance.get_token_type.return_value = 'Bearer'
+        mock_cred_store.return_value.__enter__.return_value = mock_store_instance
+
+        # Mock the requests.get response
+        mock_response = MagicMock()
+        mock_response.status_code = 200
+        mock_requests_get.return_value = mock_response
+
+        # Instantiate the class and call the method
+        instance = PyntCommand()
+        result = instance._is_business_plan_user()
+
+        # Assert the result
+        self.assertTrue(result)
+
+    @patch('pyntcli.commands.pynt_cmd.requests.get')
+    @patch('pyntcli.commands.pynt_cmd.CredStore')
+    def test_is_business_plan_user_failure(self, mock_cred_store, mock_requests_get):
+        # Mock the CredStore context manager
+        mock_store_instance = MagicMock()
+        mock_store_instance.get_access_token.return_value = 'fake_token'
+        mock_store_instance.get_token_type.return_value = 'Bearer'
+        mock_cred_store.return_value.__enter__.return_value = mock_store_instance
+
+        # Mock the requests.get response
+        mock_response = MagicMock()
+        mock_response.status_code = 403
+        mock_requests_get.return_value = mock_response
+
+        # Instantiate the class and call the method
+        instance = PyntCommand()
+        result = instance._is_business_plan_user()
+
+        # Assert the result
+        self.assertFalse(result)
+
+    @patch('pyntcli.commands.pynt_cmd.requests.get')
+    @patch('pyntcli.commands.pynt_cmd.CredStore')
+    def test_is_business_plan_user_exception(self, mock_cred_store, mock_requests_get):
+        # Mock the CredStore context manager
+        mock_store_instance = MagicMock()
+        mock_store_instance.get_access_token.return_value = 'fake_token'
+        mock_store_instance.get_token_type.return_value = 'Bearer'
+        mock_cred_store.return_value.__enter__.return_value = mock_store_instance
+
+        # Mock the requests.get to raise an exception
+        mock_requests_get.side_effect = requests.exceptions.RequestException
+
+        # Instantiate the class and call the method
+        instance = PyntCommand()
+        result = instance._is_business_plan_user()
+
+        # Assert the result
+        self.assertFalse(result)
+
+    @patch.object(PyntCommand, '_is_missing_app_id_confirmed', return_value=True)
+    def test_post_login_args_validation_missing_app_id_user_confirmed(self, _):
+        # Instantiate the class and call the method with valid arguments
+        instance = PyntCommand()
+        try:
+            result = instance._post_login_args_validation(argparse.Namespace(
+                application_id="", yes=False), "non-postman-command", True)
+        except Exception as e:
+            self.fail(f"Unexpected exception raised {e}")
+
+    @patch.object(PyntCommand, '_is_missing_app_id_confirmed', return_value=False)
+    def test_post_login_args_validation_missing_app_id_yes_flag_used(self, _):
+        # Instantiate the class and call the method with valid arguments
+        instance = PyntCommand()
+        try:
+            result = instance._post_login_args_validation(argparse.Namespace(
+                application_id="", yes=True), "non-postman-command", True)
+        except Exception as e:
+            self.fail(f"Unexpected exception raised {e}")
+
+    @patch.object(PyntCommand, '_is_missing_app_id_confirmed', return_value=False)
+    def test_post_login_args_validation_missing_app_id_user_aborted(self, _):
+        # Instantiate the class and call the method with valid arguments
+        instance = PyntCommand()
+        with self.assertRaises(UserAbortedException):
+            result = instance._post_login_args_validation(argparse.Namespace(
+                application_id="", yes=False), "non-postman-command", True)
+
+    @patch.object(PyntCommand, '_is_missing_app_id_confirmed', return_value=False)
+    def test_post_login_args_validation_missing_app_id_free_tier(self, _):
+        # Instantiate the class and call the method with valid arguments
+        instance = PyntCommand()
+        try:
+            result = instance._post_login_args_validation(argparse.Namespace(
+                application_id="", yes=False), "non-postman-command", False)
+        except Exception as e:
+            self.fail(f"Unexpected exception raised {e}")
+
+    @patch.object(PyntCommand, '_is_missing_app_id_confirmed', return_value=False)
+    def test_post_login_args_validation_missing_app_id_not_required(self, _):
+        # Instantiate the class and call the method with valid arguments
+        instance = PyntCommand()
+        try:
+            result = instance._post_login_args_validation(
+                argparse.Namespace(application_id="", yes=False), "postman", True)
+            result = instance._post_login_args_validation(
+                argparse.Namespace(application_id="", yes=False), "pynt-id", True)
+        except Exception as e:
+            self.fail(f"Unexpected exception raised {e}")
+
+    @patch('pyntcli.commands.pynt_cmd.prompt.confirmation_prompt_with_timeout', return_value=True)
+    @patch('pyntcli.commands.pynt_cmd.StateStore')
+    def test_missing_app_id_confirmation_confirmation_expired_user_confirms(self, mock_state_store, _):
+        long_time_ago = datetime.now() - timedelta(days=100)
+        # Mock the CredStore context manager
+        mock_store_instance = MagicMock()
+        mock_store_instance.get_prompts_history.return_value = {"missing_app_id": {
+            "last_confirmation": long_time_ago.strftime("%Y-%m-%d %H:%M:%S")}}
+        mock_store_instance.put_prompts_history.return_value = None
+        mock_state_store.return_value.__enter__.return_value = mock_store_instance
+
+        # Instantiate the class and call the method with app_id present
+        instance = PyntCommand()
+        result = instance._is_missing_app_id_confirmed()
+
+        # Assert the result
+        self.assertTrue(result)
+
+    @patch('pyntcli.commands.pynt_cmd.prompt.confirmation_prompt_with_timeout', return_value=False)
+    @patch('pyntcli.commands.pynt_cmd.StateStore')
+    def test_missing_app_id_confirmation_confirmation_expired_user_aborts(self, mock_state_store, _):
+        long_time_ago = datetime.now() - timedelta(days=100)
+        # Mock the CredStore context manager
+        mock_store_instance = MagicMock()
+        mock_store_instance.get_prompts_history.return_value = {"missing_app_id": {
+            "last_confirmation": long_time_ago.strftime("%Y-%m-%d %H:%M:%S")}}
+        mock_store_instance.put_prompts_history.return_value = None
+        mock_state_store.return_value.__enter__.return_value = mock_store_instance
+
+        # Instantiate the class and call the method with app_id present
+        instance = PyntCommand()
+        result = instance._is_missing_app_id_confirmed()
+
+        # Assert the result
+        self.assertFalse(result)
+
+    @patch('pyntcli.commands.pynt_cmd.prompt.confirmation_prompt_with_timeout', return_value=False)
+    @patch('pyntcli.commands.pynt_cmd.StateStore')
+    def test_missing_app_id_confirmation_user_already_confirmed(self, mock_state_store, _):
+        recently = datetime.now() - timedelta(hours=1)
+        # Mock the CredStore context manager
+        mock_store_instance = MagicMock()
+        mock_store_instance.get_prompts_history.return_value = {"missing_app_id": {
+            "last_confirmation": recently.strftime("%Y-%m-%d %H:%M:%S")}}
+        mock_store_instance.put_prompts_history.return_value = None
+        mock_state_store.return_value.__enter__.return_value = mock_store_instance
+
+        # Instantiate the class and call the method with app_id present
+        instance = PyntCommand()
+        result = instance._is_missing_app_id_confirmed()
+
+        # Assert the result
+        self.assertTrue(result)
+
+
+if __name__ == '__main__':
+    unittest.main()