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

import unittest

from cloudinit.distros.parsers import hosts


BASE_ETC = '''
# Example
127.0.0.1	localhost
192.168.1.10	foo.mydomain.org  foo
192.168.1.10 	bar.mydomain.org  bar
146.82.138.7	master.debian.org      master
209.237.226.90	www.opensource.org
'''
BASE_ETC = BASE_ETC.strip()


class TestHostsHelper(unittest.TestCase):
    def test_parse(self):
        eh = hosts.HostsConf(BASE_ETC)
        self.assertEqual(eh.get_entry('127.0.0.1'), [['localhost']])
        self.assertEqual(eh.get_entry('192.168.1.10'),
                         [['foo.mydomain.org', 'foo'],
                          ['bar.mydomain.org', 'bar']])
        eh = str(eh)
        self.assertTrue(eh.startswith('# Example'))

    def test_add(self):
        eh = hosts.HostsConf(BASE_ETC)
        eh.add_entry('127.0.0.0', 'blah')
        self.assertEqual(eh.get_entry('127.0.0.0'), [['blah']])
        eh.add_entry('127.0.0.3', 'blah', 'blah2', 'blah3')
        self.assertEqual(eh.get_entry('127.0.0.3'),
                         [['blah', 'blah2', 'blah3']])

    def test_del(self):
        eh = hosts.HostsConf(BASE_ETC)
        eh.add_entry('127.0.0.0', 'blah')
        self.assertEqual(eh.get_entry('127.0.0.0'), [['blah']])

        eh.del_entries('127.0.0.0')
        self.assertEqual(eh.get_entry('127.0.0.0'), [])

# vi: ts=4 expandtab