196 lines
8.3 KiB
Python
196 lines
8.3 KiB
Python
# Copyright 2018 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.
|
|
|
|
import mock
|
|
import os
|
|
import tempfile
|
|
|
|
from paunch.tests import base
|
|
from paunch.utils import systemd
|
|
|
|
|
|
class TestUtilsSystemd(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestUtilsSystemd, self).setUp()
|
|
self.r = mock.MagicMock()
|
|
self.r.returncode = 0
|
|
self.r.stdout = ''
|
|
self.r.stderr = ''
|
|
|
|
@mock.patch('shutil.rmtree', autospec=True)
|
|
@mock.patch('os.path.exists', autospec=True)
|
|
@mock.patch('subprocess.run', autospec=True)
|
|
@mock.patch('os.chmod')
|
|
def test_service_create(self, mock_chmod, mock_subprocess_run,
|
|
mock_exists, mock_rmtree):
|
|
mock_subprocess_run.return_value = self.r
|
|
container = 'my_app'
|
|
service = 'tripleo_' + container
|
|
cconfig = {'depends_on': ['something'], 'restart': 'unless-stopped',
|
|
'stop_grace_period': '15'}
|
|
tempdir = tempfile.mkdtemp()
|
|
systemd.service_create(container, cconfig, tempdir)
|
|
|
|
sysd_unit_f = tempdir + service + '.service'
|
|
unit = open(sysd_unit_f, 'rt').read()
|
|
self.assertIn('Wants=something.service', unit)
|
|
self.assertIn('Restart=always', unit)
|
|
self.assertIn('ExecStop=/usr/bin/podman stop -t 15 my_app', unit)
|
|
self.assertIn('ExecStopPost=/usr/bin/podman stop -t 15 my_app', unit)
|
|
self.assertIn('PIDFile=/var/run/my_app.pid', unit)
|
|
mock_chmod.assert_has_calls([mock.call(sysd_unit_f, 420)])
|
|
|
|
mock_subprocess_run.assert_has_calls([
|
|
mock.call(['systemctl', 'daemon-reload'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'is-enabled', service],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'enable', '--now', service],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
])
|
|
|
|
mock_rmtree.assert_has_calls([
|
|
mock.call(sysd_unit_f + '.requires')
|
|
])
|
|
|
|
os.rmdir(tempdir)
|
|
|
|
@mock.patch('subprocess.run', autospec=True)
|
|
@mock.patch('os.chmod')
|
|
def test_svc_extended_create(self, mock_chmod, mock_subprocess_run):
|
|
mock_subprocess_run.return_value = self.r
|
|
container = 'my_app'
|
|
service = 'tripleo_' + container
|
|
cconfig = {'depends_on': ['something'], 'restart': 'unless-stopped',
|
|
'stop_grace_period': '15',
|
|
'systemd_exec_flags': {'RootDirectory': '/srv',
|
|
'LimitCPU': '60',
|
|
'RuntimeDirectory': 'my_app foo/bar'}
|
|
}
|
|
tempdir = tempfile.mkdtemp()
|
|
systemd.service_create(container, cconfig, tempdir)
|
|
|
|
sysd_unit_f = tempdir + service + '.service'
|
|
unit = open(sysd_unit_f, 'rt').read()
|
|
self.assertIn('RootDirectory=/srv', unit)
|
|
self.assertIn('LimitCPU=60', unit)
|
|
self.assertIn('RuntimeDirectory=my_app foo/bar', unit)
|
|
|
|
os.rmdir(tempdir)
|
|
|
|
@mock.patch('shutil.rmtree', autospec=True)
|
|
@mock.patch('os.remove', autospec=True)
|
|
@mock.patch('os.path.exists', autospec=True)
|
|
@mock.patch('os.path.isfile', autospec=True)
|
|
@mock.patch('subprocess.run', autospec=True)
|
|
def test_service_delete(self, mock_subprocess_run, mock_isfile,
|
|
mock_exists, mock_rm, mock_rmtree):
|
|
mock_subprocess_run.return_value = self.r
|
|
mock_isfile.return_value = True
|
|
container = 'my_app'
|
|
service = 'tripleo_' + container
|
|
tempdir = tempfile.mkdtemp()
|
|
service_requires_d = service + '.service.requires'
|
|
|
|
systemd.service_delete(container, tempdir)
|
|
mock_rm.assert_has_calls([
|
|
mock.call(tempdir + service + '.service'),
|
|
mock.call(tempdir + service + '_healthcheck.service'),
|
|
mock.call(tempdir + service + '_healthcheck.timer'),
|
|
])
|
|
mock_subprocess_run.assert_has_calls([
|
|
mock.call(['systemctl', 'stop', service + '.service'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'disable', service + '.service'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'stop', service + '_healthcheck.service'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'disable', service +
|
|
'_healthcheck.service'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'stop', service + '_healthcheck.timer'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'disable', service +
|
|
'_healthcheck.timer'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
])
|
|
mock_rmtree.assert_has_calls([
|
|
mock.call(os.path.join(tempdir, service_requires_d)),
|
|
])
|
|
|
|
@mock.patch('os.chmod')
|
|
def test_healthcheck_create(self, mock_chmod):
|
|
container = 'my_app'
|
|
service = 'tripleo_' + container
|
|
tempdir = tempfile.mkdtemp()
|
|
healthcheck = service + '_healthcheck.service'
|
|
sysd_unit_f = tempdir + healthcheck
|
|
|
|
systemd.healthcheck_create(container, tempdir)
|
|
unit = open(sysd_unit_f, 'rt').read()
|
|
|
|
self.assertIn('Requisite=tripleo_my_app.service', unit)
|
|
self.assertIn('ExecStart=/usr/bin/podman exec --user root my_app '
|
|
'/openstack/healthcheck', unit)
|
|
mock_chmod.assert_has_calls([mock.call(sysd_unit_f, 420)])
|
|
|
|
@mock.patch('os.chmod')
|
|
def test_healthcheck_create_command(self, mock_chmod):
|
|
container = 'my_app'
|
|
service = 'tripleo_' + container
|
|
tempdir = tempfile.mkdtemp()
|
|
healthcheck = service + '_healthcheck.service'
|
|
sysd_unit_f = tempdir + healthcheck
|
|
check = '/foo/bar baz'
|
|
|
|
systemd.healthcheck_create(container, tempdir, test=check)
|
|
unit = open(sysd_unit_f, 'rt').read()
|
|
self.assertIn('ExecStart=/usr/bin/podman exec --user root my_app '
|
|
'/foo/bar baz', unit)
|
|
|
|
@mock.patch('subprocess.run', autospec=True)
|
|
@mock.patch('os.chmod')
|
|
def test_healthcheck_timer_create(self, mock_chmod,
|
|
mock_subprocess_run):
|
|
mock_subprocess_run.return_value = self.r
|
|
container = 'my_app'
|
|
service = 'tripleo_' + container
|
|
cconfig = {'check_interval': '15'}
|
|
tempdir = tempfile.mkdtemp()
|
|
healthcheck_timer = service + '_healthcheck.timer'
|
|
sysd_unit_f = tempdir + healthcheck_timer
|
|
|
|
systemd.healthcheck_timer_create(container, cconfig, tempdir)
|
|
unit = open(sysd_unit_f, 'rt').read()
|
|
|
|
self.assertIn('PartOf=%s.service' % service, unit)
|
|
self.assertIn('OnActiveSec=120', unit)
|
|
self.assertIn('OnUnitActiveSec=15', unit)
|
|
mock_chmod.assert_has_calls([mock.call(sysd_unit_f, 420)])
|
|
mock_subprocess_run.assert_has_calls([
|
|
mock.call(['systemctl', 'is-enabled', healthcheck_timer],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'enable', '--now', healthcheck_timer],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'is-enabled', service + '.service'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'add-requires', service + '.service',
|
|
healthcheck_timer],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
mock.call(['systemctl', 'daemon-reload'],
|
|
stderr=-1, stdout=-1, universal_newlines=True),
|
|
])
|