forked from sass/tipibot
Add startup scheme
This commit is contained in:
@@ -73,6 +73,11 @@ class FakePocketBase:
|
||||
async def count_records(self) -> int:
|
||||
return len(self.records)
|
||||
|
||||
async def get_collection_fields(self) -> set[str]:
|
||||
if self.schema_fields is None:
|
||||
return set(economy._default_user()) | {"user_id"}
|
||||
return set(self.schema_fields)
|
||||
|
||||
# -- test helpers -------------------------------------------------------
|
||||
def record_for(self, user_id: int) -> dict:
|
||||
for record in self.records.values():
|
||||
@@ -83,7 +88,7 @@ class FakePocketBase:
|
||||
|
||||
def _install(monkeypatch, fake: FakePocketBase) -> FakePocketBase:
|
||||
for name in ("get_record", "create_record", "update_record",
|
||||
"list_all_records", "count_records"):
|
||||
"list_all_records", "count_records", "get_collection_fields"):
|
||||
monkeypatch.setattr(pb_client, name, getattr(fake, name))
|
||||
monkeypatch.setattr(economy, "HOUSE_ID", None)
|
||||
monkeypatch.setattr(economy, "_house_pb_id", None)
|
||||
|
||||
Reference in New Issue
Block a user