roksta 0.2.6__cp313-cp313-macosx_10_13_universal2.whl → 0.3.1__cp313-cp313-macosx_10_13_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 (76) hide show
  1. roksta/ai/call_ai.cpython-313-darwin.so +0 -0
  2. roksta/ai/gemini.cpython-313-darwin.so +0 -0
  3. roksta/ai/generic.cpython-313-darwin.so +0 -0
  4. roksta/ai/llm.cpython-313-darwin.so +0 -0
  5. roksta/ai/openai.cpython-313-darwin.so +0 -0
  6. roksta/ai/tools/__init__.cpython-313-darwin.so +0 -0
  7. roksta/ai/tools/delete_file.cpython-313-darwin.so +0 -0
  8. roksta/ai/tools/edit_file.cpython-313-darwin.so +0 -0
  9. roksta/ai/tools/final_response.cpython-313-darwin.so +0 -0
  10. roksta/ai/tools/get_file_summaries.cpython-313-darwin.so +0 -0
  11. roksta/ai/tools/read_file.cpython-313-darwin.so +0 -0
  12. roksta/ai/tools/regex_replace.cpython-313-darwin.so +0 -0
  13. roksta/ai/tools/shell_any.cpython-313-darwin.so +0 -0
  14. roksta/ai/tools/shell_limited.cpython-313-darwin.so +0 -0
  15. roksta/ai/tools/tool_defs.cpython-313-darwin.so +0 -0
  16. roksta/ai/tools/tool_utils.cpython-313-darwin.so +0 -0
  17. roksta/ai/tools/write_file.cpython-313-darwin.so +0 -0
  18. roksta/balance.cpython-313-darwin.so +0 -0
  19. roksta/check_for_updates.cpython-313-darwin.so +0 -0
  20. roksta/command_handlers/__init__.cpython-313-darwin.so +0 -0
  21. roksta/command_handlers/handle_activate_command.cpython-313-darwin.so +0 -0
  22. roksta/command_handlers/handle_add_funds_command.cpython-313-darwin.so +0 -0
  23. roksta/command_handlers/handle_auto_charge_command.cpython-313-darwin.so +0 -0
  24. roksta/command_handlers/handle_auto_commit_command.cpython-313-darwin.so +0 -0
  25. roksta/command_handlers/handle_building_command.cpython-313-darwin.so +0 -0
  26. roksta/command_handlers/handle_chat_command.cpython-313-darwin.so +0 -0
  27. roksta/command_handlers/handle_dev_rate_command.cpython-313-darwin.so +0 -0
  28. roksta/command_handlers/handle_feedback_command.cpython-313-darwin.so +0 -0
  29. roksta/command_handlers/handle_goal_command.cpython-313-darwin.so +0 -0
  30. roksta/command_handlers/handle_help_command.cpython-313-darwin.so +0 -0
  31. roksta/command_handlers/handle_init_command.cpython-313-darwin.so +0 -0
  32. roksta/command_handlers/handle_linting_command.cpython-313-darwin.so +0 -0
  33. roksta/command_handlers/handle_login_command.cpython-313-darwin.so +0 -0
  34. roksta/command_handlers/handle_logout_command.cpython-313-darwin.so +0 -0
  35. roksta/command_handlers/handle_payment_details_command.cpython-313-darwin.so +0 -0
  36. roksta/command_handlers/handle_quit_command.cpython-313-darwin.so +0 -0
  37. roksta/command_handlers/handle_redeem_command.cpython-313-darwin.so +0 -0
  38. roksta/command_handlers/handle_request_activation_command.cpython-313-darwin.so +0 -0
  39. roksta/command_handlers/handle_testing_command.cpython-313-darwin.so +0 -0
  40. roksta/command_handlers/handle_usage_command.cpython-313-darwin.so +0 -0
  41. roksta/default_config.cpython-313-darwin.so +0 -0
  42. roksta/env.cpython-313-darwin.so +0 -0
  43. roksta/extended_text_area.cpython-313-darwin.so +0 -0
  44. roksta/firebase.cpython-313-darwin.so +0 -0
  45. roksta/fix_tests.cpython-313-darwin.so +0 -0
  46. roksta/gen_codebase_summaries.cpython-313-darwin.so +0 -0
  47. roksta/goal_workflow.cpython-313-darwin.so +0 -0
  48. roksta/init_codebase.cpython-313-darwin.so +0 -0
  49. roksta/lint_code.cpython-313-darwin.so +0 -0
  50. roksta/main.cpython-313-darwin.so +0 -0
  51. roksta/new_features.cpython-313-darwin.so +0 -0
  52. roksta/roksta.cpython-313-darwin.so +0 -0
  53. roksta/tips.cpython-313-darwin.so +0 -0
  54. roksta/utils.cpython-313-darwin.so +0 -0
  55. {roksta-0.2.6.dist-info → roksta-0.3.1.dist-info}/METADATA +2 -1
  56. roksta-0.3.1.dist-info/RECORD +109 -0
  57. tests/conftest.py +42 -0
  58. tests/functions/{api_v0_01 → api_v1_01}/__init__.py +1 -1
  59. tests/functions/{api_v0_01 → api_v1_01}/test__analytics.py +2 -3
  60. tests/functions/{api_v0_01 → api_v1_01}/test__gemini_proxy.py +51 -6
  61. tests/functions/{api_v0_01 → api_v1_01}/test__generic_proxy.py +31 -2
  62. tests/functions/{api_v0_01 → api_v1_01}/test__get_payment_details.py +2 -2
  63. tests/functions/{api_v0_01 → api_v1_01}/test__openai_proxy.py +50 -14
  64. tests/functions/{api_v0_01 → api_v1_01}/test__redeem_credit_code.py +2 -2
  65. tests/functions/{api_v0_01 → api_v1_01}/test__sync_emails.py +3 -2
  66. tests/functions/{api_v0_01 → api_v1_01}/test__take_payment.py +2 -2
  67. tests/functions/{api_v0_01 → api_v1_01}/test__use_activation_code.py +3 -2
  68. tests/functions/test_utils.py +484 -0
  69. roksta/ai/tools.cpython-313-darwin.so +0 -0
  70. roksta/command_handlers.cpython-313-darwin.so +0 -0
  71. roksta-0.2.6.dist-info/RECORD +0 -78
  72. tests/functions/test_utils_functions.py +0 -222
  73. {roksta-0.2.6.dist-info → roksta-0.3.1.dist-info}/WHEEL +0 -0
  74. {roksta-0.2.6.dist-info → roksta-0.3.1.dist-info}/entry_points.txt +0 -0
  75. {roksta-0.2.6.dist-info → roksta-0.3.1.dist-info}/top_level.txt +0 -0
  76. /tests/functions/{test_main_functions.py → test_main.py} +0 -0
