pyntcli 0.1.30__py3-none-any.whl → 0.1.31__py3-none-any.whl

build/lib/build/lib/build/lib/tests/auth/test_login.py

@@ -0,0 +1,96 @@
+from urllib.parse import urlparse
+import json
+from _pytest.monkeypatch import monkeypatch
+import pytest
+import requests
+import requests_mock
+import datetime
+import jwt
+import os
+from cryptography.hazmat.primitives.asymmetric import rsa
+from cryptography.hazmat.primitives import serialization
+
+from pyntcli.auth.login import Login, Timeout, InvalidTokenInEnvVarsException, is_jwt_expired, should_login, PYNT_CREDENTIALS
+from pyntcli.store import CredStore
+
+
+class TestLogin():
+    @pytest.fixture
+    def mock_webbrowser(self, mocker):
+        try:
+            mocker.patch("webbrowser.open", return_value=None)
+            yield
+        finally:
+            pass
+    
+    def get_request_url_parameters(self, req: requests.PreparedRequest) :
+        u = req.url
+        parsed_url = urlparse(u)
+        return parsed_url.query
+
+    def poll_matcher(self, request: requests.PreparedRequest):
+        assert "request_id" in self.get_request_url_parameters(request)
+
+        resp = requests.Response()
+        self.login_request_cnt += 1
+        if self.login_request_cnt < 2:
+            resp.status_code = 404
+            return resp
+        
+        resp.status_code = 200
+        resp._content = json.dumps({"token": "testToken"}).encode()
+        return resp
+
+    def test_login(self, mock_webbrowser, mock_sleep, mock_expanduser):
+        l = Login()
+        self.login_request_cnt = 0
+        with requests_mock.mock() as m: 
+            m.add_matcher(self.poll_matcher)
+            l.login() 
+
+        assert self.login_request_cnt == 2
+        c = CredStore()
+        assert c.get("token") == {"token": "testToken"}
+
+    def test_login_timeout(self, mock_webbrowser, mock_sleep, mock_expanduser):
+        l = Login() 
+        l.login_wait_period = 0
+        self.login_request_cnt = 0
+        with pytest.raises(Timeout):
+            with requests_mock.mock() as m:
+                m.add_matcher(self.poll_matcher)
+                l.get_token_using_request_id("some_id")
+    
+    
+    def test_is_jwt_expired(self):
+
+        private_key = rsa.generate_private_key(
+            public_exponent=65537,
+            key_size=2048
+        ).private_bytes(encoding=serialization.Encoding.PEM,
+                        format=serialization.PrivateFormat.PKCS8, 
+                        encryption_algorithm=serialization.NoEncryption())
+
+
+        token_data = {
+                "exp": int((datetime.datetime.now() - datetime.timedelta(days=1)).timestamp())
+                }
+
+        token = jwt.encode(token_data, private_key.decode(), algorithm="RS256").decode("utf-8")
+        assert is_jwt_expired(token) == True
+
+        token_data = {
+                "exp": int((datetime.datetime.now() + datetime.timedelta(days=1)).timestamp())
+                }
+
+        token = jwt.encode(token_data, private_key.decode(), algorithm="RS256").decode("utf-8")
+        assert is_jwt_expired(token) == False
+
+    def test_login_using_env_vars(self, mocker, mock_expanduser):
+        creds = json.dumps({"token": {"refresh_token": "some data"}})
+        mocker.patch.dict(os.environ, {PYNT_CREDENTIALS: creds})
+        assert should_login() == False 
+        
+        os.environ[PYNT_CREDENTIALS] = "some bad credentials"
+        with pytest.raises(InvalidTokenInEnvVarsException):
+            should_login()

build/lib/build/lib/build/lib/tests/conftest.py

@@ -0,0 +1,24 @@
+import tempfile
+import pytest
+import shutil
+
+import sys 
+sys.path.append("../")
+
+@pytest.fixture
+def mock_expanduser( mocker):
+    dir = tempfile.mkdtemp() 
+    try:
+        mocker.patch("os.path.expanduser", return_value=dir)
+        yield
+    finally:
+        shutil.rmtree(dir)
+
+@pytest.fixture
+def mock_sleep( mocker):
+    try:
+        mocker.patch("time.sleep", return_value=None)
+        yield
+    finally:
+        pass
+

