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 NamedStringIO(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

    21

    def testApplyHostsError(self):

    22

        with self.assertRaises(config.HostConfigError):

    23

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

    24

    25

    def testApplyHosts(self):

    26

        expected = [

    27

            ('foo', 'test', ipaddr.IPAddress('127.0.0.1')),

    28

            ('bar', 'test', ipaddr.IPAddress('127.0.0.2')),

    29

        ]

    30

    31

        for expect, host in zip(expected, config.apply_hosts_file(self.options, NamedStringIO('test', """

    32

[foo]

    33

    ip = 127.0.0.1

    34

    35

[bar]

    36

    ip = 127.0.0.2

    37

"""

    38

        ))):

    39

            hostname, domain, ip = expect

    40

    41

            self.assertEquals(str(host), hostname)

    42

            self.assertEquals(host.domain, domain)

    43

            self.assertEquals(host.ip, ip)

    44

    45

if __name__ == '__main__':

    46

    unittest.main()