def test_start_up_refreshes_workers(self):
patched_handlers = tasks.refresh_functions.copy()
patched_handlers['nodegroup_uuid'] = Mock()
self.patch(tasks, 'refresh_functions', patched_handlers)
start_up.start_up()
patched_handlers['nodegroup_uuid'].assert_called_once_with(
NodeGroup.objects.ensure_master().uuid)
def test_start_up_calls_setup_maas_avahi_service(self):
recorder = FakeMethod()
self.patch(start_up, 'setup_maas_avahi_service', recorder)
start_up.start_up()
self.assertEqual(
(1, [()]),
(recorder.call_count, recorder.extract_args()))
def test_start_up_refreshes_workers(self):
patched_handlers = tasks.refresh_functions.copy()
patched_handlers['nodegroup_uuid'] = Mock()
self.patch(tasks, 'refresh_functions', patched_handlers)
start_up.start_up()
patched_handlers['nodegroup_uuid'].assert_called_once_with(
NodeGroup.objects.ensure_master().uuid)
def test_start_up_calls_write_full_dns_config(self):
recorder = FakeMethod()
self.patch(start_up, 'write_full_dns_config', recorder)
start_up.start_up()
self.assertEqual(
(1, [()]),
(recorder.call_count, recorder.extract_args()))
def test_start_up_does_not_warn_if_boot_images_are_known(self):
# If boot images are known, there is no warning about the import
# script.
factory.make_boot_image()
recorder = self.patch(start_up, 'register_persistent_error')
start_up.start_up()
self.assertNotIn(COMPONENT.IMPORT_PXE_FILES,
[args[0][0] for args in recorder.call_args_list])
def test_start_up_does_not_warn_if_boot_images_are_known(self):
# If boot images are known, there is no warning about the import
# script.
factory.make_boot_image()
recorder = self.patch(start_up, 'register_persistent_error')
start_up.start_up()
self.assertNotIn(
COMPONENT.IMPORT_PXE_FILES,
[args[0][0] for args in recorder.call_args_list])
def test_start_up_registers_atexit_lock_cleanup(self):
filelock_mock = MagicMock()
self.patch(start_up, 'FileLock', Mock(side_effect=filelock_mock))
# Patch atexit.register to assert it's called with the right
# argument.
atexit_mock = self.patch(start_up, 'atexit')
start_up.start_up()
self.assertEqual(
[call.register(filelock_mock(start_up.LOCK_FILE_NAME).break_lock)],
atexit_mock.mock_calls)
def run(self, *args, **options):
threading = options.get('use_threading', False)
if threading:
# This is a simple backport from Django's future
# version to support threading.
class ThreadedWSGIServer(ThreadingMixIn, WSGIServer):
pass
# Monkey patch basehttp.WSGIServer.
setattr(basehttp, 'WSGIServer', ThreadedWSGIServer)
start_up()
return super(Command, self).run(*args, **options)
def test_start_up_registers_atexit_lock_cleanup(self):
filelock_mock = MagicMock()
self.patch(start_up, 'FileLock', Mock(side_effect=filelock_mock))
# Patch atexit.register to assert it's called with the right
# argument.
atexit_mock = self.patch(start_up, 'atexit')
start_up.start_up()
self.assertEqual(
[call.register(
filelock_mock(start_up.LOCK_FILE_NAME).break_lock)],
atexit_mock.mock_calls)
def run(self, *args, **options):
threading = options.get('use_threading', False)
if threading:
# This is a simple backport from Django's future
# version to support threading.
class ThreadedWSGIServer(ThreadingMixIn, WSGIServer):
pass
# Monkey patch basehttp.WSGIServer.
setattr(basehttp, 'WSGIServer', ThreadedWSGIServer)
start_up()
return super(Command, self).run(*args, **options)
def test_start_up_warns_about_missing_boot_images(self):
# If no boot images have been registered yet, that may mean that
# the import script has not been successfully run yet, or that
# the master worker is having trouble reporting its images. And
# so start_up registers a persistent warning about this.
BootImage.objects.all().delete()
discard_persistent_error(COMPONENT.IMPORT_PXE_FILES)
recorder = self.patch(start_up, 'register_persistent_error')
start_up.start_up()
self.assertIn(COMPONENT.IMPORT_PXE_FILES,
[args[0][0] for args in recorder.call_args_list])
def test_start_up_does_not_warn_if_already_warning(self):
# If there already is a warning about missing boot images, it is
# based on more precise knowledge of whether we ever heard from
# the region worker at all. It will not be replaced by a less
# knowledgeable warning.
BootImage.objects.all().delete()
register_persistent_error(COMPONENT.IMPORT_PXE_FILES,
factory.getRandomString())
recorder = self.patch(start_up, 'register_persistent_error')
start_up.start_up()
self.assertNotIn(COMPONENT.IMPORT_PXE_FILES,
[args[0][0] for args in recorder.call_args_list])
def test_start_up_warns_about_missing_boot_images(self):
# If no boot images have been registered yet, that may mean that
# the import script has not been successfully run yet, or that
# the master worker is having trouble reporting its images. And
# so start_up registers a persistent warning about this.
BootImage.objects.all().delete()
discard_persistent_error(COMPONENT.IMPORT_PXE_FILES)
recorder = self.patch(start_up, 'register_persistent_error')
start_up.start_up()
self.assertIn(
COMPONENT.IMPORT_PXE_FILES,
[args[0][0] for args in recorder.call_args_list])
def test_start_up_retries_with_wait_on_exception(self):
inner_start_up = self.patch(start_up, 'inner_start_up')
inner_start_up.side_effect = [
factory.make_exception("Boom!"),
None, # Success.
]
# We don't want to really sleep.
self.patch(start_up, "pause")
# start_up() returns without error.
start_up.start_up()
# However, it did call inner_start_up() twice; the first call resulted
# in the "Boom!" exception so it tried again.
self.expectThat(inner_start_up, MockCallsMatch(call(), call()))
# It also slept once, for 3 seconds, between those attempts.
self.expectThat(start_up.pause, MockCalledOnceWith(3.0))
def test_start_up_does_not_warn_if_already_warning(self):
# If there already is a warning about missing boot images, it is
# based on more precise knowledge of whether we ever heard from
# the region worker at all. It will not be replaced by a less
# knowledgeable warning.
BootImage.objects.all().delete()
register_persistent_error(
COMPONENT.IMPORT_PXE_FILES, factory.getRandomString())
recorder = self.patch(start_up, 'register_persistent_error')
start_up.start_up()
self.assertNotIn(
COMPONENT.IMPORT_PXE_FILES,
[args[0][0] for args in recorder.call_args_list])
def prepare(self):
"""Perform start_up of the region process."""
from maasserver.start_up import start_up
return start_up(self.master)
def test_start_up_runs_in_exclusion(self):
lock_checker = LockChecker()
self.patch(start_up, 'inner_start_up', lock_checker)
start_up.start_up()
self.assertEqual(1, lock_checker.call_count)
self.assertEqual(True, lock_checker.lock_was_held)
def test_start_up_refreshes_workers_outside_lock(self):
lock_checker = LockChecker()
self.patch(NodeGroup.objects, 'refresh_workers', lock_checker)
start_up.start_up()
self.assertEquals(False, lock_checker.lock_was_held)
# Copyright 2012 Canonical Ltd. This software is licensed under the
# GNU Affero General Public License version 3 (see the file LICENSE).
"""WSGI Application."""
from __future__ import (
absolute_import,
print_function,
unicode_literals,
)
__metaclass__ = type
__all__ = [
'application',
]
import os
import sys
import django.core.handlers.wsgi
current_path = os.path.dirname(os.path.abspath(__file__))
sys.path.append(current_path)
os.environ['DJANGO_SETTINGS_MODULE'] = 'maas.settings'
application = django.core.handlers.wsgi.WSGIHandler()
from maasserver.start_up import start_up
start_up()
def test_start_up_creates_master_nodegroup(self):
start_up.start_up()
self.assertEqual(1, NodeGroup.objects.all().count())
def test_start_up_calls_write_full_dns_config(self):
recorder = FakeMethod()
self.patch(start_up, 'write_full_dns_config', recorder)
start_up.start_up()
self.assertEqual((1, [()]),
(recorder.call_count, recorder.extract_args()))
def test_start_up_calls_setup_maas_avahi_service(self):
recorder = FakeMethod()
self.patch(start_up, 'setup_maas_avahi_service', recorder)
start_up.start_up()
self.assertEqual((1, [()]),
(recorder.call_count, recorder.extract_args()))
def test_start_up_runs_in_exclusion(self):
lock_checker = LockChecker()
self.patch(start_up, 'inner_start_up', lock_checker)
start_up.start_up()
self.assertEqual(1, lock_checker.call_count)
self.assertEqual(True, lock_checker.lock_was_held)
def _performStartUp(self):
from maasserver.start_up import start_up
start_up()
def test_start_up_refreshes_workers_outside_lock(self):
lock_checker = LockChecker()
self.patch(NodeGroup.objects, 'refresh_workers', lock_checker)
start_up.start_up()
self.assertEquals(False, lock_checker.lock_was_held)
def test_start_up_creates_master_nodegroup(self):
start_up.start_up()
self.assertEqual(1, NodeGroup.objects.all().count())
