blob: 37a4f51f1f856e25962e5fb48ee1f631c7ac7a49 (
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
|
# This file is part of cloud-init. See LICENSE file for license information.
from cloudinit import util
from cloudinit import atomic_helper
from cloudinit.tests.helpers import CiTestCase
from . import _get_distro
class TestGentoo(CiTestCase):
def test_write_hostname(self):
distro = _get_distro("gentoo")
hostname = "myhostname"
hostfile = self.tmp_path("hostfile")
distro._write_hostname(hostname, hostfile)
self.assertEqual('hostname="myhostname"\n', util.load_file(hostfile))
def test_write_existing_hostname_with_comments(self):
distro = _get_distro("gentoo")
hostname = "myhostname"
contents = '#This is the hostname\nhostname="localhost"'
hostfile = self.tmp_path("hostfile")
atomic_helper.write_file(hostfile, contents, omode="w")
distro._write_hostname(hostname, hostfile)
self.assertEqual('#This is the hostname\nhostname="myhostname"\n',
util.load_file(hostfile))
|