#! /usr/bin/env python

"""Test the errno module

   Roger E. Masse

"""

import errno

from test import test_support

import unittest

std_c_errors = frozenset(['EDOM', 'ERANGE'])

class ErrnoAttributeTests(unittest.TestCase):

    def test_for_improper_attributes(self):

        # No unexpected attributes should be on the module.

        for error_code in std_c_errors:

            self.assert_(hasattr(errno, error_code),

                            "errno is missing %s" % error_code)

    def test_using_errorcode(self):

        # Every key value in errno.errorcode should be on the module.

        for value in errno.errorcode.itervalues():

            self.assert_(hasattr(errno, value), 'no %s attr in errno' % value)

class ErrorcodeTests(unittest.TestCase):

    def test_attributes_in_errorcode(self):

        for attribute in errno.__dict__.iterkeys():

            if attribute.isupper():

                self.assert_(getattr(errno, attribute) in errno.errorcode,

                             'no %s attr in errno.errorcode' % attribute)

def test_main():

    test_support.run_unittest(ErrnoAttributeTests, ErrorcodeTests)

if __name__ == '__main__':

    test_main()