build/lib/build/lib/build/lib/tests/store/test_cred_store.py

@@ -0,0 +1,11 @@
+import tempfile
+import shutil
+import pytest
+
+from pyntcli.store import CredStore
+
+class TestCredStore():
+    def test_get_credentials(self, mock_expanduser):
+        c = CredStore()
+        c.put("data", "value")
+        assert c.get("data") == "value" 

build/lib/build/lib/pyntcli/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.30"
+__version__ = "0.1.31"

build/lib/build/lib/pyntcli/commands/proxy.py

@@ -6,6 +6,7 @@ from http import HTTPStatus
 import requests
 import time
 from subprocess import Popen, PIPE
+from build.lib.pyntcli.commands import proxy
 
 from pyntcli.store.store import CredStore
 from pyntcli.pynt_docker import pynt_container
@@ -143,3 +144,4 @@ class ProxyCommand(sub_command.PyntSubCommand):
                 f.write(report)
 
         webbrowser.open("file://{}".format(report_path))
+        proxy_docker.stop()

build/lib/pyntcli/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.30"
+__version__ = "0.1.31"

build/lib/pyntcli/commands/proxy.py

@@ -144,4 +144,3 @@ class ProxyCommand(sub_command.PyntSubCommand):
                 f.write(report)
 
         webbrowser.open("file://{}".format(report_path))
-        proxy_docker.stop()

pyntcli/__init__.py

@@ -1 +1 @@
-__version__ = "0.1.30"
+__version__ = "0.1.31"

pyntcli/commands/proxy.py

@@ -6,7 +6,6 @@ from http import HTTPStatus
 import requests
 import time
 from subprocess import Popen, PIPE
-from build.lib.pyntcli.commands import proxy
 
 from pyntcli.store.store import CredStore
 from pyntcli.pynt_docker import pynt_container

{pyntcli-0.1.30.dist-info → pyntcli-0.1.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyntcli
-Version: 0.1.30
+Version: 0.1.31
 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.30.dist-info → pyntcli-0.1.31.dist-info}/RECORD

