124 lines
3.5 KiB
Python
124 lines
3.5 KiB
Python
from typing import Any, Dict, List
|
|
|
|
from django.db import DEFAULT_DB_ALIAS, connections
|
|
from django.test.utils import CaptureQueriesContext
|
|
|
|
from openslides.core.config import config
|
|
from openslides.core.models import Projector
|
|
from openslides.users.models import User
|
|
from openslides.utils.projector import get_config, register_projector_element
|
|
|
|
|
|
class TConfig:
|
|
"""
|
|
Cachable, that fills the cache with the default values of the config variables.
|
|
"""
|
|
|
|
def get_collection_string(self) -> str:
|
|
return config.get_collection_string()
|
|
|
|
def get_elements(self) -> List[Dict[str, Any]]:
|
|
elements = []
|
|
config.key_to_id = {}
|
|
for id, item in enumerate(config.config_variables.values()):
|
|
elements.append(
|
|
{"id": id + 1, "key": item.name, "value": item.default_value}
|
|
)
|
|
config.key_to_id[item.name] = id + 1
|
|
return elements
|
|
|
|
async def restrict_elements(
|
|
self, user_id: int, elements: List[Dict[str, Any]]
|
|
) -> List[Dict[str, Any]]:
|
|
return elements
|
|
|
|
|
|
class TUser:
|
|
"""
|
|
Cachable, that fills the cache with fake users.
|
|
"""
|
|
|
|
def get_collection_string(self) -> str:
|
|
return User.get_collection_string()
|
|
|
|
def get_elements(self) -> List[Dict[str, Any]]:
|
|
return [
|
|
{
|
|
"id": 1,
|
|
"username": "admin",
|
|
"title": "",
|
|
"first_name": "",
|
|
"last_name": "Administrator",
|
|
"structure_level": "",
|
|
"number": "",
|
|
"about_me": "",
|
|
"groups_id": [4],
|
|
"is_present": False,
|
|
"is_committee": False,
|
|
"email": "",
|
|
"last_email_send": None,
|
|
"comment": "",
|
|
"is_active": True,
|
|
"default_password": "admin",
|
|
"session_auth_hash": "362d4f2de1463293cb3aaba7727c967c35de43ee",
|
|
}
|
|
]
|
|
|
|
async def restrict_elements(
|
|
self, user_id: int, elements: List[Dict[str, Any]]
|
|
) -> List[Dict[str, Any]]:
|
|
return elements
|
|
|
|
|
|
class TProjector:
|
|
"""
|
|
Cachable, that mocks the projector.
|
|
"""
|
|
|
|
def get_collection_string(self) -> str:
|
|
return Projector.get_collection_string()
|
|
|
|
def get_elements(self) -> List[Dict[str, Any]]:
|
|
return [
|
|
{"id": 1, "elements": [{"name": "test/slide1", "id": 1}]},
|
|
{"id": 2, "elements": [{"name": "test/slide2", "id": 1}]},
|
|
]
|
|
|
|
async def restrict_elements(
|
|
self, user_id: int, elements: List[Dict[str, Any]]
|
|
) -> List[Dict[str, Any]]:
|
|
return elements
|
|
|
|
|
|
def slide1(
|
|
config: Dict[str, Any], all_data: Dict[str, Dict[int, Dict[str, Any]]]
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
Slide that shows the general_event_name.
|
|
"""
|
|
return {"name": "slide1", "event_name": get_config(all_data, "general_event_name")}
|
|
|
|
|
|
def slide2(
|
|
config: Dict[str, Any], all_data: Dict[str, Dict[int, Dict[str, Any]]]
|
|
) -> Dict[str, Any]:
|
|
return {"name": "slide2"}
|
|
|
|
|
|
register_projector_element("test/slide1", slide1)
|
|
register_projector_element("test/slide2", slide2)
|
|
|
|
|
|
def count_queries(func, *args, **kwargs) -> int:
|
|
context = CaptureQueriesContext(connections[DEFAULT_DB_ALIAS])
|
|
with context:
|
|
func(*args, **kwargs)
|
|
|
|
queries = "\n".join(
|
|
f"{i}. {query['sql']}"
|
|
for i, query in enumerate(context.captured_queries, start=1)
|
|
)
|
|
|
|
print(f"{len(context)} queries executed\nCaptured queries were:\n{queries}")
|
|
return len(context)
|