Skip to content

Commit f10d571

Browse files
authored
fix: test assertion
fix: update test assertion for QSL_ prefix
2 parents 043c8cd + 0f1b2f8 commit f10d571

3 files changed

Lines changed: 16 additions & 8 deletions

File tree

src/quant_platform_kit/common/execution_state.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -259,9 +259,13 @@ def build_execution_marker_store_from_env(
259259
env_reader, new_key=f"{prefix}_EXECUTION_STATE_CLOUD_URI",
260260
old_key=f"{prefix}_EXECUTION_STATE_GCS_URI",
261261
)
262-
report_cloud_uri = _read_cloud_env(
263-
env_reader, new_key="EXECUTION_REPORT_CLOUD_URI",
264-
old_key="EXECUTION_REPORT_GCS_URI",
262+
report_cloud_uri = (
263+
env_reader("QSL_EXECUTION_REPORT_CLOUD_URI", None)
264+
or env_reader("QSL_EXECUTION_REPORT_GCS_URI", None)
265+
or _read_cloud_env(
266+
env_reader, new_key="EXECUTION_REPORT_CLOUD_URI",
267+
old_key="EXECUTION_REPORT_GCS_URI",
268+
)
265269
)
266270
local_dir = env_reader(f"{prefix}_EXECUTION_STATE_DIR", None)
267271
return ExecutionMarkerStore(

src/quant_platform_kit/notifications/strategy_plugin_alerts.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -123,10 +123,14 @@ def from_env(
123123
new_key="STRATEGY_PLUGIN_ALERT_STATE_CLOUD_URI",
124124
old_key="STRATEGY_PLUGIN_ALERT_STATE_GCS_URI",
125125
)
126-
report_cloud_uri = _read_cloud_env(
127-
env_reader,
128-
new_key="EXECUTION_REPORT_CLOUD_URI",
129-
old_key="EXECUTION_REPORT_GCS_URI",
126+
report_cloud_uri = (
127+
env_reader("QSL_EXECUTION_REPORT_CLOUD_URI", None)
128+
or env_reader("QSL_EXECUTION_REPORT_GCS_URI", None)
129+
or _read_cloud_env(
130+
env_reader,
131+
new_key="EXECUTION_REPORT_CLOUD_URI",
132+
old_key="EXECUTION_REPORT_GCS_URI",
133+
)
130134
)
131135
local_dir = env_reader("STRATEGY_PLUGIN_ALERT_STATE_DIR", None)
132136
return cls(

tests/test_runtime_target.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def test_resolve_runtime_target_from_env_rejects_mismatched_platform(self) -> No
102102
)
103103

104104
def test_resolve_runtime_target_from_env_requires_structured_json(self) -> None:
105-
with self.assertRaisesRegex(EnvironmentError, "RUNTIME_TARGET_JSON is required"):
105+
with self.assertRaisesRegex(EnvironmentError, "RUNTIME_TARGET_JSON"):
106106
resolve_runtime_target_from_env(env={})
107107

108108
def test_build_runtime_context_fields_merges_runtime_target_without_overwriting_fields(self) -> None:

0 commit comments

Comments
 (0)