1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# -*- coding: utf-8 -*-
# Copyright 2020, Red Hat, Inc.
# All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
from unittest import mock
from glance.api import common
from glance.api.v2 import cached_images
import glance.async_
from glance.common import wsgi_app
from glance.tests import utils as test_utils
class TestWsgiAppInit(test_utils.BaseTestCase):
@mock.patch('glance.common.config.load_paste_app')
@mock.patch('glance.async_.set_threadpool_model')
@mock.patch('glance.common.wsgi_app._get_config_files')
def test_wsgi_init_sets_thread_settings(self, mock_config_files,
mock_set_model,
mock_load):
mock_config_files.return_value = []
self.config(task_pool_threads=123, group='wsgi')
common.DEFAULT_POOL_SIZE = 1024
wsgi_app.init_app()
# Make sure we declared the system threadpool model as native
mock_set_model.assert_called_once_with('native')
# Make sure we set the default pool size
self.assertEqual(123, common.DEFAULT_POOL_SIZE)
mock_load.assert_called_once_with('glance-api')
@mock.patch('atexit.register')
@mock.patch('glance.common.config.load_paste_app')
@mock.patch('glance.async_.set_threadpool_model')
@mock.patch('glance.common.wsgi_app._get_config_files')
def test_wsgi_init_registers_exit_handler(self, mock_config_files,
mock_set_model,
mock_load, mock_exit):
mock_config_files.return_value = []
wsgi_app.init_app()
mock_exit.assert_called_once_with(wsgi_app.drain_workers)
@mock.patch('glance.common.config.load_paste_app')
@mock.patch('glance.async_.set_threadpool_model')
@mock.patch('glance.common.wsgi_app._get_config_files')
def test_uwsgi_init_registers_exit_handler(self, mock_config_files,
mock_set_model,
mock_load):
mock_config_files.return_value = []
with mock.patch.object(wsgi_app, 'uwsgi') as mock_u:
wsgi_app.init_app()
self.assertEqual(mock_u.atexit, wsgi_app.drain_workers)
@mock.patch('glance.api.v2.cached_images.WORKER')
@mock.patch('glance.async_._THREADPOOL_MODEL', new=None)
def test_drain_workers(self, mock_cache_worker):
# Initialize the thread pool model and tasks_pool, like API
# under WSGI would, and so we have a pointer to that exact
# pool object in the cache
glance.async_.set_threadpool_model('native')
model = common.get_thread_pool('tasks_pool')
with mock.patch.object(model.pool, 'shutdown') as mock_shutdown:
wsgi_app.drain_workers()
# Make sure that shutdown() was called on the tasks_pool
# ThreadPoolExecutor
mock_shutdown.assert_called_once_with()
# Make sure we terminated the cache worker, if present.
mock_cache_worker.terminate.assert_called_once_with()
@mock.patch('glance.async_._THREADPOOL_MODEL', new=None)
def test_drain_workers_no_cache(self):
glance.async_.set_threadpool_model('native')
model = common.get_thread_pool('tasks_pool')
with mock.patch.object(model.pool, 'shutdown'):
# Make sure that with no WORKER initialized, we do not fail
wsgi_app.drain_workers()
self.assertIsNone(cached_images.WORKER)
@mock.patch('glance.async_._THREADPOOL_MODEL', new=None)
@mock.patch('glance.common.config.load_paste_app')
@mock.patch('glance.common.wsgi_app._get_config_files')
@mock.patch('threading.Thread')
@mock.patch('glance.housekeeping.StagingStoreCleaner')
def test_runs_staging_cleanup(self, mock_cleaner, mock_Thread, mock_conf,
mock_load):
mock_conf.return_value = []
wsgi_app.init_app()
mock_Thread.assert_called_once_with(
target=mock_cleaner().clean_orphaned_staging_residue,
daemon=True)
mock_Thread.return_value.start.assert_called_once_with()
@mock.patch('glance.async_._THREADPOOL_MODEL', new=None)
@mock.patch('glance.common.config.load_paste_app')
@mock.patch('glance.common.wsgi_app._get_config_files')
@mock.patch('threading.Timer')
@mock.patch('glance.image_cache.prefetcher.Prefetcher')
def test_run_cache_prefetcher_middleware_disabled(
self, mock_prefetcher, mock_Timer, mock_conf, mock_load):
mock_conf.return_value = []
wsgi_app.init_app()
mock_Timer.assert_not_called()
|