OpenSlides/tests/utils/test_main.py

138 lines
5.4 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
import os
import sys
from django.core.exceptions import ImproperlyConfigured
2013-10-30 17:48:09 +01:00
from mock import MagicMock, patch
2013-11-04 17:21:28 +01:00
from openslides.__main__ import (
add_general_arguments,
django_command_line_utility,
runserver,
start,
syncdb)
from openslides.utils.main import (
get_browser_url,
2013-10-30 17:48:09 +01:00
get_database_path_from_settings,
get_default_settings_context,
get_default_settings_path,
get_default_user_data_path,
2013-10-30 17:48:09 +01:00
get_port,
2013-11-06 17:49:41 +01:00
PortIsBlockedError,
setup_django_settings_module,
2013-10-30 17:48:09 +01:00
start_browser,
UNIX_VERSION,
WINDOWS_PORTABLE_VERSION)
from openslides.utils.test import TestCase
class TestFunctions(TestCase):
def test_get_default_user_data_path(self):
self.assertIn(os.path.join('.local', 'share'), get_default_user_data_path(UNIX_VERSION))
def test_get_default_settings_path(self):
self.assertIn(
os.path.join('.config', 'openslides', 'settings.py'), get_default_settings_path(UNIX_VERSION))
@patch('openslides.utils.main.detect_openslides_type')
def test_get_default_settings_context_portable(self, detect_mock):
detect_mock.return_value = WINDOWS_PORTABLE_VERSION
context = get_default_settings_context()
self.assertEqual(context['openslides_user_data_path'], 'get_win32_portable_path()')
def test_setup_django_settings_module(self):
setup_django_settings_module('test_dir_dhvnghfjdh456fzheg2f/test_path_bngjdhc756dzwncshdfnx.py')
self.assertEqual(os.environ['DJANGO_SETTINGS_MODULE'], 'test_path_bngjdhc756dzwncshdfnx')
self.assertEqual(sys.path[0], 'test_dir_dhvnghfjdh456fzheg2f')
def test_setup_django_settings_module_error(self):
self.assertRaisesMessage(
ImproperlyConfigured,
"'.' is not an allowed character in the settings-file",
setup_django_settings_module,
'wrong.file.py')
def test_get_browser_url(self):
self.assertEqual(get_browser_url('123.456.789.365', 6789), 'http://123.456.789.365:6789')
self.assertEqual(get_browser_url('123.456.789.365', 80), 'http://123.456.789.365')
self.assertEqual(get_browser_url('0.0.0.0', 6789), 'http://localhost:6789')
self.assertEqual(get_browser_url('0.0.0.0', 80), 'http://localhost')
2013-10-30 17:48:09 +01:00
def test_get_port(self):
class MyException(Exception):
pass
2013-11-06 17:49:41 +01:00
def block_some_ports(mock):
if not 8000 == mock_socket.socket().bind.call_args[0][0][1]:
raise MyException
# Test open port
2013-10-30 17:48:09 +01:00
self.assertEqual(get_port('localhost', 8234), 8234)
2013-11-06 17:49:41 +01:00
# Test blocked ports
2013-10-30 17:48:09 +01:00
with patch('openslides.utils.main.socket') as mock_socket:
mock_socket.error = MyException
2013-11-06 17:49:41 +01:00
mock_socket.socket().listen = MagicMock(side_effect=block_some_ports)
2013-10-30 17:48:09 +01:00
self.assertEqual(get_port('localhost', 80), 8000)
2013-11-06 17:49:41 +01:00
self.assertRaises(PortIsBlockedError, get_port, 'localhost', 81)
2013-10-30 17:48:09 +01:00
@patch('openslides.utils.main.threading.Thread')
2013-10-30 17:48:09 +01:00
@patch('openslides.utils.main.time')
@patch('openslides.utils.main.webbrowser')
def test_start_browser(self, mock_webbrowser, mock_time, mock_Thread):
2013-10-30 17:48:09 +01:00
browser_mock = MagicMock()
mock_webbrowser.get.return_value = browser_mock
start_browser('http://localhost:8234')
self.assertTrue(mock_Thread.called)
inner_function = mock_Thread.call_args[1]['target']
inner_function()
2013-10-30 17:48:09 +01:00
browser_mock.open.assert_called_with('http://localhost:8234')
def test_get_database_path_from_settings_memory(self):
self.assertEqual(get_database_path_from_settings(), ':memory:')
2013-11-04 17:21:28 +01:00
class TestOtherFunctions(TestCase):
"""
Tests functions in openslides.__main__
"""
def test_add_general_arguments_wrong_arg(self):
self.assertRaisesMessage(
TypeError,
'The argument invalid_argument is not a valid general argument.',
add_general_arguments,
None,
['invalid_argument'])
@patch('openslides.__main__.syncdb')
@patch('openslides.__main__.runserver')
def test_start(self, mock_runserver, mock_syncdb):
mock_args = MagicMock()
start(settings=None, args=mock_args)
2013-11-20 19:32:26 +01:00
self.assertTrue(mock_syncdb.called)
2013-11-04 17:21:28 +01:00
mock_runserver.assert_called_with(None, mock_args)
2013-11-06 17:49:41 +01:00
@patch('openslides.__main__.get_port')
@patch('openslides.utils.tornado_webserver.run_tornado')
2013-11-04 17:21:28 +01:00
@patch('openslides.__main__.start_browser')
2013-11-06 17:49:41 +01:00
def test_runserver(self, mock_start_browser, mock_run_tornado, mock_get_port):
mock_get_port.return_value = 8000
2013-11-04 17:21:28 +01:00
mock_args = MagicMock()
runserver(settings=None, args=mock_args)
2013-11-20 19:32:26 +01:00
self.assertTrue(mock_run_tornado.called)
2013-11-04 17:21:28 +01:00
2013-11-20 19:32:26 +01:00
@patch('openslides.__main__.os.path.exists')
2013-11-04 17:21:28 +01:00
@patch('openslides.__main__.os.makedirs')
@patch('openslides.__main__.execute_from_command_line')
2013-11-20 19:32:26 +01:00
def test_syncdb(self, mock_execute_from_command_line, mock_os, mock_exists):
mock_exists.return_value = True
2013-11-04 17:21:28 +01:00
mock_args = MagicMock()
syncdb(settings=None, args=mock_args)
2013-11-20 19:32:26 +01:00
self.assertTrue(mock_execute_from_command_line.called)
2013-11-04 17:21:28 +01:00
@patch('openslides.__main__.execute_from_command_line')
def test_django_command_line_utility(self, mock_execute_from_command_line):
mock_args = MagicMock()
django_command_line_utility(settings=None, args=mock_args)
2013-11-20 19:32:26 +01:00
self.assertTrue(mock_execute_from_command_line.called)