128 lines
4.7 KiB
Python
128 lines
4.7 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
import unittest
|
|
|
|
from sqlalchemy import create_engine, event
|
|
from sqlalchemy.orm import Session
|
|
|
|
from govoplan_access.backend.db.models import Account, Group, User
|
|
from govoplan_core.core.change_sequence import ChangeSequenceEntry
|
|
from govoplan_core.db.base import Base
|
|
from govoplan_files.backend.db.models import (
|
|
FileAsset,
|
|
FileConnectorCredential,
|
|
FileConnectorPolicy,
|
|
FileConnectorProfile,
|
|
FileConnectorSpace,
|
|
FileFormEvidenceGrant,
|
|
)
|
|
from govoplan_files.backend.manifest import _tenant_summary_batch
|
|
|
|
|
|
class FilesTenantSummaryBatchTests(unittest.TestCase):
|
|
def test_batch_summary_uses_one_grouped_query_per_owned_table(self) -> None:
|
|
engine = create_engine("sqlite+pysqlite:///:memory:")
|
|
Base.metadata.create_all(
|
|
engine,
|
|
tables=[
|
|
Account.__table__,
|
|
User.__table__,
|
|
Group.__table__,
|
|
FileAsset.__table__,
|
|
FileConnectorCredential.__table__,
|
|
FileConnectorPolicy.__table__,
|
|
FileConnectorProfile.__table__,
|
|
FileConnectorSpace.__table__,
|
|
FileFormEvidenceGrant.__table__,
|
|
ChangeSequenceEntry.__table__,
|
|
],
|
|
)
|
|
try:
|
|
with Session(engine) as session:
|
|
session.add_all(
|
|
[
|
|
FileAsset(
|
|
id="file-1",
|
|
tenant_id="tenant-1",
|
|
owner_type="tenant",
|
|
display_path="/one.txt",
|
|
filename="one.txt",
|
|
),
|
|
FileConnectorCredential(
|
|
id="credential-1",
|
|
tenant_id="tenant-1",
|
|
label="Credential",
|
|
),
|
|
FileConnectorPolicy(
|
|
id="policy-1",
|
|
tenant_id="tenant-1",
|
|
scope_type="tenant",
|
|
),
|
|
FileConnectorProfile(
|
|
id="profile-1",
|
|
tenant_id="tenant-1",
|
|
label="Profile",
|
|
provider="webdav",
|
|
),
|
|
FileConnectorSpace(
|
|
id="space-1",
|
|
tenant_id="tenant-1",
|
|
owner_type="tenant",
|
|
label="Space",
|
|
connector_profile_id="profile-1",
|
|
provider="webdav",
|
|
),
|
|
FileFormEvidenceGrant(
|
|
id="grant-1",
|
|
tenant_id="tenant-1",
|
|
form_instance_id="form-instance-1",
|
|
form_definition_id="form-1",
|
|
form_definition_revision="1",
|
|
token_sha256="a" * 64,
|
|
idempotency_key="grant-1",
|
|
request_sha256="b" * 64,
|
|
custodian_user_id="user-1",
|
|
evidence_kind="document",
|
|
purpose="submission attachment",
|
|
expires_at=datetime.now(UTC),
|
|
max_size_bytes=1024,
|
|
),
|
|
]
|
|
)
|
|
session.commit()
|
|
query_count = 0
|
|
|
|
def count_query(*_args: object) -> None:
|
|
nonlocal query_count
|
|
query_count += 1
|
|
|
|
event.listen(engine, "before_cursor_execute", count_query)
|
|
try:
|
|
counts = _tenant_summary_batch(
|
|
session,
|
|
["tenant-1", "tenant-empty"],
|
|
)
|
|
finally:
|
|
event.remove(engine, "before_cursor_execute", count_query)
|
|
|
|
self.assertEqual(6, query_count)
|
|
self.assertEqual(
|
|
{
|
|
"files": 1,
|
|
"connector_credentials": 1,
|
|
"connector_policies": 1,
|
|
"connector_profiles": 1,
|
|
"connector_spaces": 1,
|
|
"form_evidence_upload_grants": 1,
|
|
},
|
|
counts["tenant-1"],
|
|
)
|
|
self.assertEqual(0, counts["tenant-empty"]["files"])
|
|
finally:
|
|
engine.dispose()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|