1212from app .domain .entities .user import User
1313from app .domain .value_objects .membership_status import MembershipStatus
1414
15-
1615@dataclass
1716class FakeUserRepo :
1817 by_email : dict [str , User ] = field (default_factory = dict )
@@ -33,7 +32,6 @@ async def update_last_login(self, *, user_id: UUID) -> None:
3332 if user :
3433 user .last_login_at = user .last_login_at
3534
36-
3735@dataclass
3836class FakeOrgRepo :
3937 slugs : set [str ] = field (default_factory = set )
@@ -47,7 +45,6 @@ async def create_org(self, *, org: Organization) -> Organization:
4745 self .orgs [org .id ] = org
4846 return org
4947
50-
5148@dataclass
5249class FakeMembershipRepo :
5350 memberships : list [OrganizationMembership ] = field (default_factory = list )
@@ -80,7 +77,6 @@ async def list_user_memberships(
8077 ) -> list [OrganizationMembership ]:
8178 return [m for m in self .memberships if m .user_id == user_id ]
8279
83-
8480class FakeUoW :
8581 def __init__ (
8682 self ,
@@ -106,14 +102,12 @@ async def commit(self) -> None:
106102 async def rollback (self ) -> None :
107103 pass
108104
109-
110105def uow_factory (uow : FakeUoW ):
111106 def _factory () -> FakeUoW :
112107 return uow
113108
114109 return _factory
115110
116-
117111@dataclass
118112class FakeDocumentRepo :
119113 documents : dict [UUID , Document ] = field (default_factory = dict )
@@ -185,7 +179,6 @@ def count_failed_by_org(self, org_id: str) -> int:
185179 if d .org_id == oid and d .status == DocumentStatus .FAILED
186180 )
187181
188-
189182@dataclass
190183class FakeChunkRepo :
191184 chunks : list [Any ] = field (default_factory = list )
@@ -197,7 +190,6 @@ def count_by_org(self, org_id: str) -> int:
197190 oid = UUID (org_id ) if isinstance (org_id , str ) else org_id
198191 return sum (1 for c in self .chunks if c .org_id == oid )
199192
200-
201193@dataclass
202194class FakeSyncDocumentRepo :
203195 documents : dict [UUID , Document ] = field (default_factory = dict )
@@ -224,7 +216,6 @@ def update_status(
224216 if token_count is not None :
225217 doc .token_count = token_count
226218
227-
228219@dataclass
229220class FakeSyncChunkRepo :
230221 chunks : list [Any ] = field (default_factory = list )
@@ -235,7 +226,6 @@ def add_many(self, chunks: list) -> None:
235226 def delete_by_document (self , * , document_id : UUID ) -> None :
236227 self .chunks = [c for c in self .chunks if c .document_id != document_id ]
237228
238-
239229class FakeMemoryStorage :
240230 """Minimal ObjectStorage fake."""
241231
@@ -254,12 +244,10 @@ def get_object(self, *, key: str) -> bytes:
254244 def delete_object (self , * , key : str ) -> None :
255245 self ._objects .pop (key , None )
256246
257-
258247class FakeEmbedder :
259248 def embed_texts (self , texts : list [str ]) -> list [list [float ]]:
260249 return [[0.1 , 0.2 , 0.3 ] for _ in texts ]
261250
262-
263251class FakeVectorStore :
264252 def __init__ (self ) -> None :
265253 self .upserted : list [dict ] = []
@@ -273,7 +261,6 @@ async def query_by_similarity(self, **kwargs: Any) -> list[dict]:
273261 async def delete_by_document_id (self , * , document_id : str , namespace : str ) -> None :
274262 pass
275263
276-
277264def make_document (
278265 * ,
279266 org_id : UUID | None = None ,
0 commit comments