recce-nightly 1.9.0.20250623__py3-none-any.whl → 1.10.0.20250625__py3-none-any.whl

{recce_nightly-1.9.0.20250623.dist-info → recce_nightly-1.10.0.20250625.dist-info}/RECORD

@@ -1,8 +1,9 @@
-recce/VERSION,sha256=h9fRG99db2LfCZiDVaiKOps9YxdL-lmHHrimg0QCuKs,15
+recce/VERSION,sha256=OPRTyx2o-EC6wUdTSX8AunguCb0Km1k-wtrnviGPr_k,16
 recce/__init__.py,sha256=yNb0QT-yoStex0VZALNJvUwtPLommoVCStcow31guqo,2392
 recce/artifact.py,sha256=tKQAHSrLRjiR3ppOI4sym8SxYiiLTuD3DPMYh4DWQdA,6506
-recce/cli.py,sha256=aitBalvi-dBhOmqwGCfjjvzdVYDTcVQl0nn4l6ktVFA,39627
-recce/config.py,sha256=fs22mpFj8CFIxftGbhFAV5xIsPLX2xNTwWSer3UYn5k,4658
+recce/cli.py,sha256=sb9h1Wp48cjp7EWv-lnodmSk78twSJA5YAvJ7z6-dAw,40710
+recce/config.py,sha256=A4CbKcIdwQ4A9Q2ba4riHUshVQw1D-qDelMOdlt2khU,4661
+recce/connect_to_cloud.py,sha256=_SkX2pdyXa9FNyaHvItyYVPF2nZxy2JnCyd_o_psh2Y,4750
 recce/core.py,sha256=3Nv2QWBGz3yj1QI0wyPNRaDj1rH80DsX2qRQn9ydpg0,10747
 recce/diff.py,sha256=L2_bzQ3__PO-0aeir8PHF8FvSOUmQ8WcDXgML1-mHdY,748
 recce/exceptions.py,sha256=SclQ678GrHGjw7p_ZFJ3vZaL_yMU5xABeIAm2u_W2bk,592
@@ -10,8 +11,8 @@ recce/git.py,sha256=8Eg-6NzL-KjA3rT-ibbAyaCwGlzV0JqH3yGikrJNMDA,2344
 recce/github.py,sha256=PEpM6ZRiinsPbXSWj4aJCKbZrN1jUXzpzAfJq_CGah4,7420
 recce/pull_request.py,sha256=aW0B1NE2LUKTam1S4TQ7smXB9KLE1DV8GnyBqNXA6j8,3832
 recce/run.py,sha256=LAjbWUF8loZ9cL25d_maQELwROHHmfaFn7iqFRy5O1o,13567
-recce/server.py,sha256=cp5XW0X1N314b5lOz75yU7rHMQZtlh_4oGZM9K6_CdE,20311
-recce/state.py,sha256=VPzKyEdmyBeZ45BQArB587QN5C3Sfr7O5Oh_JRBcKrc,30616
+recce/server.py,sha256=RFbZQP9RAsbtGEgtP-ltLc9ZvBZZM6T9qEn9nIwh3HY,20944
+recce/state.py,sha256=fOzy09rn96XqOQuA0RcMW-k2KpjNVH9FQhihEXLvlA0,30634
 recce/summary.py,sha256=Mbxvxr9KazR5o9icqhhjiGHsoAiWxQU4PdN7HytBJ1c,19154
 recce/adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 recce/adapter/base.py,sha256=T_JNeLHgiHSaegw-DbrvHOaYjMyZcjj2Qtg5cWh_fco,3548
@@ -23,12 +24,13 @@ recce/apis/check_api.py,sha256=KMCXSMl1qqzx2jQgRqCrD4j_cY3EHBbM3H2-t-6saAU,6227
 recce/apis/check_func.py,sha256=gktbCcyk3WGvWRJJ-wDnwv7NrIny2nTHWLl1-kdiVRo,4183
 recce/apis/run_api.py,sha256=eOaxOxXDkH59uqGCd4blld7edavUx7JU_DCd2WAYrL8,3416
 recce/apis/run_func.py,sha256=6wC8TDU-h7TLr2VZH7HNsWaUVlQ9HBN5N_dwqfi4lMY,7440
