roksta 0.2.7__cp311-cp311-macosx_10_9_universal2.whl → 0.3.1__cp311-cp311-macosx_10_9_universal2.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of roksta might be problematic. Click here for more details.

Files changed (72) hide show
  1. roksta/ai/call_ai.cpython-311-darwin.so +0 -0
  2. roksta/ai/gemini.cpython-311-darwin.so +0 -0
  3. roksta/ai/generic.cpython-311-darwin.so +0 -0
  4. roksta/ai/llm.cpython-311-darwin.so +0 -0
  5. roksta/ai/openai.cpython-311-darwin.so +0 -0
  6. roksta/ai/tools/__init__.cpython-311-darwin.so +0 -0
  7. roksta/ai/tools/delete_file.cpython-311-darwin.so +0 -0
  8. roksta/ai/tools/edit_file.cpython-311-darwin.so +0 -0
  9. roksta/ai/tools/final_response.cpython-311-darwin.so +0 -0
  10. roksta/ai/tools/get_file_summaries.cpython-311-darwin.so +0 -0
  11. roksta/ai/tools/read_file.cpython-311-darwin.so +0 -0
  12. roksta/ai/tools/regex_replace.cpython-311-darwin.so +0 -0
  13. roksta/ai/tools/shell_any.cpython-311-darwin.so +0 -0
  14. roksta/ai/tools/shell_limited.cpython-311-darwin.so +0 -0
  15. roksta/ai/tools/tool_defs.cpython-311-darwin.so +0 -0
  16. roksta/ai/tools/tool_utils.cpython-311-darwin.so +0 -0
  17. roksta/ai/tools/write_file.cpython-311-darwin.so +0 -0
  18. roksta/balance.cpython-311-darwin.so +0 -0
  19. roksta/command_handlers/__init__.cpython-311-darwin.so +0 -0
  20. roksta/command_handlers/handle_activate_command.cpython-311-darwin.so +0 -0
  21. roksta/command_handlers/handle_add_funds_command.cpython-311-darwin.so +0 -0
  22. roksta/command_handlers/handle_auto_charge_command.cpython-311-darwin.so +0 -0
  23. roksta/command_handlers/handle_auto_commit_command.cpython-311-darwin.so +0 -0
  24. roksta/command_handlers/handle_building_command.cpython-311-darwin.so +0 -0
  25. roksta/command_handlers/handle_chat_command.cpython-311-darwin.so +0 -0
  26. roksta/command_handlers/handle_dev_rate_command.cpython-311-darwin.so +0 -0
  27. roksta/command_handlers/handle_feedback_command.cpython-311-darwin.so +0 -0
  28. roksta/command_handlers/handle_goal_command.cpython-311-darwin.so +0 -0
  29. roksta/command_handlers/handle_help_command.cpython-311-darwin.so +0 -0
  30. roksta/command_handlers/handle_init_command.cpython-311-darwin.so +0 -0
  31. roksta/command_handlers/handle_linting_command.cpython-311-darwin.so +0 -0
  32. roksta/command_handlers/handle_login_command.cpython-311-darwin.so +0 -0
  33. roksta/command_handlers/handle_logout_command.cpython-311-darwin.so +0 -0
  34. roksta/command_handlers/handle_payment_details_command.cpython-311-darwin.so +0 -0
  35. roksta/command_handlers/handle_quit_command.cpython-311-darwin.so +0 -0
  36. roksta/command_handlers/handle_redeem_command.cpython-311-darwin.so +0 -0
  37. roksta/command_handlers/handle_request_activation_command.cpython-311-darwin.so +0 -0
  38. roksta/command_handlers/handle_testing_command.cpython-311-darwin.so +0 -0
  39. roksta/command_handlers/handle_usage_command.cpython-311-darwin.so +0 -0
  40. roksta/env.cpython-311-darwin.so +0 -0
  41. roksta/extended_text_area.cpython-311-darwin.so +0 -0
  42. roksta/firebase.cpython-311-darwin.so +0 -0
  43. roksta/fix_tests.cpython-311-darwin.so +0 -0
  44. roksta/goal_workflow.cpython-311-darwin.so +0 -0
  45. roksta/lint_code.cpython-311-darwin.so +0 -0
  46. roksta/main.cpython-311-darwin.so +0 -0
  47. roksta/new_features.cpython-311-darwin.so +0 -0
  48. roksta/roksta.cpython-311-darwin.so +0 -0
  49. roksta/tips.cpython-311-darwin.so +0 -0
  50. roksta/utils.cpython-311-darwin.so +0 -0
  51. {roksta-0.2.7.dist-info → roksta-0.3.1.dist-info}/METADATA +2 -1
  52. roksta-0.3.1.dist-info/RECORD +109 -0
  53. tests/conftest.py +42 -0
  54. tests/functions/{api_v0_01 → api_v1_01}/__init__.py +1 -1
  55. tests/functions/{api_v0_01 → api_v1_01}/test__analytics.py +2 -3
  56. tests/functions/{api_v0_01 → api_v1_01}/test__gemini_proxy.py +51 -6
  57. tests/functions/{api_v0_01 → api_v1_01}/test__generic_proxy.py +31 -2
  58. tests/functions/{api_v0_01 → api_v1_01}/test__get_payment_details.py +2 -2
  59. tests/functions/{api_v0_01 → api_v1_01}/test__openai_proxy.py +50 -14
  60. tests/functions/{api_v0_01 → api_v1_01}/test__redeem_credit_code.py +2 -2
  61. tests/functions/{api_v0_01 → api_v1_01}/test__sync_emails.py +3 -2
  62. tests/functions/{api_v0_01 → api_v1_01}/test__take_payment.py +2 -2
  63. tests/functions/{api_v0_01 → api_v1_01}/test__use_activation_code.py +3 -2
  64. tests/functions/test_utils.py +484 -0
  65. roksta/ai/tools.cpython-311-darwin.so +0 -0
  66. roksta/command_handlers.cpython-311-darwin.so +0 -0
  67. roksta-0.2.7.dist-info/RECORD +0 -78
  68. tests/functions/test_utils_functions.py +0 -222
  69. {roksta-0.2.7.dist-info → roksta-0.3.1.dist-info}/WHEEL +0 -0
  70. {roksta-0.2.7.dist-info → roksta-0.3.1.dist-info}/entry_points.txt +0 -0
  71. {roksta-0.2.7.dist-info → roksta-0.3.1.dist-info}/top_level.txt +0 -0
  72. /tests/functions/{test_main_functions.py → test_main.py} +0 -0
