roksta 0.2.4__cp311-cp311-win_amd64.whl → 0.2.5__cp311-cp311-win_amd64.whl

tests/functions/test_utils_functions.py

@@ -0,0 +1,222 @@
+import importlib
+import sys
+import types
+import json
+import pytest
+
+
+def _setup_basic_stubs(secret_data=b'MY_SECRET', secret_side_effect=None, auth_verify=lambda token: {'uid': 'u1'}):
+    """Install minimal stub modules for google.cloud.secretmanager, firebase_functions and firebase_admin.
+    Returns a dict of original sys.modules entries for cleanup.
+    """
+    orig = {}
+    names = [
+        'google',
+        'google.cloud',
+        'google.cloud.secretmanager',
+        'firebase_functions',
+        'firebase_admin',
+        'firebase_admin.auth',
+    ]
+    for name in names:
+        orig[name] = sys.modules.get(name)
+
+    # google package and submodules
+    google_mod = types.ModuleType('google')
+    cloud_mod = types.ModuleType('google.cloud')
+    secretmanager_mod = types.ModuleType('google.cloud.secretmanager')
+
+    class _Payload:
+        def __init__(self, data):
+            self.data = data
+
+    class _Response:
+        def __init__(self, data):
+            self.payload = _Payload(data)
+
+    class SecretManagerServiceClient:
+        def access_secret_version(self, request=None):
+            if secret_side_effect:
+                # raise the provided exception instance for testing
+                raise secret_side_effect
+            return _Response(secret_data)
+
+    secretmanager_mod.SecretManagerServiceClient = SecretManagerServiceClient
+    cloud_mod.secretmanager = secretmanager_mod
+    google_mod.cloud = cloud_mod
+
+    sys.modules['google'] = google_mod
+    sys.modules['google.cloud'] = cloud_mod
+    sys.modules['google.cloud.secretmanager'] = secretmanager_mod
+
+    # firebase_functions stub with a Response type expected by utils.create_json_response
+    firebase_functions_mod = types.ModuleType('firebase_functions')
+
+    class FakeResponse:
+        def __init__(self, response=None, mimetype=None, status=200, **kwargs):
+            self.status_code = status
+            if isinstance(response, (dict, list)):
+                self._body_text = json.dumps(response)
+            else:
+                self._body_text = '' if response is None else response
+            self.headers = kwargs.get('headers', {})
+
+        def get_data(self, as_text=False):
+            if as_text:
+                return self._body_text
+            return self._body_text.encode('utf-8')
+
+    firebase_functions_mod.https_fn = types.SimpleNamespace(Request=object, Response=FakeResponse)
+    sys.modules['firebase_functions'] = firebase_functions_mod
+
+    # firebase_admin + auth stub
+    firebase_admin_mod = types.ModuleType('firebase_admin')
+    auth_mod = types.ModuleType('firebase_admin.auth')
+
+    def _verify_id_token(token):
+        return auth_verify(token)
+
+    auth_mod.verify_id_token = _verify_id_token
+    firebase_admin_mod.auth = auth_mod
+    sys.modules['firebase_admin'] = firebase_admin_mod
+    sys.modules['firebase_admin.auth'] = auth_mod
+
+    return orig
+
+
+def _restore_orig(orig):
+    for name, val in orig.items():
+        if val is None:
+            if name in sys.modules:
+                del sys.modules[name]
+        else:
+            sys.modules[name] = val
+
+
+def test_get_secret_key_returns_decoded_string_on_success():
+    orig = _setup_basic_stubs(secret_data=b'MY_SECRET')
+    try:
+        if 'utils' in sys.modules:
+            del sys.modules['utils']
+        utils = importlib.import_module('utils')
+        res = utils.get_secret_key('any')
+        assert res == 'MY_SECRET'
+    finally:
+        _restore_orig(orig)
+
+
+def test_get_secret_key_raises_exception_on_client_error():
+    orig = _setup_basic_stubs(secret_side_effect=Exception('boom'))
+    try:
+        if 'utils' in sys.modules:
+            del sys.modules['utils']
+        utils = importlib.import_module('utils')
+        with pytest.raises(Exception) as exc:
+            utils.get_secret_key('any')
+        assert 'Failed to retrieve secret key' in str(exc.value)
+        assert 'boom' in str(exc.value)
+    finally:
+        _restore_orig(orig)
+
+
+def test_verify_firebase_token_raises_when_header_missing_or_invalid():
+    orig = _setup_basic_stubs()
+    try:
+        if 'utils' in sys.modules:
+            del sys.modules['utils']
+        utils = importlib.import_module('utils')
+
+        class DummyRequest:
+            def __init__(self, headers=None):
+                self.headers = headers or {}
+
+        # Missing header
+        req = DummyRequest(headers={})
+        with pytest.raises(Exception) as exc:
+            utils.verify_firebase_token(req)
+        assert 'Missing or invalid Authorization header' in str(exc.value)
+
+        # Header present but not starting with 'Bearer '
+        req2 = DummyRequest(headers={'Authorization': 'Token abc'})
+        with pytest.raises(Exception) as exc2:
+            utils.verify_firebase_token(req2)
+        assert 'Missing or invalid Authorization header' in str(exc2.value)
+    finally:
+        _restore_orig(orig)
+
+
+def test_verify_firebase_token_calls_auth_verify_and_returns_payload():
+    orig = _setup_basic_stubs(auth_verify=lambda token: {'uid': 'user_1'})
+    try:
+        if 'utils' in sys.modules:
+            del sys.modules['utils']
+        utils = importlib.import_module('utils')
+
+        class DummyRequest:
+            def __init__(self, headers=None):
+                self.headers = headers or {}
+
+        req = DummyRequest(headers={'Authorization': 'Bearer TOK'})
+        res = utils.verify_firebase_token(req)
+        assert res == {'uid': 'user_1'}
+    finally:
+        _restore_orig(orig)
+
+
+def test_verify_firebase_token_wraps_auth_exceptions():
+    def _raiser(token):
+        raise Exception('bad token')
+
+    orig = _setup_basic_stubs(auth_verify=_raiser)
+    try:
+        if 'utils' in sys.modules:
+            del sys.modules['utils']
+        utils = importlib.import_module('utils')
+
+        class DummyRequest:
+            def __init__(self, headers=None):
+                self.headers = headers or {}
+
+        req = DummyRequest(headers={'Authorization': 'Bearer TOK'})
+        with pytest.raises(Exception) as exc:
+            utils.verify_firebase_token(req)
+        assert 'Authentication failed' in str(exc.value)
+        assert 'bad token' in str(exc.value)
+    finally:
+        _restore_orig(orig)
+
+
+def test_get_api_key_uses_lowercase_family_name_and_get_secret_key(monkeypatch):
+    orig = _setup_basic_stubs()
+    try:
+        if 'utils' in sys.modules:
+            del sys.modules['utils']
+        utils = importlib.import_module('utils')
+
+        called = {}
+
+        def fake_get_secret_key(name):
+            called['name'] = name
+            return 'THE_KEY'
+
+        monkeypatch.setattr(utils, 'get_secret_key', fake_get_secret_key)
+        from enums import LlmFamily
+        res = utils.get_api_key(LlmFamily.OPENAI)
+        assert res == 'THE_KEY'
+        assert called['name'] == 'openai-api-key'
+    finally:
+        _restore_orig(orig)
+
+
+def test_create_json_response_returns_formatted_response():
+    orig = _setup_basic_stubs()
+    try:
+        if 'utils' in sys.modules:
+            del sys.modules['utils']
+        utils = importlib.import_module('utils')
+        resp = utils.create_json_response(True, {'a': 1}, 201)
+        assert getattr(resp, 'status_code', None) == 201
+        data = json.loads(resp.get_data(as_text=True))
+        assert data == {"success": True, "payload": {"a": 1}}
+    finally:
+        _restore_orig(orig)

