OpenSlides/tests/integration/helpers.py

136 lines
3.8 KiB
Python
Raw Normal View History

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
2018-12-23 11:05:38 +01:00
from openslides.core.models import Projector
from openslides.users.models import User
2019-01-19 21:36:30 +01:00
from openslides.utils.projector import AllData, 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()):
2019-01-06 16:22:33 +01:00
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(
2019-01-06 16:22:33 +01:00
self, user_id: int, elements: List[Dict[str, Any]]
) -> List[Dict[str, Any]]:
return elements
class TUser:
"""
2018-12-23 11:05:38 +01:00
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 [
2019-01-06 16:22:33 +01:00
{
"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(
2019-01-06 16:22:33 +01:00
self, user_id: int, elements: List[Dict[str, Any]]
) -> List[Dict[str, Any]]:
return elements
2018-12-23 11:05:38 +01:00
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}]},
2018-12-23 11:05:38 +01:00
]
async def restrict_elements(
self, user_id: int, elements: List[Dict[str, Any]]
) -> List[Dict[str, Any]]:
return elements
2019-01-19 21:36:30 +01:00
def slide1(element: Dict[str, Any], all_data: AllData) -> Dict[str, Any]:
2018-12-23 11:05:38 +01:00
"""
Slide that shows the general_event_name.
"""
return {"name": "slide1", "event_name": get_config(all_data, "general_event_name")}
2019-01-19 21:36:30 +01:00
def slide2(element: Dict[str, Any], all_data: AllData) -> Dict[str, Any]:
2018-12-23 11:05:38 +01:00
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)
2019-01-12 23:01:42 +01:00
queries = "\n".join(
f"{i}. {query['sql']}"
for i, query in enumerate(context.captured_queries, start=1)
2019-01-06 16:22:33 +01:00
)
2019-01-12 23:01:42 +01:00
print(f"{len(context)} queries executed\nCaptured queries were:\n{queries}")
return len(context)
2019-01-19 21:36:30 +01:00
def all_data_config() -> AllData:
return {
TConfig().get_collection_string(): {
element["id"]: element for element in TConfig().get_elements()
}
}
def all_data_users() -> AllData:
return {
TUser().get_collection_string(): {
element["id"]: element for element in TUser().get_elements()
}
}