143 lines
6.5 KiB
Python
143 lines
6.5 KiB
Python
from unittest.mock import MagicMock, patch
|
|
|
|
from openslides.projector import api as projector_api
|
|
from openslides.utils.test import TestCase
|
|
|
|
|
|
class ApiFunctions(TestCase):
|
|
@patch('openslides.projector.api.default_slide')
|
|
def test_get_projector_content(self, mock_default_slide):
|
|
mock_slide = MagicMock()
|
|
mock_slide.return_value = 'slide content'
|
|
|
|
with patch.dict('openslides.projector.api.slide_callback',
|
|
values={'mock_slide': mock_slide}):
|
|
value = projector_api.get_projector_content({'callback': 'mock_slide'})
|
|
self.assertEqual(value, 'slide content')
|
|
|
|
projector_api.get_projector_content({'callback': 'unknown_slide'})
|
|
self.assertTrue(mock_default_slide.called)
|
|
|
|
with patch('openslides.projector.api.config',
|
|
{'projector_active_slide': {'callback': 'mock_slide'}}):
|
|
value = projector_api.get_projector_content()
|
|
self.assertEqual(value, 'slide content')
|
|
|
|
mock_slide.side_effect = projector_api.SlideError
|
|
projector_api.get_projector_content({'callback': 'mock_slide'})
|
|
self.assertTrue(mock_default_slide.called)
|
|
|
|
@patch('openslides.projector.api.render_to_string')
|
|
def test_default_slide(self, mock_render_to_string):
|
|
projector_api.default_slide()
|
|
mock_render_to_string.assert_called_with('projector/default_slide.html')
|
|
|
|
@patch('openslides.projector.api.projector_overlays')
|
|
def test_get_overlays(self, mock_projector_overlays):
|
|
mock_overlay = MagicMock()
|
|
mock_overlay.name = 'mock_overlay'
|
|
mock_projector_overlays.send.return_value = ((None, mock_overlay), )
|
|
value = projector_api.get_overlays()
|
|
self.assertEqual(value, {'mock_overlay': mock_overlay})
|
|
|
|
@patch('openslides.projector.api.projector_overlays')
|
|
def test_get_overlays_inactive(self, mock_projector_overlays):
|
|
mock_overlay = MagicMock()
|
|
mock_overlay.name = 'mock_overlay_2'
|
|
mock_overlay.is_active.return_value = False
|
|
mock_projector_overlays.send.return_value = ((None, mock_overlay), )
|
|
value = projector_api.get_overlays(only_active=True)
|
|
self.assertNotEqual(value, {'mock_overlay_2': mock_overlay})
|
|
|
|
@patch('openslides.projector.api.get_overlays')
|
|
def test_get_projector_overlays_js(self, mock_get_overlays):
|
|
overlay = MagicMock()
|
|
mock_get_overlays.return_value = {'overlay': overlay}
|
|
|
|
# Test with inactive overlay
|
|
overlay.is_active.return_value = False
|
|
value = projector_api.get_projector_overlays_js()
|
|
self.assertEqual(value, [])
|
|
|
|
# Test with active overlay without js
|
|
overlay.is_active.return_value = True
|
|
overlay.get_javascript.return_value = None
|
|
value = projector_api.get_projector_overlays_js()
|
|
self.assertEqual(value, [])
|
|
|
|
# Test with active overlay with js
|
|
overlay.get_javascript.return_value = 'some javascript'
|
|
value = projector_api.get_projector_overlays_js()
|
|
self.assertEqual(value, ['some javascript'])
|
|
|
|
def test_register_slide(self):
|
|
mock_slide_callback = {}
|
|
mock_slide_model = {}
|
|
with patch('openslides.projector.api.slide_model', mock_slide_model):
|
|
with patch('openslides.projector.api.slide_callback', mock_slide_callback):
|
|
projector_api.register_slide('some name', 'some callback')
|
|
projector_api.register_slide('other name', 'other callback', 'Model')
|
|
self.assertEqual(mock_slide_callback, {'some name': 'some callback',
|
|
'other name': 'other callback'})
|
|
self.assertEqual(mock_slide_model, {'other name': 'Model'})
|
|
|
|
@patch('openslides.projector.api.render_to_string')
|
|
@patch('openslides.projector.api.register_slide')
|
|
def test_register_slide_model(self, mock_register_slide, mock_render_to_string):
|
|
mock_SlideModel = MagicMock()
|
|
mock_SlideModel.slide_callback_name = 'mock_callback_name'
|
|
mock_SlideModel.DoesNotExist = Exception
|
|
mock_slide_object = MagicMock()
|
|
mock_slide_object.get_slide_context.return_value = 'some context'
|
|
mock_SlideModel.objects.get.return_value = mock_slide_object
|
|
|
|
projector_api.register_slide_model(mock_SlideModel, 'some template')
|
|
used_args, __ = mock_register_slide.call_args
|
|
self.assertEqual(used_args[0], 'mock_callback_name')
|
|
self.assertEqual(used_args[2], mock_SlideModel)
|
|
|
|
# Test the generated slide function
|
|
used_args[1](pk=1)
|
|
mock_render_to_string.assert_called_with('some template', 'some context')
|
|
|
|
# Test with non existing object
|
|
mock_SlideModel.objects.get.side_effect = Exception
|
|
self.assertRaises(projector_api.SlideError, used_args[1], pk=1)
|
|
|
|
def test_set_active_slide(self):
|
|
mock_config = {}
|
|
with patch('openslides.projector.api.config', mock_config):
|
|
projector_api.set_active_slide('callback_name', some='kwargs')
|
|
self.assertEqual(mock_config,
|
|
{'projector_active_slide': {'callback': 'callback_name',
|
|
'some': 'kwargs'}})
|
|
|
|
def test_get_active_slide(self):
|
|
mock_config = {'projector_active_slide': 'value'}
|
|
with patch('openslides.projector.api.config', mock_config):
|
|
value = projector_api.get_active_slide()
|
|
self.assertEqual(value, 'value')
|
|
|
|
def test_get_active_object(self):
|
|
mock_Model = MagicMock()
|
|
mock_Model.DoesNotExist = Exception
|
|
mock_slide_model = {'mock_model': mock_Model}
|
|
mock_active_slide = {'callback': 'unknown'}
|
|
mock_get_active_slide = MagicMock(return_value=mock_active_slide)
|
|
|
|
with patch('openslides.projector.api.get_active_slide', mock_get_active_slide):
|
|
with patch('openslides.projector.api.slide_model', mock_slide_model):
|
|
# test unknwon slide_callback_name
|
|
self.assertIsNone(projector_api.get_active_object())
|
|
|
|
# test unknown object
|
|
mock_Model.objects.get.side_effect = Exception
|
|
mock_active_slide.update(callback='mock_model', pk=42)
|
|
self.assertIsNone(projector_api.get_active_object())
|
|
mock_Model.objects.get.assert_called_with(pk=42)
|
|
|
|
# test success
|
|
mock_Model.objects.get.side_effect = None
|
|
mock_Model.objects.get.return_value = 'success'
|
|
self.assertEqual(projector_api.get_active_object(), 'success')
|