-recce/data/404.html,sha256=2jEAEilUmYNl2p4bzngbU0WR5YZZ7vuv3rATHhy7IKU,26574
+recce/data/404.html,sha256=UdzrC9_oa4co5DMQq0RUcPz7DGbXKyqnSi6yVhwL9Do,26574
+recce/data/auth_callback.html,sha256=2QnhHRc2eLPiaxCYl-HndNsGDGmM6jZAMbqoMlehal4,94783
 recce/data/favicon.ico,sha256=B2mBumUOnzvUrXrqNkrc5QfdDXjzEXRcWkWur0fJ6sM,2565
-recce/data/index.html,sha256=BavWLVZ03JFLaUxaIJjFFYff2qTGxldM4x0jiEPAim8,42155
-recce/data/index.txt,sha256=XA3HBkMkVX3dqELQKElyiviGFm12dIW_C-TsaufLSb4,4462
-recce/data/_next/static/WrRUb3nV8BhAZG_R8kVma/_buildManifest.js,sha256=sjy6FmOqWlkXWmod2oXwqKv7L582_MTycs9THCE6G-c,224
-recce/data/_next/static/WrRUb3nV8BhAZG_R8kVma/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
+recce/data/index.html,sha256=O9LAzdl-b1ys9yxjr2mVqFZ6D1zYc0fNZnegI5IhTu8,42155
+recce/data/index.txt,sha256=gDl4Z1wxKosGQ_29sYbNzFLziX3C9ZDy84O5VoLG-XU,4462
+recce/data/_next/static/abCX3x3UoIdRLEDWxx4xd/_buildManifest.js,sha256=sjy6FmOqWlkXWmod2oXwqKv7L582_MTycs9THCE6G-c,224
+recce/data/_next/static/abCX3x3UoIdRLEDWxx4xd/_ssgManifest.js,sha256=Z49s4suAsf5y_GfnQSvm4qtq2ggxEbZPfEDTXjy6XgA,80
 recce/data/_next/static/chunks/181-acc61ddada3bc0ca.js,sha256=2zRVkcaI906lYW_aM2mNqnt2HbLxmUaYFjft7eNcuTQ,310191
 recce/data/_next/static/chunks/1bff33f1-1ef85cf5e658a751.js,sha256=wTH-_PbE1XdP-wmGaQStF5y2yH6XcZXeYX_hDaGMfh4,153777
 recce/data/_next/static/chunks/217-879a84d70f7a907c.js,sha256=kNJkLcwaIlWnQY7akbrFimbcHHdK6lMVTbTYCNKXIjc,124102
@@ -45,7 +47,7 @@ recce/data/_next/static/chunks/7a8a3e83-edf6dc64b5d5f0a5.js,sha256=NfgBK9op5SzWm
 recce/data/_next/static/chunks/7f27ae6c-d5f0438edd5c2a5b.js,sha256=ewMLioJbTUUDeqV0J9NDKE0Bj-XCFg8QqMLOmLPgpcc,777
 recce/data/_next/static/chunks/86730205-cfb14e3f051bab35.js,sha256=qKkzgWsNqSrn-PUGBloZopIlacGCy8hH0BZA9tpqQoQ,172837
 recce/data/_next/static/chunks/8d700b6a.8bb140898499c512.js,sha256=NPGWvqv8Y_rbNaKxYnoq16mi6FDoQHxOOjpH4tJnUkE,119972
-recce/data/_next/static/chunks/92-7ab55ae02606193c.js,sha256=f0YhpleWOSyqlxfS32YC25CQnb0EJAujrN4BWncRKdY,80497
+recce/data/_next/static/chunks/92-607cd1af83c41f43.js,sha256=0ThWZlPjuwxRRFkwB0qN9Q6QdMxJzdGhuIyLAOHO4lw,80466
 recce/data/_next/static/chunks/9746af58-a42b7d169cacadf0.js,sha256=lV-V6BYpd4HEt-pFtT03G6qQD2Wa-2L_jp6aH9VgoeM,4177
 recce/data/_next/static/chunks/a30376cd-de84559016d7e133.js,sha256=qzgyjrX5PdPKGeSwb-Gh0RfM_NEF2zjMvMK14ixq228,864
 recce/data/_next/static/chunks/b63b1b3f-4282bdcf459e075c.js,sha256=s2EZ59zDBJDmy2kh0cHkQ_pAWxiQr_yCPlnttCj0x_k,572