@@ -1,78 +0,0 @@
1
- roksta-0.2.6.dist-info/RECORD,,
2
- roksta-0.2.6.dist-info/WHEEL,sha256=YOKbiIEZep2WTRrpKV0dvEfsSeIOVH2NqduVxoKQt6I,142
3
- roksta-0.2.6.dist-info/entry_points.txt,sha256=mzRdYg_DlzZRwjxYUt9-gyoRCkM1QBTeTbwETgiTdGw,44
4
- roksta-0.2.6.dist-info/top_level.txt,sha256=lvciNZQ1dPGXpiCLdWVXK03n9fKHjbQdwjqQbnUjeYM,13
5
- roksta-0.2.6.dist-info/METADATA,sha256=gBXNX4iwCT9vychi5wNnmihf7ptAXoiAymfGKYchiNo,1427
6
- roksta/command_handlers.cpython-313-darwin.so,sha256=qda6tnEvSVJYlZxOuCWcYGAkkQqF2TWxCZZ9ZksY20s,950088
7
- roksta/analytics.cpython-313-darwin.so,sha256=iJvu-a1EGDlio1bxufPn9tk7QzQlv_oCIY7P7VKdq6I,273168
8
- roksta/check_for_updates.cpython-313-darwin.so,sha256=9UTWaI6rhY9MJC7jW8pvhiLTDH-z1fZ2y0UMCOvKkQs,206904
9
- roksta/response_formats.cpython-313-darwin.so,sha256=9MmD_FwyFDeKd8-Zr8xjmc68_lqPX1FZZcygbJQdu1M,120968
10
- roksta/utils.cpython-313-darwin.so,sha256=z3_9_CpCNlsqJEAjCdMY1zoUPM424LK28yX3i_wng3g,589128
11
- roksta/main.cpython-313-darwin.so,sha256=3NyhZAgrWzmR2dYqq5NANg69St9Jwdo9udE-6HNJmvw,122184
12
- roksta/env.cpython-313-darwin.so,sha256=RGJ2m5ZSV6n25Sc8K21zrQl6WhI--VZNUf9iGjPzmDs,87832
13
- roksta/get_failing_tests.cpython-313-darwin.so,sha256=tLzNpROqCTf-X8_B2Avv_dDSfU-_7D064lxoJ29GN3Q,224296
14
- roksta/fix_tests.cpython-313-darwin.so,sha256=EdUGStPlnC3BNk7DNeZpxSwb8WOKdMZISFR748CYpnQ,240704
15
- roksta/firebase_auth_web.cpython-313-darwin.so,sha256=LduXEbNR5K7Ri7GNmsiAyaZW7Lq3jY9fiw4ovlrPLK4,472664
16
- roksta/build_project.cpython-313-darwin.so,sha256=yOFozWXQoqi9yZ_AhrOWH1hO8i1G_ETEbsPNAVl_Vwo,191184
17
- roksta/rewrite_goal.cpython-313-darwin.so,sha256=syFaNtvZI0UAUsfsTBG6seUCP6FBUKM5vDhYfvi9pcY,191088
18
- roksta/run_cli_goal.cpython-313-darwin.so,sha256=LnBKx9tOvohdAsLqlprdzk0FOx9aI8anNBKoFDXAtdE,223968
19
- roksta/clarify_goal.cpython-313-darwin.so,sha256=TWlse-HG03UP5ks8esK7M0KEGxZYbdeMEkd9eO-5hp8,207808
20
- roksta/roksta.cpython-313-darwin.so,sha256=xZ4dfNZM2p1N-6gTgXG3roP4m0QX9fHTi44DR3SPTTU,571632
21
- roksta/default_config.cpython-313-darwin.so,sha256=CEOgjGLlkENoK58rGQ2CeppwFvsR2fO21NyyTOJyRJo,102248
22
- roksta/__init__.cpython-313-darwin.so,sha256=9NPz72fxGkIQXx2OtFYdnleayh6ielrbliTgVxswNCc,85648
23
- roksta/codebase_listing.cpython-313-darwin.so,sha256=MrLu0cfolSbjQKv6Y9HZP8SRMzVOtSpmVmJS7YidByk,155608
24
- roksta/make_issue.cpython-313-darwin.so,sha256=BRmIoFb5UL_FqoY-NTHKyJFGRzw2GbjddJI02uS9nZ8,221856
25
- roksta/create_default_config.cpython-313-darwin.so,sha256=yWxoCIZpvdvv3nSW29_3hKY2xe9R3Hph7GWPYvZJ4bc,121608
26
- roksta/gen_codebase_summaries.cpython-313-darwin.so,sha256=Q4jftsyTfDIIxibQpfCBj2x1PdXPH6il7tXHA0WJZeQ,621568
27
- roksta/propose_solution.cpython-313-darwin.so,sha256=mGEgZ6KldraDTbvBdP_x7Vvd5bZTy6VMWQmZ10Jh9x8,207496
28
- roksta/balance.cpython-313-darwin.so,sha256=OSdYm3L2CHEunnka4bcXFcZ5iw7pI-U18geHk2YjOpo,340432
29
- roksta/logger.cpython-313-darwin.so,sha256=Twgl3_qXUJ0c3gVV5ek-pHZUtqaAzVsmt4rlNIaMCYg,122032
30
- roksta/lint_code.cpython-313-darwin.so,sha256=aHh_KZ-bBRXLOe9t0f-Vw6rgwVorID3YPjmLG09EWIM,208048
31
- roksta/extended_text_area.cpython-313-darwin.so,sha256=qiKWR3-mZMyrD2tytHEA9zrlrVhGIFNOMkHFMfkov74,307208
32
- roksta/checkpoints.cpython-313-darwin.so,sha256=kyBklHRV54PUjhwdeKD0xBPXVb2AQTp-IuqRDSufJrc,254320
33
- roksta/get_codebase_structure.cpython-313-darwin.so,sha256=SCMaKJp2Mf_GkOp-TW-Rc7pFnYovHdZz44B3CG7-IDk,206240
34
- roksta/init_codebase.cpython-313-darwin.so,sha256=iMX6o1d5yIIZyMu6mmJW-Fk7QSQ0J4bX0VeMxsUKYSw,291824
35
- roksta/gen_one_line_goal.cpython-313-darwin.so,sha256=j1_I9b-5ZBfRga_YSATSsKpPFv24LosQ2b-uchv5cOc,207768
36
- roksta/new_features.cpython-313-darwin.so,sha256=In_WCFLFSYl__Uhu2PlIisavXZlFnSol1_OMLb-r7A4,85728
37
- roksta/firebase_config.cpython-313-darwin.so,sha256=mvrDiJcrsSU4E5c4SQakAnPT0gXvYEzex8H1IdCobdI,86328
38
- roksta/chat_workflow.cpython-313-darwin.so,sha256=KNPbclDrtzlWl7SWehDTQBhI9jEh06PrnzwpPx4g7Xk,273888
39
- roksta/select_files.cpython-313-darwin.so,sha256=FjCx9U4P4K14hK4IA5sULYTvnd-LraTlOaOvb9JDPfc,207792
40
- roksta/tips.cpython-313-darwin.so,sha256=ChSe7J5lQhO7xm1IqKIR0_qAQmOCD0AttuaNS6zTay4,85688
41
- roksta/write_code.cpython-313-darwin.so,sha256=CPbozpwjCoRmDnhVs4TyuEp6Veo32kdigdB8NO4ozQ0,191152
42
- roksta/enums.cpython-313-darwin.so,sha256=N0YUW8hXLJ_Zr0dwbnyszPlU-Xt5pb1NETdnYCKPY-k,338344
43
- roksta/parse_readme.cpython-313-darwin.so,sha256=MjcG8euvVPS3m4UsEdUm9f_lFLRdwIREsJlsspD_qtg,207280
44
- roksta/firebase.cpython-313-darwin.so,sha256=z7M7rhJEIr5YTJ45Io-5_Gm7XTkTviyN9PbzyUpeMxU,488176
45
- roksta/goal_workflow.cpython-313-darwin.so,sha256=DRLEnFs8PPKFIHxwulpu0n0uPjZGn0Cfgkllqr2bK90,504496
46
- roksta/ai/generic.cpython-313-darwin.so,sha256=wic1Tq1BNy1yrNvTXwykMrlOeroud80ueDf1xEeTqy0,306944
47
- roksta/ai/__init__.cpython-313-darwin.so,sha256=gnpXADu43zbYSsPXDFkSMAm1_tnAAFb9g3UClsQVlSQ,85648
48
- roksta/ai/tools.cpython-313-darwin.so,sha256=jgY3s6Z7ogwMg51-U_FmnePYLrQWJ0D4K26cmlx1hpU,653976
49
- roksta/ai/openai.cpython-313-darwin.so,sha256=PzQfRWUzZQP6tPTB3pFPcqy8zM1tWFGwETllE8sW2pQ,306928
50
- roksta/ai/call_ai.cpython-313-darwin.so,sha256=gfKcuUEKzcgUxeKbjAgGhbxqJaTQAwreVoyMKIl6UBs,240784
51
- roksta/ai/llm.cpython-313-darwin.so,sha256=vgstbNLxeOb5fQve1RgEcrw9X_jYzoGG76hpCmIqrko,271496
52
- roksta/ai/gemini.cpython-313-darwin.so,sha256=s9HCxBZihX1bv9kZomcticx7HqhP3LlO0ApPgXRl7fU,422624
53
- tests/conftest.py,sha256=WBja46X2wduZRF2t4LARhbNvLlU8oRhSW_HB6ara8Yg,6295
54
- tests/__init__.py,sha256=J6ztnXNqM8TSQCgHXMWb6NRq1kN_h4ql63Jq-TYR8XE,124
55
- tests/functions/test_auth.py,sha256=I3oFTw2LTuUsnQ75v5SCfgUPnpel_pDKqUH0VAJpXQw,992
56
- tests/functions/__init__.py,sha256=mLXx3UARcv07vs_fGJuy3L5ZRn9IRsF9zIuL5IOnwdk,95
57
- tests/functions/test_utils_functions.py,sha256=gS-SxnK0QnVUooxcXlJKodjqOB1HVmUxufwzUPQ7qAk,7368
58
- tests/functions/test_main_functions.py,sha256=h0W_4ISo95WzRy2HpXZw3cuNOTcYk_2QCIZffZJEfEU,2575
59
- tests/functions/api_v1_00/test__sync_emails.py,sha256=OTyf2YbBK7e4pWIQTUGYR4x6il1Hb3g4FYqQ_fRj1nA,12699
60
- tests/functions/api_v1_00/test__take_payment.py,sha256=I-XJasGuLW5wTMAu4nAgFkA8Mo_zVpGPvkJ2LrrJGCk,18022
61
- tests/functions/api_v1_00/test__gemini_proxy.py,sha256=mf8ubRmNxxxN80H6EA-iDPwu6Oue6Rmgxl6tiFNpfrU,13344
62
- tests/functions/api_v1_00/test__analytics.py,sha256=oKjd7_zjnttcb-5BuA93yllkm0OOF4YWnfTUW1-icnM,15806
63
- tests/functions/api_v1_00/__init__.py,sha256=3RCSoj_5UQPw9Z5NeAgYz2s4gfcZIi7g36qk_Ed5TiA,124
64
- tests/functions/api_v1_00/test__use_activation_code.py,sha256=tNyTkkRtrpjei-7F31UNwrcU-PbP-QnkWNU7h2n5Yqo,16960
65
- tests/functions/api_v1_00/test__redeem_credit_code.py,sha256=7GD3tGgO2Lk0BCjUmzZACJiwocEhLBvXrrut0x13IWU,6347
66
- tests/functions/api_v1_00/test__openai_proxy.py,sha256=xlO2zmG64LlrKrbjFz-ds6Map7vy4SjIuHfZJDAqGK8,17726
67
- tests/functions/api_v1_00/test__get_payment_details.py,sha256=i5Px3NkLxqw8D3tVPrYdRJJF5jlNmysSuX_NNolbIa4,13735
68
- tests/functions/api_v1_00/test__generic_proxy.py,sha256=KZkOtmc71aOnWlz2L7B6CFhRNEV73bExG3GR14oD2L0,17673
69
- tests/functions/api_v0_01/test__sync_emails.py,sha256=zCfuh1M6qJlGaCpCWuUE0Ao9Pp5pVkBJXXjoctldkpk,12698
70
- tests/functions/api_v0_01/test__take_payment.py,sha256=2WDv_7cYTozVMl7hDLNDTF9gmfrgna3oR8xgfRa-plA,18022
71
- tests/functions/api_v0_01/test__gemini_proxy.py,sha256=R-lzyVsNtqJFGILKhgy1cUIHg1SYPSFKNDbKp2fua-k,11152
72
- tests/functions/api_v0_01/test__analytics.py,sha256=nMs_aYuXDEFWz9aY21MYwbgugF-czRbucAzis4-8kgc,15807
73
- tests/functions/api_v0_01/__init__.py,sha256=BOIFrFcANRAbpnX0Bez6rtkj6xJnOli1fqzFX5SpkV0,124
74
- tests/functions/api_v0_01/test__use_activation_code.py,sha256=EgifjM9wkW1fkQoBK9z5PEEM5xwpRHyt_g7QWy6xrec,16959
75
- tests/functions/api_v0_01/test__redeem_credit_code.py,sha256=WSs7CE2pkHsied_tg3XCR3RJ9blwQEkOzy-s3lkV_w0,6347
76
- tests/functions/api_v0_01/test__openai_proxy.py,sha256=Nda-RDUVYBQl2wI5mxxr9Ca-9k9HJV5EgEmdIvyytGs,15761
77
- tests/functions/api_v0_01/test__get_payment_details.py,sha256=3LYAp9m1RDsf928HGJ6P_YTji7F4srkDv0iWcKTwW54,13735
78
- tests/functions/api_v0_01/test__generic_proxy.py,sha256=cJftek6njsphK_yesUX1QxphGcD5TEkhQNC5ROvnaIg,16190
@@ -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