roksta 0.2.6__cp314-cp314t-macosx_10_13_universal2.whl → 0.3.1__cp314-cp314t-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-314t-darwin.so +0 -0
  2. roksta/ai/gemini.cpython-314t-darwin.so +0 -0
  3. roksta/ai/generic.cpython-314t-darwin.so +0 -0
  4. roksta/ai/llm.cpython-314t-darwin.so +0 -0
  5. roksta/ai/openai.cpython-314t-darwin.so +0 -0
  6. roksta/ai/tools/__init__.cpython-314t-darwin.so +0 -0
  7. roksta/ai/tools/delete_file.cpython-314t-darwin.so +0 -0
  8. roksta/ai/tools/edit_file.cpython-314t-darwin.so +0 -0
  9. roksta/ai/tools/final_response.cpython-314t-darwin.so +0 -0
  10. roksta/ai/tools/get_file_summaries.cpython-314t-darwin.so +0 -0
  11. roksta/ai/tools/read_file.cpython-314t-darwin.so +0 -0
  12. roksta/ai/tools/regex_replace.cpython-314t-darwin.so +0 -0
  13. roksta/ai/tools/shell_any.cpython-314t-darwin.so +0 -0
  14. roksta/ai/tools/shell_limited.cpython-314t-darwin.so +0 -0
  15. roksta/ai/tools/tool_defs.cpython-314t-darwin.so +0 -0
  16. roksta/ai/tools/tool_utils.cpython-314t-darwin.so +0 -0
  17. roksta/ai/tools/write_file.cpython-314t-darwin.so +0 -0
  18. roksta/balance.cpython-314t-darwin.so +0 -0
  19. roksta/check_for_updates.cpython-314t-darwin.so +0 -0
  20. roksta/command_handlers/__init__.cpython-314t-darwin.so +0 -0
  21. roksta/command_handlers/handle_activate_command.cpython-314t-darwin.so +0 -0
  22. roksta/command_handlers/handle_add_funds_command.cpython-314t-darwin.so +0 -0
  23. roksta/command_handlers/handle_auto_charge_command.cpython-314t-darwin.so +0 -0
  24. roksta/command_handlers/handle_auto_commit_command.cpython-314t-darwin.so +0 -0
  25. roksta/command_handlers/handle_building_command.cpython-314t-darwin.so +0 -0
  26. roksta/command_handlers/handle_chat_command.cpython-314t-darwin.so +0 -0
  27. roksta/command_handlers/handle_dev_rate_command.cpython-314t-darwin.so +0 -0
  28. roksta/command_handlers/handle_feedback_command.cpython-314t-darwin.so +0 -0
  29. roksta/command_handlers/handle_goal_command.cpython-314t-darwin.so +0 -0
  30. roksta/command_handlers/handle_help_command.cpython-314t-darwin.so +0 -0
  31. roksta/command_handlers/handle_init_command.cpython-314t-darwin.so +0 -0
  32. roksta/command_handlers/handle_linting_command.cpython-314t-darwin.so +0 -0
  33. roksta/command_handlers/handle_login_command.cpython-314t-darwin.so +0 -0
  34. roksta/command_handlers/handle_logout_command.cpython-314t-darwin.so +0 -0
  35. roksta/command_handlers/handle_payment_details_command.cpython-314t-darwin.so +0 -0
  36. roksta/command_handlers/handle_quit_command.cpython-314t-darwin.so +0 -0
  37. roksta/command_handlers/handle_redeem_command.cpython-314t-darwin.so +0 -0
  38. roksta/command_handlers/handle_request_activation_command.cpython-314t-darwin.so +0 -0
  39. roksta/command_handlers/handle_testing_command.cpython-314t-darwin.so +0 -0
  40. roksta/command_handlers/handle_usage_command.cpython-314t-darwin.so +0 -0
  41. roksta/default_config.cpython-314t-darwin.so +0 -0
  42. roksta/env.cpython-314t-darwin.so +0 -0
  43. roksta/extended_text_area.cpython-314t-darwin.so +0 -0
  44. roksta/firebase.cpython-314t-darwin.so +0 -0
  45. roksta/fix_tests.cpython-314t-darwin.so +0 -0
  46. roksta/gen_codebase_summaries.cpython-314t-darwin.so +0 -0
  47. roksta/goal_workflow.cpython-314t-darwin.so +0 -0
  48. roksta/init_codebase.cpython-314t-darwin.so +0 -0
  49. roksta/lint_code.cpython-314t-darwin.so +0 -0
  50. roksta/main.cpython-314t-darwin.so +0 -0
  51. roksta/new_features.cpython-314t-darwin.so +0 -0
  52. roksta/roksta.cpython-314t-darwin.so +0 -0
  53. roksta/tips.cpython-314t-darwin.so +0 -0
  54. roksta/utils.cpython-314t-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-314t-darwin.so +0 -0
  70. roksta/command_handlers.cpython-314t-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=ERcDdRDi5tXqNjUI3Zpgr5_CkfUS0k_e9FvQ5FQhNCU,143
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/default_config.cpython-314t-darwin.so,sha256=UbpObJ3ght1_cN2HRPv1ddK15Fy29unc4F_aHqDhnPA,102424
7
- roksta/get_failing_tests.cpython-314t-darwin.so,sha256=TgiPDwlMZlUssV_VAxkv2iITeamCP2-QGiK5sS8mI9o,241096
8
- roksta/select_files.cpython-314t-darwin.so,sha256=IBoV8bVdIWrbQ3zRj722ui0HatrA1ZzfHZyLILn-U2I,224704
9
- roksta/run_cli_goal.cpython-314t-darwin.so,sha256=ttg_PJw87gmjhnXLV7cmYL7u9xJvmXaqemiGIOcHuPY,240624
10
- roksta/tips.cpython-314t-darwin.so,sha256=EkRn4Yk5LAKjB7o2SfEugkyqUVRVR6cpgwtBvcretm0,85864
11
- roksta/codebase_listing.cpython-314t-darwin.so,sha256=RA8iTkU9V8T-OwmCvMWgvJyB8dmIriM7ByPW8oXAE_g,188856
12
- roksta/firebase.cpython-314t-darwin.so,sha256=nFv-wCYiUcYGi23zHTZMJ1j_jWYZCGuq4ky6E_S7Spk,554432
13
- roksta/init_codebase.cpython-314t-darwin.so,sha256=ByG6ec-bIUYuKgtBDI6AxepXnTDnQ8Jkg-6VY5lVtnc,325160
14
- roksta/checkpoints.cpython-314t-darwin.so,sha256=N1gnpCyta-9dtwXbfYk4YTEerexM8NMpjx70IEsRwho,271088
15
- roksta/balance.cpython-314t-darwin.so,sha256=S9ibTGwHFDyIgJQ3LDLLqXzOz3wCnsjwNy2p7cD-3R4,357312
16
- roksta/create_default_config.cpython-314t-darwin.so,sha256=HBTlwgQSQu3BbOnUt-B2K_3g3ZdwlZdad-YxrAxhXtM,121888
17
- roksta/lint_code.cpython-314t-darwin.so,sha256=ualsyOdhOamAlssOVc34QhjAhoraigeFZmke_V9rPp8,224944
18
- roksta/clarify_goal.cpython-314t-darwin.so,sha256=m_Teljpp7EjbXTYHChiwf6eV9aHV88k9MhgaXf4DDxQ,224720
19
- roksta/firebase_auth_web.cpython-314t-darwin.so,sha256=ylHP98rxSwtAGyB15TAfJp4uGWDGUptioOc9jgM6AQ0,539064
20
- roksta/roksta.cpython-314t-darwin.so,sha256=VflDwjSxWzOdlVExGw8JaYoP4JFRw3uB-WvixyplmJw,638080
21
- roksta/firebase_config.cpython-314t-darwin.so,sha256=50G4FEKgT6xC1EkRY7t9K_sV-qaZfJ7kxhN2S9Z3Ujg,86424
22
- roksta/propose_solution.cpython-314t-darwin.so,sha256=Q2CzZCJ5OJuKk4MaiVrHbW4JeFz6VydkQwRYZBRlne0,224408
23
- roksta/parse_readme.cpython-314t-darwin.so,sha256=wt3fxkkhrw9sb7Mrx03_7l1FOACFtDHH75rX2DsONNs,224144
24
- roksta/gen_one_line_goal.cpython-314t-darwin.so,sha256=WKgBUMKhdtc-KhXRF3YdYie_ewhti8O0rdd8Re9Kof0,224680
25
- roksta/check_for_updates.cpython-314t-darwin.so,sha256=riocU3YTCHFRqEQFHVyjxK0dFO2b_dWXzsrsFVg4qZY,223688
26
- roksta/gen_codebase_summaries.cpython-314t-darwin.so,sha256=NTXYXXDWR09rqaoJcyB3ccydt_ygP6SbvuBn_HEznqk,704192
27
- roksta/main.cpython-314t-darwin.so,sha256=LIEkYChJsmyOzavCOOy9ZP0X36SmGOCSMLHeD9K3kAo,155432
28
- roksta/enums.cpython-314t-darwin.so,sha256=zy6rRIEXtNW37BA76zR9tPzMA_j-wG-Si6f2e6C7gKc,322320
29
- roksta/__init__.cpython-314t-darwin.so,sha256=wi8Vybo_OIgiP0nw3dYRY1WuU-_XToX7Kg214k5KG1Q,85824
30
- roksta/goal_workflow.cpython-314t-darwin.so,sha256=j5LmOhSASpWPdr_1W9T7eS2y3bZ-p2oVsWpMu4U05QQ,554376
31
- roksta/analytics.cpython-314t-darwin.so,sha256=QPUKzsnhI6HbbYK4ek6kWkYgtzAofem0JSL8hV5Qy1Q,322928
32
- roksta/write_code.cpython-314t-darwin.so,sha256=bmPIEDgwjHioODH89S8OvOp-2_L1C533MTvmBBJ5MR8,207920
33
- roksta/env.cpython-314t-darwin.so,sha256=Cw3neGkv8Jp4jloXVlpzQ087XJt03v9MxvPWu_oA96g,87928
34
- roksta/new_features.cpython-314t-darwin.so,sha256=OjOT_2DDy3kv1Nh5FMkNQbAcFarJQ9Nr2dg6lDpgVAs,85904
35
- roksta/build_project.cpython-314t-darwin.so,sha256=4F7-EKDWTNCFQ3laC2_qU_5Y7hB7YhUntTpCY_84WNM,207960
36
- roksta/fix_tests.cpython-314t-darwin.so,sha256=F-kVhZDW9Is5MqeQSyN_ggOj077Z9Rr_FwqHTFdI9BM,257632
37
- roksta/logger.cpython-314t-darwin.so,sha256=upcimtFREbc9WKSwGoEBypFX_uM0jwZqjfvOvIhyKjw,155216
38
- roksta/chat_workflow.cpython-314t-darwin.so,sha256=Sskw4w_0WaxhEaLN3diC3fbnUK_9lPH_0mzpoQX7hy0,290840
39
- roksta/response_formats.cpython-314t-darwin.so,sha256=8pWaDZuRf-9ZUpR8p8kZkIOg0ovPKGbQ44DE1i2ZRng,121208
40
- roksta/get_codebase_structure.cpython-314t-darwin.so,sha256=L4OJLaFY0K-rjQYH3ur_w01ag2SC3JojmJapfcEl_Ts,239520
41
- roksta/make_issue.cpython-314t-darwin.so,sha256=Maf34ULyKoPtPrHwISHikTunVNkQ0ATIJBbea8tQYWw,222240
42
- roksta/rewrite_goal.cpython-314t-darwin.so,sha256=-MtU1o7UfREP8kOXzpmtz5zTxu6gcEoRYh_I2_byl8k,207872
43
- roksta/utils.cpython-314t-darwin.so,sha256=HhRj4g1rERoQ2IhI8IKsgmXbdTBHW35MQTmUVXRVEVs,671920
44
- roksta/command_handlers.cpython-314t-darwin.so,sha256=1eQMu1_T-GqbQ_V2lIgGyiFGNgSW0DRSOhWCIKpSJnc,1033016
45
- roksta/extended_text_area.cpython-314t-darwin.so,sha256=AjevxgJUgNEkvoMUnXJk9nchm9f0qywZy8icE2jvMN8,324072
46
- roksta/ai/openai.cpython-314t-darwin.so,sha256=j4yCqiK_uElxKvQMfCaN4z6RTzc9o3QpJXhRTM4AC1M,323840
47
- roksta/ai/tools.cpython-314t-darwin.so,sha256=1VygY_YkIya9zupdAmLkf-KRR1-yLCm7Q0oxWEwpSHg,720256
48
- roksta/ai/call_ai.cpython-314t-darwin.so,sha256=-Ce5avfqFGG2rm43Glb8NhGM229oVUxSI7D6OP3dgaE,257648
49
- roksta/ai/__init__.cpython-314t-darwin.so,sha256=ADxSpO91HkA6gk1RUfnjAxO8etJfH2MS2Ku7jcZI2JI,85824
50
- roksta/ai/generic.cpython-314t-darwin.so,sha256=R-If_Yba-rBzCC0KOpffmHudR39AMCVGo-FHY4DSUAA,340256
51
- roksta/ai/gemini.cpython-314t-darwin.so,sha256=4P7ZdIqfGpkGPX2glwNA7Bgpb-nkRQ4ti_AuyfSqecQ,439584
52
- roksta/ai/llm.cpython-314t-darwin.so,sha256=njLeHP8mcz_BeERAgcvf8pjX3dwZMEMQhXYpRChVxeU,271880
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