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

    34

    def testApplyHostsError(self):

    35

        with self.assertRaises(config.HostConfigError):

    36

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

    37

    38

    def testApplyHosts(self):

    39

        expected = [

    40

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

    41

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

    42

        ]

    43

    44

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

    45

[foo]

    46

    ip = 127.0.0.1

    47

    48

[bar]

    49

    ip = 127.0.0.2

    50

"""

    51

        ))):

    52

            hostname, domain, ip = expect

    53

    54

            self.assertEquals(str(host), hostname)

    55

            self.assertEquals(host.domain, domain)

    56

            self.assertEquals(host.ip, ip)

    57

    58

if __name__ == '__main__':

    59

    unittest.main()