/Lib/test/test_nis.py

 1from test import test_support
 2import unittest
 3import nis
 4
 5class NisTests(unittest.TestCase):
 6    def test_maps(self):
 7        try:
 8            maps = nis.maps()
 9        except nis.error, msg:
10            # NIS is probably not active, so this test isn't useful
11            if test_support.verbose:
12                print "Test Skipped:", msg
13            # Can't raise TestSkipped as regrtest only recognizes the exception
14            #   import time.
15            return
16        try:
17            # On some systems, this map is only accessible to the
18            # super user
19            maps.remove("passwd.adjunct.byname")
20        except ValueError:
21            pass
22
23        done = 0
24        for nismap in maps:
25            mapping = nis.cat(nismap)
26            for k, v in mapping.items():
27                if not k:
28                    continue
29                if nis.match(k, nismap) != v:
30                    self.fail("NIS match failed for key `%s' in map `%s'" % (k, nismap))
31                else:
32                    # just test the one key, otherwise this test could take a
33                    # very long time
34                    done = 1
35                    break
36            if done:
37                break
38
39def test_main():
40    test_support.run_unittest(NisTests)
41
42if __name__ == '__main__':
43    test_main()