roksta-0.2.4.dist-info/RECORD

@@ -1,51 +0,0 @@
-roksta/__init__.cp311-win_amd64.pyd,sha256=NN88Wx3_X0ecA31XMHYDQDYO5tjmpG9X5nkzZaLQ6Cg,15872
-roksta/analytics.cp311-win_amd64.pyd,sha256=JgkQWRm48hjNQbik0JYlLv6BVNe3Xh2PAkxDD8XBz1I,93184
-roksta/balance.cp311-win_amd64.pyd,sha256=w3vMmvH55SCGPY5cVNsPAonqsrf7IM1FW3tm49JNLEs,117760
-roksta/build_project.cp311-win_amd64.pyd,sha256=QWrKrfDDdblvSCB0zRuhbxJbqPanmzHhToZyvirsveU,59904
-roksta/chat_workflow.cp311-win_amd64.pyd,sha256=tL7jPMCOTm5R0gAQo-pcJzrvyg8tndMc_mv69r6YtcM,77312
-roksta/check_for_updates.cp311-win_amd64.pyd,sha256=g-5Cmmg3gfTEcGmnvHvssmSx5owttOmPiRJ9C3pkqXI,62464
-roksta/checkpoints.cp311-win_amd64.pyd,sha256=muMMlgxXNXwqMvSXJSZk0m_y9XlplDSRbylcVLs2hoY,81408
-roksta/clarify_goal.cp311-win_amd64.pyd,sha256=fwvAB6Bertcv77xZN9GpYoSGWYKOVRYFIxnPia_jiq0,64000
-roksta/codebase_listing.cp311-win_amd64.pyd,sha256=axThLcqE2DfQ4zB9w-7_5k9dqaWGH0qIuSgnsj31daQ,46592
-roksta/command_handlers.cp311-win_amd64.pyd,sha256=Ii6YvllBXZc3IYwmI5pHIvVuDmfvFPFgFXgQOcqA_0o,341504
-roksta/create_default_config.cp311-win_amd64.pyd,sha256=8ZPQgew_nsVDmYMH5FSnfC4md1tAW77Y_WL1mMiKQdQ,31744
-roksta/default_config.cp311-win_amd64.pyd,sha256=wtW0xTHZEkR2KMICMEw-IUmUllODPO5e-r6TYuO2fgI,30720
-roksta/enums.cp311-win_amd64.pyd,sha256=WjqjHkZuhntAvee0WahHN09vqWTpKVdLtabVe-YdHd0,135168
-roksta/env.cp311-win_amd64.pyd,sha256=EqIHdDnOdxWXHLKL23k1xGm7eJdc-oEI_zmEsi3L5l4,26624
-roksta/extended_text_area.cp311-win_amd64.pyd,sha256=M-cJEi5DGkoQsA_s_AdtOXIyPB8SSpbOedoG1-Z5rR8,105984
-roksta/firebase.cp311-win_amd64.pyd,sha256=dchZr8MdUT1jKpWFYRlzDLrgAPx_5_78Hf4NxlEd2vk,167936
-roksta/firebase_auth_web.cp311-win_amd64.pyd,sha256=rrQH0fRTUkuyZRywB_At67WWYIOInNnMTCzKyt1r5P0,169984
-roksta/firebase_config.cp311-win_amd64.pyd,sha256=Rf8JUcCLRYebRH7xPHqwyUk-OyPerbVZqAkT6KY20F0,21504
-roksta/fix_tests.cp311-win_amd64.pyd,sha256=d18SrOZ0oGZjDzyp6TRpyBYU7bhWREDxszrvCcemNIs,73728
-roksta/gen_codebase_summaries.cp311-win_amd64.pyd,sha256=oxVIXTItyFLnblCNbUPC4ioweSrSS6AH2FnVIdaYL98,225280
-roksta/gen_one_line_goal.cp311-win_amd64.pyd,sha256=vUpXBAMwxuOHKEoTPImzmv7BQW8dFQYOFffoNDjRHMw,63488
-roksta/get_codebase_structure.cp311-win_amd64.pyd,sha256=Kd2mL6M7Q99LJvXfrH8gyvHNZ473EaHeXV9GzmNi5XY,66048
-roksta/get_failing_tests.cp311-win_amd64.pyd,sha256=ffoIZOfvbAIUprTw0t6ljrv-dD_hIOetuVEykenl3Mk,69120
-roksta/goal_workflow.cp311-win_amd64.pyd,sha256=zdgkCRG1me_jRYKO3ZpfFt5MhjIpZcrw_C564ORA8tA,183296
-roksta/init_codebase.cp311-win_amd64.pyd,sha256=jX--btTXF2D9pJMTjY_RA-1KWPu4GTJnoX6_fE9ik9E,95232
-roksta/lint_code.cp311-win_amd64.pyd,sha256=cii0tl9lQGWbnxGB1uhzuy6QofgHbNLcoHz2MrW0xtU,69632
-roksta/logger.cp311-win_amd64.pyd,sha256=sNarurnlnHvSoRjSx5UQEps6wVIU6GrBq-YF7xYblJc,38400
-roksta/main.cp311-win_amd64.pyd,sha256=1GG77Hy9uVkacn9MqFHVnYGTz5ZXNaZKBNg2_eERQt0,22528
-roksta/make_issue.cp311-win_amd64.pyd,sha256=0RvNeZN5_HlEOjBvLbS2eVXC4DTGT3hHhgfiFRdmkBc,66048
-roksta/new_features.cp311-win_amd64.pyd,sha256=rQnl_SrGdiMRGQJ78MdWLdJtAISjbs2lYWPOKQmoQ2g,16896
-roksta/parse_readme.cp311-win_amd64.pyd,sha256=t6IAC4J-OZIkwT6yEk8KPNf34TfzQMDRgsXG3u7ClyA,60416
-roksta/propose_solution.cp311-win_amd64.pyd,sha256=7nJTN-TzUNDDzLbuUpRawNg_AEBCScm9RwaDIIMR4EI,61440
-roksta/response_formats.cp311-win_amd64.pyd,sha256=PG1zBWdjfILEDJJbYlsMrzziXZZWvV4Qnzv5f3zz-TA,41984
-roksta/rewrite_goal.cp311-win_amd64.pyd,sha256=T1mofuBqkfLHU_k5Ps_nCLz8qZ41S1dU-NVp_yqpBxA,55808
-roksta/roksta.cp311-win_amd64.pyd,sha256=XyYJNipce_ZqxwKHxkIHCbwsMEpB8hRxkosihx3gcbQ,227328
-roksta/select_files.cp311-win_amd64.pyd,sha256=iRZ81TB90mQzQzB24LMPmbj7-QaBcvh4IjBTb3DndWY,66048
-roksta/tips.cp311-win_amd64.pyd,sha256=aowSWThim6IubscB6ocL-BXxX02858fLoZq4-ZiJV1c,20480
-roksta/utils.cp311-win_amd64.pyd,sha256=AmuKXvjwEIBMiN1eDk7-CMu0Bf_4WCgZBZ64_iRLpjE,205312
-roksta/write_code.cp311-win_amd64.pyd,sha256=5ASCp4-ZKGrv9BzXAFcwW5hoDsdV4QtS6-ZwP5dR1cw,58880
-roksta/ai/__init__.cp311-win_amd64.pyd,sha256=16CFHgYfDta6frg8sxvMFhMcyWGZ7LyOm2GChNfDT-I,15872
-roksta/ai/call_ai.cp311-win_amd64.pyd,sha256=RZBZDWsR8UMaIJSI5-hr9QCRfkj7toaR1SEal9C9EbE,72192
-roksta/ai/gemini.cp311-win_amd64.pyd,sha256=krZDStSO9yndEqbOfD_vEiSVBvv7b7yjKyndezO40VA,151040
-roksta/ai/generic.cp311-win_amd64.pyd,sha256=C_2GRKxWFZJO1Lq298IkIL0MeR_lWvv4lwU0YDm_5-4,111616
-roksta/ai/llm.cp311-win_amd64.pyd,sha256=bbYVnUxc1RCj_mCKhtzmpUMDr7BoR0T9p2XDHd2Kflo,94208
-roksta/ai/openai.cp311-win_amd64.pyd,sha256=juaiNluktM-JixeBAn2NEuD8QiNlG4gF6mhQF2kXM50,109568
-roksta/ai/tools.cp311-win_amd64.pyd,sha256=oT1fC_ilgqLS2CQYw7WBs_f8vkvFYZPbBXy4GJHPcGk,230912
-roksta-0.2.4.dist-info/METADATA,sha256=cYaARo086rNt4toq7tM9PKFVxdk1Flt9-16gSjo8rVI,1466
-roksta-0.2.4.dist-info/WHEEL,sha256=JLOMsP7F5qtkAkINx5UnzbFguf8CqZeraV8o04b0I8I,101
-roksta-0.2.4.dist-info/entry_points.txt,sha256=mzRdYg_DlzZRwjxYUt9-gyoRCkM1QBTeTbwETgiTdGw,44
-roksta-0.2.4.dist-info/top_level.txt,sha256=Worlz0mMqoxRSgPmoN1j_P-sJgIrmzD0uJ08qJwpWpQ,7
-roksta-0.2.4.dist-info/RECORD,,