@@ -60,7 +62,7 @@ recce/data/_next/static/chunks/main-b5b3ae20a1405261.js,sha256=28hJH2VsiTdxZGZrQ
 recce/data/_next/static/chunks/polyfills-42372ed130431b0a.js,sha256=CXPB1kyIrcjjyVBBDLWLKI9yEY1ZZbeASUON648vloM,112594
 recce/data/_next/static/chunks/webpack-7b49d5ba7e3a434d.js,sha256=lcK59Juk4EZqi3jIsY35lQqAlB5TLTPy-neri0BIZL0,3929
 recce/data/_next/static/chunks/app/layout-177a410a97e0d018.js,sha256=Hmy6Lmrt4GPYYa0f1qhxTiMj4K47PCmkAkrm1Rbnd7w,1731
-recce/data/_next/static/chunks/app/page-59241c42b7dd4fcf.js,sha256=5G46SScopZfjA1KxuSNwUunWAz7FIP_81LQ0Oc5DoI4,156420
+recce/data/_next/static/chunks/app/page-da6e046a8235dbfc.js,sha256=pgWvu9fFwA7namOsi52ywwJQlJjjVNjG9-9F3iuTEAk,156424
 recce/data/_next/static/chunks/app/_not-found/page-01ed58b7f971d311.js,sha256=_l3JSAaIJxInWqyqdDsobkhJ3JP6DZuwXkw-MsRakQA,1751
 recce/data/_next/static/chunks/pages/_app-437c455677d62394.js,sha256=FaxURD1wuAmnMU1zJlsQKT9rRdfuspR7yx7BckcoCLY,284
 recce/data/_next/static/chunks/pages/_error-e7650df18ca04bde.js,sha256=0Vzx_hCqw4RcptsFzHfUigpI13sToBnFw38owTP7ruo,250
@@ -116,10 +118,11 @@ recce/util/pydantic_model.py,sha256=KumKuyCjbTzEMsKLE4-b-eZfp0gLhYDdmVtw1-hxiJw,
 recce/util/recce_cloud.py,sha256=YgRs9PhER3EhESaJInJo6mf3uwvnp0OavHBY1OYHPXw,7958
 recce/util/singleton.py,sha256=1cU99I0f9tjuMQLMJyLsK1oK3fZJMsO5-TbRHAMXqds,627
 recce/yaml/__init__.py,sha256=EgXYlFeJZchatUClRDXbIC5Oqb2_nBvB2NqItYVihio,1292
-recce_nightly-1.9.0.20250623.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
+recce_nightly-1.10.0.20250625.dist-info/licenses/LICENSE,sha256=CQjjMy9aYPhfe8xG_bcpIfKtNkdxLZ5IOb8oPygtUhY,11343
 tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_cli.py,sha256=_ikuZW4bYEGyj5A_V7nDSyBa1tVZqc0TVm-TJ0rEYVo,5467
 tests/test_config.py,sha256=ODDFe_XF6gphmSmmc422dGLBaCCmG-IjDzTkD5SJsJE,1557
+tests/test_connect_to_cloud.py,sha256=b2fgV8L1iQBdEwh6RumMsIIyYg7GtMOMRz1dvE3WRPg,3059
 tests/test_core.py,sha256=jfn9dV-k89O2TywUYHmMTSdvyfBUips1ful1cIY2oFY,799
 tests/test_dbt.py,sha256=VzXvdoJNwwEmKNhJJDNB1N_qZYwp1aoJQ1sLcoyRBmk,1316
 tests/test_pull_request.py,sha256=HmZo5MoDaoKSgPwbLxJ3Ur3ajZ7IxhkzJxaOmhg6bwE,3562
@@ -144,8 +147,8 @@ tests/tasks/test_row_count.py,sha256=21PaP2aq-x8-pqwzWHRT1sixhQ8g3CQNRWOZTTmbK0s
 tests/tasks/test_schema.py,sha256=7ds4Vx8ixaiIWDR49Lvjem4xlPkRP1cXazDRY3roUak,3121
 tests/tasks/test_top_k.py,sha256=YR_GS__DJsbDlQVaEEdJvNQ3fh1VmV5Nb3G7lb0r6YM,1779
 tests/tasks/test_valuediff.py,sha256=_xQJGgxsXoy2NYk_Z6Hsw2FlVh6zk2nN_iUueyRN1e8,2046
