218 lines
6.4 KiB
Python
218 lines
6.4 KiB
Python
# Copyright 2014 Red Hat, Inc.
|
|
#
|
|
# 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 tests.base
|
|
|
|
|
|
class TestOsSvcDaemon(tests.base.ScriptTestBase):
|
|
def setUp(self):
|
|
super(TestOsSvcDaemon, self).setUp()
|
|
self._stub_script('map-services', 'echo $1')
|
|
|
|
def test_standard_systemd(self):
|
|
self._stub_script('dib-init-system', 'echo systemd')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'bar',
|
|
'-c', 'baz', 'arg',
|
|
])
|
|
expected = """[Unit]
|
|
Description=foo Service
|
|
After=os-refresh-config.service
|
|
Requires=foo-create-dir.service
|
|
|
|
[Service]
|
|
ExecStart=/opt/stack/venvs/bar/bin/baz arg
|
|
User=bar
|
|
|
|
|
|
[Install]
|
|
WantedBy=multi-user.target
|
|
Alias=foo.service
|
|
[Unit]
|
|
Description=Create /var/run/bar
|
|
|
|
[Service]
|
|
ExecStartPre=/bin/mkdir -p /var/run/bar
|
|
ExecStartPre=/usr/local/bin/restore-selinux-file-context /var/run/bar
|
|
ExecStart=/bin/chown -R bar:bar /var/run/bar
|
|
|
|
[Install]
|
|
RequiredBy=foo.service
|
|
"""
|
|
self.assertEqual(expected, output)
|
|
|
|
def test_standard_upstart(self):
|
|
self._stub_script('dib-init-system', 'echo upstart')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'bar',
|
|
'-c', 'baz', 'a',
|
|
])
|
|
expected = """start on runlevel [2345]
|
|
stop on runlevel [016]
|
|
|
|
|
|
env OS_SVC_ENABLE_CONTROL=1
|
|
export OS_SVC_ENABLE_CONTROL
|
|
|
|
pre-start script
|
|
mkdir -p /var/run/bar
|
|
chown -R bar:bar /var/run/bar
|
|
end script
|
|
|
|
respawn
|
|
# the default post-start of 1 second sleep delays respawning enough to
|
|
# not hit the default of 10 times in 5 seconds. Make it 2 times in 5s.
|
|
respawn limit 2 5
|
|
|
|
exec start-stop-daemon --start -c bar --exec /opt/stack/venvs/bar/bin/baz -- a
|
|
|
|
post-start exec sleep 1
|
|
"""
|
|
self.assertEqual(expected, output)
|
|
|
|
def test_dir_only_systemd(self):
|
|
self._stub_script('dib-init-system', 'echo systemd')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'bar',
|
|
])
|
|
expected = """[Unit]
|
|
Description=Create /var/run/bar
|
|
|
|
[Service]
|
|
ExecStartPre=/bin/mkdir -p /var/run/bar
|
|
ExecStartPre=/usr/local/bin/restore-selinux-file-context /var/run/bar
|
|
ExecStart=/bin/chown -R bar:bar /var/run/bar
|
|
|
|
[Install]
|
|
RequiredBy=foo.service
|
|
"""
|
|
self.assertEqual(expected, output)
|
|
|
|
def test_dir_only_upstart(self):
|
|
self._stub_script('dib-init-system', 'echo upstart')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'bar',
|
|
])
|
|
expected = """start on runlevel [2345]
|
|
stop on runlevel [016]
|
|
|
|
|
|
env OS_SVC_ENABLE_CONTROL=1
|
|
export OS_SVC_ENABLE_CONTROL
|
|
|
|
pre-start script
|
|
mkdir -p /var/run/bar
|
|
chown -R bar:bar /var/run/bar
|
|
end script
|
|
"""
|
|
self.assertEqual(expected, output)
|
|
|
|
def test_install_dir_systemd(self):
|
|
self._stub_script('dib-init-system', 'echo systemd')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'foo',
|
|
'-i', '/test/dir',
|
|
'-c', 'foo', 'arg',
|
|
])
|
|
self.assertIn('ExecStart=/test/dir/bin/foo arg', output)
|
|
|
|
def test_install_dir_upstart(self):
|
|
self._stub_script('dib-init-system', 'echo upstart')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'foo',
|
|
'-i', '/test/dir',
|
|
'-c', 'foo', 'arg',
|
|
])
|
|
self.assertIn('--exec /test/dir/bin/foo -- arg', output)
|
|
|
|
def test_environment_systemd(self):
|
|
self._stub_script('dib-init-system', 'echo systemd')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'foo',
|
|
'-e', '"foo=bar"',
|
|
'-c', 'foo', 'arg',
|
|
])
|
|
self.assertIn('Environment="foo=bar"', output)
|
|
|
|
def test_environment_upstart(self):
|
|
self._stub_script('dib-init-system', 'echo upstart')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'foo',
|
|
'-e', 'foo=bar',
|
|
'-c', 'foo', 'arg',
|
|
])
|
|
self.assertIn('env foo=bar', output)
|
|
|
|
def test_post_start_upstart(self):
|
|
self._stub_script('dib-init-system', 'echo upstart')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'foo',
|
|
'-s', 'bar',
|
|
'-c', 'foo', 'arg',
|
|
])
|
|
self.assertIn('post-start bar', output)
|
|
|
|
def test_runtime_dir_systemd(self):
|
|
self._stub_script('dib-init-system', 'echo systemd')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'bar',
|
|
'-d', 'baz',
|
|
'-c', 'foo', 'arg',
|
|
])
|
|
self.assertIn('ExecStartPre=/bin/mkdir -p /var/run/baz', output)
|
|
self.assertIn('ExecStart=/bin/chown -R bar:bar /var/run/baz', output)
|
|
|
|
def test_runtime_dir_upstart(self):
|
|
self._stub_script('dib-init-system', 'echo upstart')
|
|
output = self._run_command(
|
|
['elements/os-svc-install/bin/os-svc-daemon',
|
|
'-p',
|
|
'-n', 'foo',
|
|
'-u', 'bar',
|
|
'-d', 'baz',
|
|
'-c', 'foo', 'arg',
|
|
])
|
|
self.assertIn('mkdir -p /var/run/baz', output)
|
|
self.assertIn('chown -R bar:bar /var/run/baz', output)
|