@@ -1,4 +1,32 @@
-build/lib/build/lib/pyntcli/__init__.py,sha256=2GUJJyX8g8EAXKUqyj7DGVzG-jNXOGaqVSWilvGYuX8,23
+build/lib/build/lib/build/lib/pyntcli/__init__.py,sha256=i-fDEsQ0iAiPKXFaj9eERDqcxl3BqNnavaCEqpNxmVI,23
+build/lib/build/lib/build/lib/pyntcli/main.py,sha256=_K2y9rj5Ysb2GucX_C8c3hH7rvFAGLhRT7GnQhoiu4M,3073
+build/lib/build/lib/build/lib/pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+build/lib/build/lib/build/lib/pyntcli/analytics/send.py,sha256=2zcncrCkC8F8GoTbFNZcQtYg5og1z1bA-Gng6CMb7pY,1862
+build/lib/build/lib/build/lib/pyntcli/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+build/lib/build/lib/build/lib/pyntcli/auth/login.py,sha256=6y1GzvBRihCOWIjMpJCZ8leKNHTsdpVKIjDAizhxbNg,4703
+build/lib/build/lib/build/lib/pyntcli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+build/lib/build/lib/build/lib/pyntcli/commands/har.py,sha256=CA61j_lO4Bi1OpDeQF2Krg835yGal8Ld8D-XXAeRhxw,2950
+build/lib/build/lib/build/lib/pyntcli/commands/id_command.py,sha256=2J5oEa39uyBMMSCRHzuKh3_11CeUEYi7YLoBk1yuu_I,942
+build/lib/build/lib/build/lib/pyntcli/commands/newman.py,sha256=H73-TYmHslAl4O-w3cH25FTVKYYka_d64-WHhDfri9U,4268
+build/lib/build/lib/build/lib/pyntcli/commands/postman.py,sha256=c65nTeg9RkgorzJP1ZE0a4B98DS6wvWicFHlEbacu4c,2558
+build/lib/build/lib/build/lib/pyntcli/commands/proxy.py,sha256=iiz_0DZcoXc2U-R-0vilZQKF-or4S8ytqpEueSzUpm8,7024
+build/lib/build/lib/build/lib/pyntcli/commands/pynt_cmd.py,sha256=tDGu3n-YwknSl8fMEv7eXPg_EuVbXLY2IWC2TwLzuFU,1412
+build/lib/build/lib/build/lib/pyntcli/commands/root.py,sha256=c8iM4Kyr3GU9DZyyE6wIWXTJjXeTioP-AWOcP_gTpkQ,2103
+build/lib/build/lib/build/lib/pyntcli/commands/sub_command.py,sha256=POEBodnU8vdfPRtagFG83QVbxvnVN4LIJov-mNi1jkw,375
+build/lib/build/lib/build/lib/pyntcli/commands/util.py,sha256=NqS9ZwMP0HrPqb6Ll5-olQn4qNL7xeRlzO1y6ncWLmU,719
+build/lib/build/lib/build/lib/pyntcli/pynt_docker/__init__.py,sha256=PQIOVxc7XXtMLfEX7ojgwf_Z3mmTllO3ZvzUZTPOxQY,30
+build/lib/build/lib/build/lib/pyntcli/pynt_docker/pynt_container.py,sha256=zveBFt0APHc5KnFDwV9qmGaBzmtIbP8rL9nVxYJgp54,5415
+build/lib/build/lib/build/lib/pyntcli/store/__init__.py,sha256=xuS9OB21F6B1sUx5XPGxz_6WpG6-KTMbuq50RrZS5OY,29
+build/lib/build/lib/build/lib/pyntcli/store/json_connector.py,sha256=UGs3uORw3iyn0YJ8kzab-veEZToA6d-ByXYuqEleWsA,560
+build/lib/build/lib/build/lib/pyntcli/store/store.py,sha256=-ecy9JnvoKtVfl8pznSUR42wnn4_BLnw_YSnmcztlOM,1676
+build/lib/build/lib/build/lib/pyntcli/store/store_connector.py,sha256=w4LzcpRZesUZL1f63RmLlWEFRtJ6Y6rcS6PkkGtO4MA,357
+build/lib/build/lib/build/lib/pyntcli/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+build/lib/build/lib/build/lib/pyntcli/ui/progress.py,sha256=UJY6YNJkml9K9fQbtCwcpxusfaX6oSrNHskYLbIO2Lg,737
+build/lib/build/lib/build/lib/pyntcli/ui/ui_thread.py,sha256=D_4PF89Rd97XTA5hd8fCSS2iY5T0PW0OemR-UykXtr8,4603
+build/lib/build/lib/build/lib/tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
+build/lib/build/lib/build/lib/tests/auth/test_login.py,sha256=M6JRFTQRZrL6M2-iph_r-aBSQMMiFDncQbVYeObBFYU,3296
+build/lib/build/lib/build/lib/tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
+build/lib/build/lib/pyntcli/__init__.py,sha256=i-fDEsQ0iAiPKXFaj9eERDqcxl3BqNnavaCEqpNxmVI,23
 build/lib/build/lib/pyntcli/main.py,sha256=_K2y9rj5Ysb2GucX_C8c3hH7rvFAGLhRT7GnQhoiu4M,3073
 build/lib/build/lib/pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 build/lib/build/lib/pyntcli/analytics/send.py,sha256=2zcncrCkC8F8GoTbFNZcQtYg5og1z1bA-Gng6CMb7pY,1862
@@ -9,7 +37,7 @@ build/lib/build/lib/pyntcli/commands/har.py,sha256=CA61j_lO4Bi1OpDeQF2Krg835yGal
 build/lib/build/lib/pyntcli/commands/id_command.py,sha256=2J5oEa39uyBMMSCRHzuKh3_11CeUEYi7YLoBk1yuu_I,942
 build/lib/build/lib/pyntcli/commands/newman.py,sha256=H73-TYmHslAl4O-w3cH25FTVKYYka_d64-WHhDfri9U,4268
 build/lib/build/lib/pyntcli/commands/postman.py,sha256=c65nTeg9RkgorzJP1ZE0a4B98DS6wvWicFHlEbacu4c,2558