-recce_nightly-1.9.0.20250623.dist-info/METADATA,sha256=HlG1PEA0-VELPkwBVp7BfxNfoWqPqNK8aAxhM82xSHE,9397
-recce_nightly-1.9.0.20250623.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-recce_nightly-1.9.0.20250623.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
-recce_nightly-1.9.0.20250623.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
-recce_nightly-1.9.0.20250623.dist-info/RECORD,,
+recce_nightly-1.10.0.20250625.dist-info/METADATA,sha256=7mItLT4k75wBtKN04-E_OJX5gllsTv9iLmxFE8aI1OE,9367
+recce_nightly-1.10.0.20250625.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+recce_nightly-1.10.0.20250625.dist-info/entry_points.txt,sha256=oqoY_IiwIqXbgrIsPnlqUqao2eiIeP2dprowkOlmeyg,40
+recce_nightly-1.10.0.20250625.dist-info/top_level.txt,sha256=6PKGVpf75idP0C6KEaldDzzZUauIxNu1ZDstau1pI4I,12
+recce_nightly-1.10.0.20250625.dist-info/RECORD,,

tests/test_connect_to_cloud.py

@@ -0,0 +1,82 @@
+import base64
+import unittest
+from unittest.mock import patch
+from urllib.parse import quote
+
+from cryptography.hazmat.primitives import hashes
+from cryptography.hazmat.primitives.asymmetric import padding
+
+from recce.connect_to_cloud import (
+    connect_to_cloud_background_task,
+    decrypt_code,
+    generate_key_pair,
+    is_callback_server_running,
+    prepare_connection_url,
+)
+
+
+class ConnectToCloudTests(unittest.TestCase):
+
+    def test_generate_key_pair(self):
+        private_key, public_key = generate_key_pair()
+        self.assertIsNotNone(private_key)
+        self.assertIsNotNone(public_key)
+        self.assertEqual(private_key.public_key().public_numbers(), public_key.public_numbers())
+
+    def test_prepare_connection_url(self):
+        _, public_key = generate_key_pair()
+        url, port = prepare_connection_url(public_key)
+        self.assertIn("connect?", url)
+        self.assertTrue(port >= 10000 and port <= 15000)
+
+    def test_decrypt_code(self):
+        private_key, public_key = generate_key_pair()
+        test_string = "recce-api-token-123"
+        ciphertext = public_key.encrypt(
+            test_string.encode(),
+            padding.OAEP(mgf=padding.MGF1(algorithm=hashes.SHA1()), algorithm=hashes.SHA1(), label=None),
+        )
+        b64_ciphertext = base64.b64encode(ciphertext).decode()
+        result = decrypt_code(private_key, b64_ciphertext)
+        self.assertEqual(result, test_string)
+
+    @patch("recce.connect_to_cloud.update_recce_api_token")
+    @patch("recce.connect_to_cloud.update_onboarding_state")
+    @patch("recce.connect_to_cloud.RecceCloud")
+    def test_handle_callback_request_success(self, mock_recce_cloud, mock_update_state, mock_update_token):
+        private_key, public_key = generate_key_pair()
+
+        # Prepare encrypted token
+        test_token = "recce-api-token-xyz"
+        ciphertext = public_key.encrypt(
+            test_token.encode(),
+            padding.OAEP(mgf=padding.MGF1(algorithm=hashes.SHA1()), algorithm=hashes.SHA1(), label=None),
+        )
+        encrypted_b64 = base64.b64encode(ciphertext).decode()
+
+        # Set up mocks
+        mock_recce_cloud.return_value.verify_token.return_value = True
+
+        from recce.connect_to_cloud import handle_callback_request
+
+        result = handle_callback_request(f"code={quote(encrypted_b64)}", private_key)
+
+        assert result == test_token
+        mock_update_token.assert_called_once_with(test_token)
+        mock_update_state.assert_called_once_with(test_token, False)
+
+    def test_is_callback_server_running(self):
+        # Should return False by default
+        self.assertFalse(is_callback_server_running())
+
+    @patch("recce.connect_to_cloud.run_one_time_http_server")
+    def test_connect_to_cloud_background_task_runs(self, mock_server):
+        private_key, public_key = generate_key_pair()
+        url, port = prepare_connection_url(public_key)
+
+        connect_to_cloud_background_task(private_key, port, url)
+        mock_server.assert_called_once()
+
+
+if __name__ == "__main__":
+    unittest.main()