apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-06 22:12:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

test: rename SearchSpace -> Workspace across tests + fixtures (Phase 2 Wave F)

Browse source 
Apply the same rename to surfsense_backend/tests: workspace_id fields/vars,
Workspace* classes/schemas, table name searchspaces -> workspaces in raw SQL,
and the API URL spellings -> /workspaces. Preserves the carve-out wire literals
tests assert (Celery task names, OTel key "search_space.id").
This commit is contained in:
CREDO23 2026-06-26 18:36:46 +02:00
parent 56826a63bc
commit ca9bd28934
124 changed files with 1269 additions and 1269 deletions
Download patch file Download diff file Expand all files Collapse all files

14
surfsense_backend/tests/unit/tasks/chat/streaming/test_llm_bundle.py
View file

@ -32,12 +32,12 @@ def _patch_common_bundle_dependencies(monkeypatch: pytest.MonkeyPatch):
_CapturedChatLiteLLM.calls = []
async def _fake_search_space(
_session: Any, _search_space_id: int
async def _fake_workspace(
_session: Any, _workspace_id: int
) -> SimpleNamespace:
return SimpleNamespace(id=42, user_id="user-1")
monkeypatch.setattr(llm_bundle, "_load_search_space", _fake_search_space)
monkeypatch.setattr(llm_bundle, "_load_workspace", _fake_workspace)
monkeypatch.setattr(llm_bundle, "SanitizedChatLiteLLM", _CapturedChatLiteLLM)
monkeypatch.setattr(llm_bundle, "register_model_usage_metadata", lambda **_kw: None)
monkeypatch.setattr(
@ -65,9 +65,9 @@ async def test_load_llm_bundle_enables_streaming_for_db_models(
connection=connection,
)
async def _fake_db_model(_session: Any, *, model_id: int, search_space: Any) -> Any:
async def _fake_db_model(_session: Any, *, model_id: int, workspace: Any) -> Any:
assert model_id == 7
assert search_space.id == 42
assert workspace.id == 42
return model
monkeypatch.setattr(llm_bundle, "_load_db_model", _fake_db_model)
@ -83,7 +83,7 @@ async def test_load_llm_bundle_enables_streaming_for_db_models(
llm, agent_config, error = await llm_bundle.load_llm_bundle(
object(),
config_id=7,
search_space_id=42,
workspace_id=42,
)
assert error is None
@ -140,7 +140,7 @@ async def test_load_llm_bundle_enables_streaming_for_global_models(
llm, agent_config, error = await llm_bundle.load_llm_bundle(
object(),
config_id=-11,
search_space_id=42,
workspace_id=42,
)
assert error is None

36
surfsense_backend/tests/unit/tasks/test_video_presentation_billing.py
View file

@ -145,14 +145,14 @@ async def test_billable_call_invoked_with_correct_kwargs_for_free_config(monkeyp
user_id = uuid4()
async def _fake_resolver(sess, search_space_id, *, thread_id=None):
assert search_space_id == 777
async def _fake_resolver(sess, workspace_id, *, thread_id=None):
assert workspace_id == 777
assert thread_id == 99
return user_id, "free", "openrouter/some-free-model"
monkeypatch.setattr(
video_presentation_tasks,
"_resolve_agent_billing_for_search_space",
"_resolve_agent_billing_for_workspace",
_fake_resolver,
)
monkeypatch.setattr(video_presentation_tasks, "billable_call", _ok_billable_call)
@ -169,7 +169,7 @@ async def test_billable_call_invoked_with_correct_kwargs_for_free_config(monkeyp
result = await video_presentation_tasks._generate_video_presentation(
video_presentation_id=11,
source_content="content",
search_space_id=777,
workspace_id=777,
user_prompt=None,
)
@ -180,7 +180,7 @@ async def test_billable_call_invoked_with_correct_kwargs_for_free_config(monkeyp
assert len(_CALL_LOG) == 1
call = _CALL_LOG[0]
assert call["user_id"] == user_id
assert call["search_space_id"] == 777
assert call["workspace_id"] == 777
assert call["billing_tier"] == "free"
assert call["base_model"] == "openrouter/some-free-model"
assert call["usage_type"] == "video_presentation_generation"
@ -213,12 +213,12 @@ async def test_billable_call_invoked_with_premium_tier(monkeypatch):
user_id = uuid4()
async def _fake_resolver(sess, search_space_id, *, thread_id=None):
async def _fake_resolver(sess, workspace_id, *, thread_id=None):
return user_id, "premium", "gpt-5.4"
monkeypatch.setattr(
video_presentation_tasks,
"_resolve_agent_billing_for_search_space",
"_resolve_agent_billing_for_workspace",
_fake_resolver,
)
monkeypatch.setattr(video_presentation_tasks, "billable_call", _ok_billable_call)
@ -235,7 +235,7 @@ async def test_billable_call_invoked_with_premium_tier(monkeypatch):
await video_presentation_tasks._generate_video_presentation(
video_presentation_id=11,
source_content="content",
search_space_id=777,
workspace_id=777,
user_prompt=None,
)
@ -256,12 +256,12 @@ async def test_quota_insufficient_marks_video_failed_and_skips_graph(monkeypatch
lambda: _FakeSessionMaker(session),
)
async def _fake_resolver(sess, search_space_id, *, thread_id=None):
async def _fake_resolver(sess, workspace_id, *, thread_id=None):
return uuid4(), "premium", "gpt-5.4"
monkeypatch.setattr(
video_presentation_tasks,
"_resolve_agent_billing_for_search_space",
"_resolve_agent_billing_for_workspace",
_fake_resolver,
)
monkeypatch.setattr(
@ -283,7 +283,7 @@ async def test_quota_insufficient_marks_video_failed_and_skips_graph(monkeypatch
result = await video_presentation_tasks._generate_video_presentation(
video_presentation_id=12,
source_content="content",
search_space_id=777,
workspace_id=777,
user_prompt=None,
)
@ -309,12 +309,12 @@ async def test_billing_settlement_failure_marks_video_failed(monkeypatch):
lambda: _FakeSessionMaker(session),
)
async def _fake_resolver(sess, search_space_id, *, thread_id=None):
async def _fake_resolver(sess, workspace_id, *, thread_id=None):
return uuid4(), "premium", "gpt-5.4"
monkeypatch.setattr(
video_presentation_tasks,
"_resolve_agent_billing_for_search_space",
"_resolve_agent_billing_for_workspace",
_fake_resolver,
)
monkeypatch.setattr(
@ -335,7 +335,7 @@ async def test_billing_settlement_failure_marks_video_failed(monkeypatch):
result = await video_presentation_tasks._generate_video_presentation(
video_presentation_id=14,
source_content="content",
search_space_id=777,
workspace_id=777,
user_prompt=None,
)
@ -360,12 +360,12 @@ async def test_resolver_failure_marks_video_failed(monkeypatch):
lambda: _FakeSessionMaker(session),
)
async def _failing_resolver(sess, search_space_id, *, thread_id=None):
raise ValueError("Search space 777 not found")
async def _failing_resolver(sess, workspace_id, *, thread_id=None):
raise ValueError("Workspace 777 not found")
monkeypatch.setattr(
video_presentation_tasks,
"_resolve_agent_billing_for_search_space",
"_resolve_agent_billing_for_workspace",
_failing_resolver,
)
@ -384,7 +384,7 @@ async def test_resolver_failure_marks_video_failed(monkeypatch):
result = await video_presentation_tasks._generate_video_presentation(
video_presentation_id=13,
source_content="content",
search_space_id=777,
workspace_id=777,
user_prompt=None,
)