-build/lib/build/lib/pyntcli/commands/proxy.py,sha256=iiz_0DZcoXc2U-R-0vilZQKF-or4S8ytqpEueSzUpm8,7024
+build/lib/build/lib/pyntcli/commands/proxy.py,sha256=XCza_gALmcnKyCQJFqnpzyBFduMO1Rtl57-50tT7NM4,7097
 build/lib/build/lib/pyntcli/commands/pynt_cmd.py,sha256=tDGu3n-YwknSl8fMEv7eXPg_EuVbXLY2IWC2TwLzuFU,1412
 build/lib/build/lib/pyntcli/commands/root.py,sha256=c8iM4Kyr3GU9DZyyE6wIWXTJjXeTioP-AWOcP_gTpkQ,2103
 build/lib/build/lib/pyntcli/commands/sub_command.py,sha256=POEBodnU8vdfPRtagFG83QVbxvnVN4LIJov-mNi1jkw,375
@@ -26,7 +54,7 @@ build/lib/build/lib/pyntcli/ui/ui_thread.py,sha256=D_4PF89Rd97XTA5hd8fCSS2iY5T0P
 build/lib/build/lib/tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
 build/lib/build/lib/tests/auth/test_login.py,sha256=M6JRFTQRZrL6M2-iph_r-aBSQMMiFDncQbVYeObBFYU,3296
 build/lib/build/lib/tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
-build/lib/pyntcli/__init__.py,sha256=2GUJJyX8g8EAXKUqyj7DGVzG-jNXOGaqVSWilvGYuX8,23
+build/lib/pyntcli/__init__.py,sha256=i-fDEsQ0iAiPKXFaj9eERDqcxl3BqNnavaCEqpNxmVI,23
 build/lib/pyntcli/main.py,sha256=_K2y9rj5Ysb2GucX_C8c3hH7rvFAGLhRT7GnQhoiu4M,3073
 build/lib/pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 build/lib/pyntcli/analytics/send.py,sha256=2zcncrCkC8F8GoTbFNZcQtYg5og1z1bA-Gng6CMb7pY,1862
@@ -37,7 +65,7 @@ build/lib/pyntcli/commands/har.py,sha256=CA61j_lO4Bi1OpDeQF2Krg835yGal8Ld8D-XXAe
 build/lib/pyntcli/commands/id_command.py,sha256=2J5oEa39uyBMMSCRHzuKh3_11CeUEYi7YLoBk1yuu_I,942
 build/lib/pyntcli/commands/newman.py,sha256=H73-TYmHslAl4O-w3cH25FTVKYYka_d64-WHhDfri9U,4268
 build/lib/pyntcli/commands/postman.py,sha256=c65nTeg9RkgorzJP1ZE0a4B98DS6wvWicFHlEbacu4c,2558
-build/lib/pyntcli/commands/proxy.py,sha256=XCza_gALmcnKyCQJFqnpzyBFduMO1Rtl57-50tT7NM4,7097
+build/lib/pyntcli/commands/proxy.py,sha256=8iJH0ApTYMpNO5EQoKQRDcTrJVbL5s7uvmPwWTFzUfw,7069
 build/lib/pyntcli/commands/pynt_cmd.py,sha256=tDGu3n-YwknSl8fMEv7eXPg_EuVbXLY2IWC2TwLzuFU,1412
 build/lib/pyntcli/commands/root.py,sha256=c8iM4Kyr3GU9DZyyE6wIWXTJjXeTioP-AWOcP_gTpkQ,2103
 build/lib/pyntcli/commands/sub_command.py,sha256=POEBodnU8vdfPRtagFG83QVbxvnVN4LIJov-mNi1jkw,375
@@ -54,7 +82,7 @@ build/lib/pyntcli/ui/ui_thread.py,sha256=D_4PF89Rd97XTA5hd8fCSS2iY5T0PW0OemR-Uyk
 build/lib/tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
 build/lib/tests/auth/test_login.py,sha256=M6JRFTQRZrL6M2-iph_r-aBSQMMiFDncQbVYeObBFYU,3296
 build/lib/tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
