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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
# This file is part of cloud-init. See LICENSE file for license information.
"""Tests for cloudinit.util"""
import logging
import cloudinit.util as util
from cloudinit.tests.helpers import CiTestCase, mock
LOG = logging.getLogger(__name__)
MOUNT_INFO = [
'68 0 8:3 / / ro,relatime shared:1 - btrfs /dev/sda1 ro,attr2,inode64',
'153 68 254:0 / /home rw,relatime shared:101 - xfs /dev/sda2 rw,attr2'
]
class FakeCloud(object):
def __init__(self, hostname, fqdn):
self.hostname = hostname
self.fqdn = fqdn
self.calls = []
def get_hostname(self, fqdn=None, metadata_only=None):
myargs = {}
if fqdn is not None:
myargs['fqdn'] = fqdn
if metadata_only is not None:
myargs['metadata_only'] = metadata_only
self.calls.append(myargs)
if fqdn:
return self.fqdn
return self.hostname
class TestUtil(CiTestCase):
def test_parse_mount_info_no_opts_no_arg(self):
result = util.parse_mount_info('/home', MOUNT_INFO, LOG)
self.assertEqual(('/dev/sda2', 'xfs', '/home'), result)
def test_parse_mount_info_no_opts_arg(self):
result = util.parse_mount_info('/home', MOUNT_INFO, LOG, False)
self.assertEqual(('/dev/sda2', 'xfs', '/home'), result)
def test_parse_mount_info_with_opts(self):
result = util.parse_mount_info('/', MOUNT_INFO, LOG, True)
self.assertEqual(
('/dev/sda1', 'btrfs', '/', 'ro,relatime'),
result
)
@mock.patch('cloudinit.util.get_mount_info')
def test_mount_is_rw(self, m_mount_info):
m_mount_info.return_value = ('/dev/sda1', 'btrfs', '/', 'rw,relatime')
is_rw = util.mount_is_read_write('/')
self.assertEqual(is_rw, True)
@mock.patch('cloudinit.util.get_mount_info')
def test_mount_is_ro(self, m_mount_info):
m_mount_info.return_value = ('/dev/sda1', 'btrfs', '/', 'ro,relatime')
is_rw = util.mount_is_read_write('/')
self.assertEqual(is_rw, False)
class TestShellify(CiTestCase):
def test_input_dict_raises_type_error(self):
self.assertRaisesRegex(
TypeError, 'Input.*was.*dict.*xpected',
util.shellify, {'mykey': 'myval'})
def test_input_str_raises_type_error(self):
self.assertRaisesRegex(
TypeError, 'Input.*was.*str.*xpected', util.shellify, "foobar")
def test_value_with_int_raises_type_error(self):
self.assertRaisesRegex(
TypeError, 'shellify.*int', util.shellify, ["foo", 1])
def test_supports_strings_and_lists(self):
self.assertEqual(
'\n'.join(["#!/bin/sh", "echo hi mom", "'echo' 'hi dad'",
"'echo' 'hi' 'sis'", ""]),
util.shellify(["echo hi mom", ["echo", "hi dad"],
('echo', 'hi', 'sis')]))
class TestGetHostnameFqdn(CiTestCase):
def test_get_hostname_fqdn_from_only_cfg_fqdn(self):
"""When cfg only has the fqdn key, derive hostname and fqdn from it."""
hostname, fqdn = util.get_hostname_fqdn(
cfg={'fqdn': 'myhost.domain.com'}, cloud=None)
self.assertEqual('myhost', hostname)
self.assertEqual('myhost.domain.com', fqdn)
def test_get_hostname_fqdn_from_cfg_fqdn_and_hostname(self):
"""When cfg has both fqdn and hostname keys, return them."""
hostname, fqdn = util.get_hostname_fqdn(
cfg={'fqdn': 'myhost.domain.com', 'hostname': 'other'}, cloud=None)
self.assertEqual('other', hostname)
self.assertEqual('myhost.domain.com', fqdn)
def test_get_hostname_fqdn_from_cfg_hostname_with_domain(self):
"""When cfg has only hostname key which represents a fqdn, use that."""
hostname, fqdn = util.get_hostname_fqdn(
cfg={'hostname': 'myhost.domain.com'}, cloud=None)
self.assertEqual('myhost', hostname)
self.assertEqual('myhost.domain.com', fqdn)
def test_get_hostname_fqdn_from_cfg_hostname_without_domain(self):
"""When cfg has a hostname without a '.' query cloud.get_hostname."""
mycloud = FakeCloud('cloudhost', 'cloudhost.mycloud.com')
hostname, fqdn = util.get_hostname_fqdn(
cfg={'hostname': 'myhost'}, cloud=mycloud)
self.assertEqual('myhost', hostname)
self.assertEqual('cloudhost.mycloud.com', fqdn)
self.assertEqual(
[{'fqdn': True, 'metadata_only': False}], mycloud.calls)
def test_get_hostname_fqdn_from_without_fqdn_or_hostname(self):
"""When cfg has neither hostname nor fqdn cloud.get_hostname."""
mycloud = FakeCloud('cloudhost', 'cloudhost.mycloud.com')
hostname, fqdn = util.get_hostname_fqdn(cfg={}, cloud=mycloud)
self.assertEqual('cloudhost', hostname)
self.assertEqual('cloudhost.mycloud.com', fqdn)
self.assertEqual(
[{'fqdn': True, 'metadata_only': False},
{'metadata_only': False}], mycloud.calls)
def test_get_hostname_fqdn_from_passes_metadata_only_to_cloud(self):
"""Calls to cloud.get_hostname pass the metadata_only parameter."""
mycloud = FakeCloud('cloudhost', 'cloudhost.mycloud.com')
hostname, fqdn = util.get_hostname_fqdn(
cfg={}, cloud=mycloud, metadata_only=True)
self.assertEqual(
[{'fqdn': True, 'metadata_only': True},
{'metadata_only': True}], mycloud.calls)
# vi: ts=4 expandtab
|
