summaryrefslogtreecommitdiff
path: root/tests/unittests/distros/test_gentoo.py
blob: dadf5df5ea82e3c1d412dde53c49765fb514874c (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
# This file is part of cloud-init. See LICENSE file for license information.

from cloudinit import atomic_helper, util
from tests.unittests.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),
        )