-pyntcli/__init__.py,sha256=2GUJJyX8g8EAXKUqyj7DGVzG-jNXOGaqVSWilvGYuX8,23
+pyntcli/__init__.py,sha256=i-fDEsQ0iAiPKXFaj9eERDqcxl3BqNnavaCEqpNxmVI,23
 pyntcli/main.py,sha256=_K2y9rj5Ysb2GucX_C8c3hH7rvFAGLhRT7GnQhoiu4M,3073
 pyntcli/analytics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pyntcli/analytics/send.py,sha256=2zcncrCkC8F8GoTbFNZcQtYg5og1z1bA-Gng6CMb7pY,1862
@@ -65,7 +93,7 @@ pyntcli/commands/har.py,sha256=CA61j_lO4Bi1OpDeQF2Krg835yGal8Ld8D-XXAeRhxw,2950
 pyntcli/commands/id_command.py,sha256=2J5oEa39uyBMMSCRHzuKh3_11CeUEYi7YLoBk1yuu_I,942
 pyntcli/commands/newman.py,sha256=H73-TYmHslAl4O-w3cH25FTVKYYka_d64-WHhDfri9U,4268
 pyntcli/commands/postman.py,sha256=c65nTeg9RkgorzJP1ZE0a4B98DS6wvWicFHlEbacu4c,2558
-pyntcli/commands/proxy.py,sha256=8iJH0ApTYMpNO5EQoKQRDcTrJVbL5s7uvmPwWTFzUfw,7069
+pyntcli/commands/proxy.py,sha256=iiz_0DZcoXc2U-R-0vilZQKF-or4S8ytqpEueSzUpm8,7024
 pyntcli/commands/pynt_cmd.py,sha256=tDGu3n-YwknSl8fMEv7eXPg_EuVbXLY2IWC2TwLzuFU,1412
 pyntcli/commands/root.py,sha256=c8iM4Kyr3GU9DZyyE6wIWXTJjXeTioP-AWOcP_gTpkQ,2103
 pyntcli/commands/sub_command.py,sha256=POEBodnU8vdfPRtagFG83QVbxvnVN4LIJov-mNi1jkw,375
@@ -82,8 +110,8 @@ pyntcli/ui/ui_thread.py,sha256=D_4PF89Rd97XTA5hd8fCSS2iY5T0PW0OemR-UykXtr8,4603
 tests/conftest.py,sha256=gToq5K74GtgeGQXjFvXSzMaE6axBYxAzcFG5XJPOXjI,427
 tests/auth/test_login.py,sha256=M6JRFTQRZrL6M2-iph_r-aBSQMMiFDncQbVYeObBFYU,3296
 tests/store/test_cred_store.py,sha256=_7-917EtNC9eKEumO2_lt-7KuDmCwOZFaowCm7DbA_A,254
-pyntcli-0.1.30.dist-info/METADATA,sha256=XfrJh7NFYfSxfcFY-45EYX6YlIAScJhCFJqlMRkMWuk,274
-pyntcli-0.1.30.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-pyntcli-0.1.30.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
-pyntcli-0.1.30.dist-info/top_level.txt,sha256=cfPl5WA9Rf7lwTBVIVy_tX9tJ2H9TFVYd_4L9cSUcXQ,36
-pyntcli-0.1.30.dist-info/RECORD,,
+pyntcli-0.1.31.dist-info/METADATA,sha256=N-wMulQPs1gC-4ZZ904mv5w7GBT3Ow3asEg2y1lMJJ8,274
+pyntcli-0.1.31.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
+pyntcli-0.1.31.dist-info/entry_points.txt,sha256=kcGmqAxXDttNk2EPRcqunc_LTVp61gzakz0v-GEE2SY,43
+pyntcli-0.1.31.dist-info/top_level.txt,sha256=cfPl5WA9Rf7lwTBVIVy_tX9tJ2H9TFVYd_4L9cSUcXQ,36
+pyntcli-0.1.31.dist-info/RECORD,,