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