@@ -1,78 +0,0 @@
1
- roksta/logger.cpython-311-darwin.so,sha256=xch275vL4nC3HeSUxVdjTpnXmRXq2Jd1auVsd-a3bNs,122016
2
- roksta/lint_code.cpython-311-darwin.so,sha256=HpTWqrsxDMAiu0DqNWSZiEOBvbWtDVkq3S4eooB9tWY,207840
3
- roksta/get_codebase_structure.cpython-311-darwin.so,sha256=gFFoEJU2H9vJgmbRWjcqY2tSqc_7eUJPWIGTDgSQZk4,189984
4
- roksta/checkpoints.cpython-311-darwin.so,sha256=4vZMS8o8V6fez18V-ln64U4tXsOBPqbGgLE7pyYhZkQ,238032
5
- roksta/extended_text_area.cpython-311-darwin.so,sha256=W8q52wI8SmPLQ74ySEXm8eCqsQTdPji5-l9ejdx6TQ0,290904
6
- roksta/firebase_config.cpython-311-darwin.so,sha256=BKPNge6g8lDyI4M6dAxXVasD1T-j9stNUT--pB-AGT8,86568
7
- roksta/init_codebase.cpython-311-darwin.so,sha256=P9kuJXGfk4uk9VnGozUvnOyP66V0PZ4xbSS22N7i890,275024
8
- roksta/gen_one_line_goal.cpython-311-darwin.so,sha256=LGggcZkhy9W8RTUIzj9vcACw0KZs7Big3_TpTXE3IZc,207560
9
- roksta/new_features.cpython-311-darwin.so,sha256=J0KVA_OrPS0ADGKw05jLDTDF9rfSiJWDN-A1GD4m2Ew,85632
10
- roksta/chat_workflow.cpython-311-darwin.so,sha256=uQ4mon110StmmfbIALtZ9JOI76d6yIqOu_WUsxZLbgw,257232
11
- roksta/select_files.cpython-311-darwin.so,sha256=Qbce978o2zTA1wLVw2pHtNvhJJWhhUGHRuC5C57H_XM,207600
12
- roksta/parse_readme.cpython-311-darwin.so,sha256=2vXGMqTmmn9vm1twZkudrs4IXL6RTm0TYBcJwUlY0_E,207088
13
- roksta/firebase.cpython-311-darwin.so,sha256=AhLG5dRIzo31Qakmg5nS_Mi-Fxt6mUt892vm840fPw4,438704
14
- roksta/goal_workflow.cpython-311-darwin.so,sha256=mJjIcJJOoZFPulhvR-0Zz78o1bg8zJU0T4MWOF-jq7Q,505312
15
- roksta/tips.cpython-311-darwin.so,sha256=zGp4O9s2I30pSHADOZsnNzbQJxCITQzWJGrHwQUgcwc,85592
16
- roksta/write_code.cpython-311-darwin.so,sha256=U_6AsReL8Kxv8TwuscxLySI4jL7dQa-rFgft7kLeYCQ,190944
17
- roksta/enums.cpython-311-darwin.so,sha256=rGABZEs5M_6PDdxssR9-gwfDhqMoor9kOyoxf79VFnI,404232
18
- roksta/analytics.cpython-311-darwin.so,sha256=lXYObBh_e8QiM7PQHipcKWtl5sD04ygEi71p2dvtc9k,256576
19
- roksta/command_handlers.cpython-311-darwin.so,sha256=eObQAXdqcclOiTaqV6LPz6JgE3POLjLmEheS5nywimA,966456
20
- roksta/main.cpython-311-darwin.so,sha256=1fbrCeAjr_jN7FPj5kjKnV_BnNPSRGp3s71e0CZ8DeI,122152
21
- roksta/response_formats.cpython-311-darwin.so,sha256=BthAKv6MpQLL6SPni865ADEgw9VQwy-w1kgEEev-xGY,153928
22
- roksta/check_for_updates.cpython-311-darwin.so,sha256=quPqiJn9zI5elA4bJ5l3u9x_o-tpzpTVkfq2meCAqNA,290200
23
- roksta/utils.cpython-311-darwin.so,sha256=D-xhVO-Jl14vV2e3mADPUkEa8vJ-yDtCFE_ET4qSGyk,572456
24
- roksta/fix_tests.cpython-311-darwin.so,sha256=Xj4xMy9Tqv7hh65YvwdFCODEbvK_Jpjul-uBXZ0LSPY,223984
25
- roksta/firebase_auth_web.cpython-311-darwin.so,sha256=U51PxZ0Qocf5yE5WfMacrIC2VAvYqRxraf5DvKAeNfM,439656
26
- roksta/env.cpython-311-darwin.so,sha256=K7F-kgnErvOpliK1AewNczx9GYtlUL_tEqkwtXkApSo,87992
27
- roksta/get_failing_tests.cpython-311-darwin.so,sha256=1jvAcZKA8gsnsrq6-HNsro-rQkqesYZIi9nKhf8g5m8,207576
28
- roksta/clarify_goal.cpython-311-darwin.so,sha256=9GRKv1QrjNCoulhg7BxAMwKItzLaI1XGcADo0lFCaYY,207616
29
- roksta/roksta.cpython-311-darwin.so,sha256=zoV2VaeJtXpW8Ky1U63PMkv7bsNJcdM7Iy5SmKhxEsk,555120
30
- roksta/build_project.cpython-311-darwin.so,sha256=vUvI_82FBp5at73GbjzytM40VdyuxNrMSoy-TF0_hW0,190992
31
- roksta/rewrite_goal.cpython-311-darwin.so,sha256=Ef34h7nGUNKgGcRQY2EKLKkSrcNLKp4xzKE9JTmg6QY,174512
32
- roksta/run_cli_goal.cpython-311-darwin.so,sha256=mIDRX9GJIZUYock-CE_VUBvR-zLvYAkXQRCCdMUXlU4,207120
33
- roksta/codebase_listing.cpython-311-darwin.so,sha256=yqwkwIMP3MKag60iU2LEYWt36jCXgjmnFL7UuKiUUaQ,155656
34
- roksta/default_config.cpython-311-darwin.so,sha256=okjKjvIXlHp_m_82TZIFdMT9fgEAfr1R3EodQhIX564,102152
35
- roksta/__init__.cpython-311-darwin.so,sha256=0ErvE_Ab31A1ApkmxEhfJuljCTrEA6VccImUD3_9z8U,85488
36
- roksta/create_default_config.cpython-311-darwin.so,sha256=6XuYR51E9xZ54eSdTRZzlsN1ywGP5EuVUTHnC6L0ljU,121672
37
- roksta/make_issue.cpython-311-darwin.so,sha256=sPSYITShHG4HpdqDow8XbLhADJ-as4J4Jl_bviKOx0Q,222064
38
- roksta/propose_solution.cpython-311-darwin.so,sha256=wCKK0fp40j_ewRIzxF1PYqmUGkE3i4iGnp6tTB0A9Vc,190920
39
- roksta/balance.cpython-311-darwin.so,sha256=QpRQuVdggsFhQDW6uvrA9fSY7lHskliMkm4oVOMc9S8,323856
40
- roksta/gen_codebase_summaries.cpython-311-darwin.so,sha256=KYIPW6UaQV_XJEMttZHuUCyyHlPAQDVOvZwMNr2SIzg,621328
41
- roksta/ai/tools.cpython-311-darwin.so,sha256=fspljuxt2iNDW95TlN3c100r5-0AUSFHp1geznGh9G8,620904
42
- roksta/ai/openai.cpython-311-darwin.so,sha256=KL8BSiLkXb8p0uM6Yl3GZSozT3ZUqSBxuuKABOhqN0o,290352
43
- roksta/ai/call_ai.cpython-311-darwin.so,sha256=ha__TxFymsSiqhMesb7AJScC5UMt2YkDMoTuHxYND2k,224048
44
- roksta/ai/llm.cpython-311-darwin.so,sha256=I7JC3AWtct1AkiffsUBCOU9wA3CYBUWaXP_ZlNf70Fw,288232
45
- roksta/ai/gemini.cpython-311-darwin.so,sha256=Ch9BhetpHHmy0PwCe3nJtIIBR4P5IgLqU2NX5BfNw4A,389600
46
- roksta/ai/generic.cpython-311-darwin.so,sha256=YUrNelYAEHhh8OOsU1uf4fYkp4uxrd_XHl6mnctOAGQ,290368
47
- roksta/ai/__init__.cpython-311-darwin.so,sha256=dpXQ14GwJxUat0fxONmbamSbzHo2JsnZHVeNP3LY_No,85488
48
- tests/conftest.py,sha256=WBja46X2wduZRF2t4LARhbNvLlU8oRhSW_HB6ara8Yg,6295
49
- tests/__init__.py,sha256=J6ztnXNqM8TSQCgHXMWb6NRq1kN_h4ql63Jq-TYR8XE,124
50
- tests/functions/test_auth.py,sha256=I3oFTw2LTuUsnQ75v5SCfgUPnpel_pDKqUH0VAJpXQw,992
51
- tests/functions/__init__.py,sha256=mLXx3UARcv07vs_fGJuy3L5ZRn9IRsF9zIuL5IOnwdk,95
52
- tests/functions/test_utils_functions.py,sha256=gS-SxnK0QnVUooxcXlJKodjqOB1HVmUxufwzUPQ7qAk,7368
53
- tests/functions/test_main_functions.py,sha256=h0W_4ISo95WzRy2HpXZw3cuNOTcYk_2QCIZffZJEfEU,2575
54
- tests/functions/api_v1_00/test__sync_emails.py,sha256=OTyf2YbBK7e4pWIQTUGYR4x6il1Hb3g4FYqQ_fRj1nA,12699
55
- tests/functions/api_v1_00/test__take_payment.py,sha256=I-XJasGuLW5wTMAu4nAgFkA8Mo_zVpGPvkJ2LrrJGCk,18022
56
- tests/functions/api_v1_00/test__gemini_proxy.py,sha256=mf8ubRmNxxxN80H6EA-iDPwu6Oue6Rmgxl6tiFNpfrU,13344
57
- tests/functions/api_v1_00/test__analytics.py,sha256=oKjd7_zjnttcb-5BuA93yllkm0OOF4YWnfTUW1-icnM,15806
58
- tests/functions/api_v1_00/__init__.py,sha256=3RCSoj_5UQPw9Z5NeAgYz2s4gfcZIi7g36qk_Ed5TiA,124
59
- tests/functions/api_v1_00/test__use_activation_code.py,sha256=tNyTkkRtrpjei-7F31UNwrcU-PbP-QnkWNU7h2n5Yqo,16960
60
- tests/functions/api_v1_00/test__redeem_credit_code.py,sha256=7GD3tGgO2Lk0BCjUmzZACJiwocEhLBvXrrut0x13IWU,6347
61
- tests/functions/api_v1_00/test__openai_proxy.py,sha256=xlO2zmG64LlrKrbjFz-ds6Map7vy4SjIuHfZJDAqGK8,17726
62
- tests/functions/api_v1_00/test__get_payment_details.py,sha256=i5Px3NkLxqw8D3tVPrYdRJJF5jlNmysSuX_NNolbIa4,13735
63
- tests/functions/api_v1_00/test__generic_proxy.py,sha256=KZkOtmc71aOnWlz2L7B6CFhRNEV73bExG3GR14oD2L0,17673
64
- tests/functions/api_v0_01/test__sync_emails.py,sha256=zCfuh1M6qJlGaCpCWuUE0Ao9Pp5pVkBJXXjoctldkpk,12698
65
- tests/functions/api_v0_01/test__take_payment.py,sha256=2WDv_7cYTozVMl7hDLNDTF9gmfrgna3oR8xgfRa-plA,18022
66
- tests/functions/api_v0_01/test__gemini_proxy.py,sha256=R-lzyVsNtqJFGILKhgy1cUIHg1SYPSFKNDbKp2fua-k,11152
67
- tests/functions/api_v0_01/test__analytics.py,sha256=nMs_aYuXDEFWz9aY21MYwbgugF-czRbucAzis4-8kgc,15807
68
- tests/functions/api_v0_01/__init__.py,sha256=BOIFrFcANRAbpnX0Bez6rtkj6xJnOli1fqzFX5SpkV0,124
69
- tests/functions/api_v0_01/test__use_activation_code.py,sha256=EgifjM9wkW1fkQoBK9z5PEEM5xwpRHyt_g7QWy6xrec,16959
70
- tests/functions/api_v0_01/test__redeem_credit_code.py,sha256=WSs7CE2pkHsied_tg3XCR3RJ9blwQEkOzy-s3lkV_w0,6347
71
- tests/functions/api_v0_01/test__openai_proxy.py,sha256=Nda-RDUVYBQl2wI5mxxr9Ca-9k9HJV5EgEmdIvyytGs,15761
72
- tests/functions/api_v0_01/test__get_payment_details.py,sha256=3LYAp9m1RDsf928HGJ6P_YTji7F4srkDv0iWcKTwW54,13735
73
- tests/functions/api_v0_01/test__generic_proxy.py,sha256=cJftek6njsphK_yesUX1QxphGcD5TEkhQNC5ROvnaIg,16190
74
- roksta-0.2.7.dist-info/RECORD,,
75
- roksta-0.2.7.dist-info/WHEEL,sha256=VWNc5NH4DDoQDkc74R-nMZPza3RGJ6ac2abjVbvC-0k,141
76
- roksta-0.2.7.dist-info/entry_points.txt,sha256=mzRdYg_DlzZRwjxYUt9-gyoRCkM1QBTeTbwETgiTdGw,44
77
- roksta-0.2.7.dist-info/top_level.txt,sha256=lvciNZQ1dPGXpiCLdWVXK03n9fKHjbQdwjqQbnUjeYM,13
78
- roksta-0.2.7.dist-info/METADATA,sha256=sAEp7h8ZrxTSvu8itkn48UfRLVshhC1sOT7urCnt1xM,1427
@@ -1,222 +0,0 @@
1
- import importlib
2
- import sys
3
- import types
4
- import json
5
- import pytest
6
-
7
-
8
- def _setup_basic_stubs(secret_data=b'MY_SECRET', secret_side_effect=None, auth_verify=lambda token: {'uid': 'u1'}):
9
- """Install minimal stub modules for google.cloud.secretmanager, firebase_functions and firebase_admin.
10
- Returns a dict of original sys.modules entries for cleanup.
11
- """
12
- orig = {}
13
- names = [
14
- 'google',
15
- 'google.cloud',
16
- 'google.cloud.secretmanager',
17
- 'firebase_functions',
18
- 'firebase_admin',
19
- 'firebase_admin.auth',
20
- ]
21
- for name in names:
22
- orig[name] = sys.modules.get(name)
23
-
24
- # google package and submodules
25
- google_mod = types.ModuleType('google')
26
- cloud_mod = types.ModuleType('google.cloud')
27
- secretmanager_mod = types.ModuleType('google.cloud.secretmanager')
28
-
29
- class _Payload:
30
- def __init__(self, data):
31
- self.data = data
32
-
33
- class _Response:
34
- def __init__(self, data):
35
- self.payload = _Payload(data)
36
-
37
- class SecretManagerServiceClient:
38
- def access_secret_version(self, request=None):
39
- if secret_side_effect:
40
- # raise the provided exception instance for testing
41
- raise secret_side_effect
42
- return _Response(secret_data)
43
-
44
- secretmanager_mod.SecretManagerServiceClient = SecretManagerServiceClient
45
- cloud_mod.secretmanager = secretmanager_mod
46
- google_mod.cloud = cloud_mod
47
-
48
- sys.modules['google'] = google_mod
49
- sys.modules['google.cloud'] = cloud_mod
50
- sys.modules['google.cloud.secretmanager'] = secretmanager_mod
51
-
52
- # firebase_functions stub with a Response type expected by utils.create_json_response
53
- firebase_functions_mod = types.ModuleType('firebase_functions')
54
-
55
- class FakeResponse:
56
- def __init__(self, response=None, mimetype=None, status=200, **kwargs):
57
- self.status_code = status
58
- if isinstance(response, (dict, list)):
59
- self._body_text = json.dumps(response)
60
- else:
61
- self._body_text = '' if response is None else response
62
- self.headers = kwargs.get('headers', {})
63
-
64
- def get_data(self, as_text=False):
65
- if as_text:
66
- return self._body_text
67
- return self._body_text.encode('utf-8')
68
-
69
- firebase_functions_mod.https_fn = types.SimpleNamespace(Request=object, Response=FakeResponse)
70
- sys.modules['firebase_functions'] = firebase_functions_mod
71
-
72
- # firebase_admin + auth stub
73
- firebase_admin_mod = types.ModuleType('firebase_admin')
74
- auth_mod = types.ModuleType('firebase_admin.auth')
75
-
76
- def _verify_id_token(token):
77
- return auth_verify(token)
78
-
79
- auth_mod.verify_id_token = _verify_id_token
80
- firebase_admin_mod.auth = auth_mod
81
- sys.modules['firebase_admin'] = firebase_admin_mod
82
- sys.modules['firebase_admin.auth'] = auth_mod
83
-
84
- return orig
85
-
86
-
87
- def _restore_orig(orig):
88
- for name, val in orig.items():
89
- if val is None:
90
- if name in sys.modules:
91
- del sys.modules[name]
92
- else:
93
- sys.modules[name] = val
94
-
95
-
96
- def test_get_secret_key_returns_decoded_string_on_success():
97
- orig = _setup_basic_stubs(secret_data=b'MY_SECRET')
98
- try:
99
- if 'utils' in sys.modules:
100
- del sys.modules['utils']
101
- utils = importlib.import_module('utils')
102
- res = utils.get_secret_key('any')
103
- assert res == 'MY_SECRET'
104
- finally:
105
- _restore_orig(orig)
106
-
107
-
108
- def test_get_secret_key_raises_exception_on_client_error():
109
- orig = _setup_basic_stubs(secret_side_effect=Exception('boom'))
110
- try:
111
- if 'utils' in sys.modules:
112
- del sys.modules['utils']
113
- utils = importlib.import_module('utils')
114
- with pytest.raises(Exception) as exc:
115
- utils.get_secret_key('any')
116
- assert 'Failed to retrieve secret key' in str(exc.value)
117
- assert 'boom' in str(exc.value)
118
- finally:
119
- _restore_orig(orig)
120
-
121
-
122
- def test_verify_firebase_token_raises_when_header_missing_or_invalid():
123
- orig = _setup_basic_stubs()
124
- try:
125
- if 'utils' in sys.modules:
126
- del sys.modules['utils']
127
- utils = importlib.import_module('utils')
128
-
129
- class DummyRequest:
130
- def __init__(self, headers=None):
131
- self.headers = headers or {}
132
-
133
- # Missing header
134
- req = DummyRequest(headers={})
135
- with pytest.raises(Exception) as exc:
136
- utils.verify_firebase_token(req)
137
- assert 'Missing or invalid Authorization header' in str(exc.value)
138
-
139
- # Header present but not starting with 'Bearer '
140
- req2 = DummyRequest(headers={'Authorization': 'Token abc'})
141
- with pytest.raises(Exception) as exc2:
142
- utils.verify_firebase_token(req2)
143
- assert 'Missing or invalid Authorization header' in str(exc2.value)
144
- finally:
145
- _restore_orig(orig)
146
-
147
-
148
- def test_verify_firebase_token_calls_auth_verify_and_returns_payload():
149
- orig = _setup_basic_stubs(auth_verify=lambda token: {'uid': 'user_1'})
150
- try:
151
- if 'utils' in sys.modules:
152
- del sys.modules['utils']
153
- utils = importlib.import_module('utils')
154
-
155
- class DummyRequest:
156
- def __init__(self, headers=None):
157
- self.headers = headers or {}
158
-
159
- req = DummyRequest(headers={'Authorization': 'Bearer TOK'})
160
- res = utils.verify_firebase_token(req)
161
- assert res == {'uid': 'user_1'}
162
- finally:
163
- _restore_orig(orig)
164
-
165
-
166
- def test_verify_firebase_token_wraps_auth_exceptions():
167
- def _raiser(token):
168
- raise Exception('bad token')
169
-
170
- orig = _setup_basic_stubs(auth_verify=_raiser)
171
- try:
172
- if 'utils' in sys.modules:
173
- del sys.modules['utils']
174
- utils = importlib.import_module('utils')
175
-
176
- class DummyRequest:
177
- def __init__(self, headers=None):
178
- self.headers = headers or {}
179
-
180
- req = DummyRequest(headers={'Authorization': 'Bearer TOK'})
181
- with pytest.raises(Exception) as exc:
182
- utils.verify_firebase_token(req)
183
- assert 'Authentication failed' in str(exc.value)
184
- assert 'bad token' in str(exc.value)
185
- finally:
186
- _restore_orig(orig)
187
-
188
-
189
- def test_get_api_key_uses_lowercase_family_name_and_get_secret_key(monkeypatch):
190
- orig = _setup_basic_stubs()
191
- try:
192
- if 'utils' in sys.modules:
193
- del sys.modules['utils']
194
- utils = importlib.import_module('utils')
195
-
196
- called = {}
197
-
198
- def fake_get_secret_key(name):
199
- called['name'] = name
200
- return 'THE_KEY'
201
-
202
- monkeypatch.setattr(utils, 'get_secret_key', fake_get_secret_key)
203
- from enums import LlmFamily
204
- res = utils.get_api_key(LlmFamily.OPENAI)
205
- assert res == 'THE_KEY'
206
- assert called['name'] == 'openai-api-key'
207
- finally:
208
- _restore_orig(orig)
209
-
210
-
211
- def test_create_json_response_returns_formatted_response():
212
- orig = _setup_basic_stubs()
213
- try:
214
- if 'utils' in sys.modules:
215
- del sys.modules['utils']
216
- utils = importlib.import_module('utils')
217
- resp = utils.create_json_response(True, {'a': 1}, 201)
218
- assert getattr(resp, 'status_code', None) == 201
219
- data = json.loads(resp.get_data(as_text=True))
220
- assert data == {"success": True, "payload": {"a": 1}}
221
- finally:
222
- _restore_orig(orig)
File without changes