|
Packit Service |
a04d08 |
# This file is part of cloud-init. See LICENSE file for license information.
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
"""cloud-init Integration Test Verify Script."""
|
|
Packit Service |
a04d08 |
from tests.cloud_tests.testcases import base
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
class TestFinalMessage(base.CloudTestCase):
|
|
Packit Service |
a04d08 |
"""Test cloud init module `cc_final_message`."""
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
subs_char = '$'
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def get_final_message_config(self):
|
|
Packit Service |
a04d08 |
"""Get config for final message."""
|
|
Packit Service |
a04d08 |
self.assertIn('final_message', self.cloud_config)
|
|
Packit Service |
a04d08 |
return self.cloud_config['final_message']
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def get_final_message(self):
|
|
Packit Service |
a04d08 |
"""Get final message from log."""
|
|
Packit Service |
a04d08 |
out = self.get_data_file('cloud-init-output.log')
|
|
Packit Service |
a04d08 |
lines = len(self.get_final_message_config().splitlines())
|
|
Packit Service |
a04d08 |
return '\n'.join(out.splitlines()[-1 * lines:])
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def test_final_message_string(self):
|
|
Packit Service |
a04d08 |
"""Ensure final handles regular strings."""
|
|
Packit Service |
a04d08 |
for actual, config in zip(
|
|
Packit Service |
a04d08 |
self.get_final_message().splitlines(),
|
|
Packit Service |
a04d08 |
self.get_final_message_config().splitlines()):
|
|
Packit Service |
a04d08 |
if self.subs_char not in config:
|
|
Packit Service |
a04d08 |
self.assertEqual(actual, config)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
def test_final_message_subs(self):
|
|
Packit Service |
a04d08 |
"""Test variable substitution in final message."""
|
|
Packit Service |
a04d08 |
# TODO: add verification of other substitutions
|
|
Packit Service |
a04d08 |
patterns = {'$datasource': self.get_datasource()}
|
|
Packit Service |
a04d08 |
for key, expected in patterns.items():
|
|
Packit Service |
a04d08 |
index = self.get_final_message_config().splitlines().index(key)
|
|
Packit Service |
a04d08 |
actual = self.get_final_message().splitlines()[index]
|
|
Packit Service |
a04d08 |
self.assertEqual(actual, expected)
|
|
Packit Service |
a04d08 |
|
|
Packit Service |
a04d08 |
# vi: ts=4 expandtab
|