1

import ipaddr

     2

import unittest

     3

     4

from pvl.hosts import config

     5

from StringIO import StringIO

     6

     7

class Options(object):

     8

    hosts_charset   = 'utf-8'

     9

    hosts_domain    = None

    10

    hosts_include   = None

    11

    12

class ConfFile(StringIO):

    13

    def __init__(self, name, buffer):

    14

        StringIO.__init__(self, buffer)

    15

        self.name = name

    16

    17

class TestConfig(unittest.TestCase):

    18

    def setUp(self):

    19

        self.options = Options()

    20

    27

    def assertHostsEqual(self, hosts, expected):

    28

        for host, expect in zip(hosts, expected):

    29

            host_str, attrs = expect

    30

    32

    33

    def testApplyHostsFileError(self):

    34

        with self.assertRaises(config.HostConfigError):

    35

            list(config.apply_hosts(self.options, ['nonexistant']))

    36

    37

    def testApplyHosts(self):

    38

        conf_file = ConfFile('test', """

    39

[foo]

    40

    ip = 127.0.0.1

    41

    42

[bar]

    43

    ip = 127.0.0.2

    44

        """)

    45

        expected = [

    46

                ('foo@test', dict(ip=ipaddr.IPAddress('127.0.0.1'))),

    47

                ('bar@test', dict(ip=ipaddr.IPAddress('127.0.0.2'))),

    48

        ]

    49

    50

        self.assertHostsEqual(config.apply_hosts_file(self.options, conf_file), expected)

    51

    52

    def testApply(self):

    53

        self.assertHostsEqual(config.apply(self.options, ['etc/hosts/test']), [

    54

                ('foo@test', dict(

    55

                    ip          = ipaddr.IPAddress('127.0.0.1'),

    56

                    ethernet    = {None: '00:11:22:33:44:55'},

    57

                )),

    58

                ('bar@test', dict(

    59

                    ip          = ipaddr.IPAddress('127.0.0.2'),

    60

                    ethernet    = {None: '01:23:45:67:89:ab'},

    61

                )),

    62

        ])

    63

    64

    def testApplyHostsExpand(self):

    65

        self.assertHostsEqual(config.apply_host_config(self.options, 'asdf', 'asdf{1-3}', ip='10.100.100.$'), [

    66

                ('asdf1@asdf', dict(ip=ipaddr.IPAddress('10.100.100.1'))),

    67

                ('asdf2@asdf', dict(ip=ipaddr.IPAddress('10.100.100.2'))),

    68

                ('asdf3@asdf', dict(ip=ipaddr.IPAddress('10.100.100.3'))),

    69

        ])

    80

    def testApplyHostsConfigError(self):

    81

        with self.assertRaises(config.HostConfigError):

    82

            config.apply_host(self.options, 'foo', 'test', {

    83

                'ethernet': 'foo',

    84

                'ethernet.eth0': 'bar',

    85

            })

    86

    87

if __name__ == '__main__':

    88

    unittest.main()