Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 7 additions & 10 deletions tests/unittests/distros/test_hostname.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,4 @@
# This file is part of cloud-init. See LICENSE file for license information.

import unittest

from cloudinit.distros.parsers import hostname

BASE_HOSTNAME = """
Expand All @@ -13,27 +10,27 @@
BASE_HOSTNAME = BASE_HOSTNAME.strip()


class TestHostnameHelper(unittest.TestCase):
class TestHostnameHelper:
def test_parse_same(self):
hn = hostname.HostnameConf(BASE_HOSTNAME)
self.assertEqual(str(hn).strip(), BASE_HOSTNAME)
self.assertEqual(hn.hostname, "blahblah")
assert str(hn).strip() == BASE_HOSTNAME
assert hn.hostname == "blahblah"

def test_no_adjust_hostname(self):
hn = hostname.HostnameConf(BASE_HOSTNAME)
prev_name = hn.hostname
hn.set_hostname("")
self.assertEqual(hn.hostname, prev_name)
assert hn.hostname == prev_name

def test_adjust_hostname(self):
hn = hostname.HostnameConf(BASE_HOSTNAME)
prev_name = hn.hostname
self.assertEqual(prev_name, "blahblah")
assert prev_name == "blahblah"
hn.set_hostname("bbbbd")
self.assertEqual(hn.hostname, "bbbbd")
assert hn.hostname == "bbbbd"
expected_out = """
# My super-duper-hostname

bbbbd
"""
self.assertEqual(str(hn).strip(), expected_out.strip())
assert str(hn).strip() == expected_out.strip()