Lib/test/test_dircache.py

   1 """
   2   Test cases for the dircache module
   3   Nick Mathewson
   4 """
   5
   6 import unittest
   7 from test.test_support import run_unittest, import_module
   8 dircache = import_module('dircache', deprecated=True)
   9 import os, time, sys, tempfile
  10
  11
  12 class DircacheTests(unittest.TestCase):
  13     def setUp(self):
  14         self.tempdir = tempfile.mkdtemp()
  15
  16     def tearDown(self):
  17         for fname in os.listdir(self.tempdir):
  18             self.delTemp(fname)
  19         os.rmdir(self.tempdir)
  20
  21     def writeTemp(self, fname):
  22         f = open(os.path.join(self.tempdir, fname), 'w')
  23         f.close()
  24
  25     def mkdirTemp(self, fname):
  26         os.mkdir(os.path.join(self.tempdir, fname))
  27
  28     def delTemp(self, fname):
  29         fname = os.path.join(self.tempdir, fname)
  30         if os.path.isdir(fname):
  31             os.rmdir(fname)
  32         else:
  33             os.unlink(fname)
  34
  35     def test_listdir(self):
  36         ## SUCCESSFUL CASES
  37         entries = dircache.listdir(self.tempdir)
  38         self.assertEquals(entries, [])
  39
  40         # Check that cache is actually caching, not just passing through.
  41         self.assertTrue(dircache.listdir(self.tempdir) is entries)
  42
  43         # Directories aren't "files" on Windows, and directory mtime has
  44         # nothing to do with when files under a directory get created.
  45         # That is, this test can't possibly work under Windows -- dircache
  46         # is only good for capturing a one-shot snapshot there.
  47
  48         if sys.platform[:3] not in ('win', 'os2'):
  49             # Sadly, dircache has the same granularity as stat.mtime, and so
  50             # can't notice any changes that occurred within 1 sec of the last
  51             # time it examined a directory.
  52             time.sleep(1)
  53             self.writeTemp("test1")
  54             entries = dircache.listdir(self.tempdir)
  55             self.assertEquals(entries, ['test1'])
  56             self.assertTrue(dircache.listdir(self.tempdir) is entries)
  57
  58         ## UNSUCCESSFUL CASES
  59         self.assertRaises(OSError, dircache.listdir, self.tempdir+"_nonexistent")
  60
  61     def test_annotate(self):
  62         self.writeTemp("test2")
  63         self.mkdirTemp("A")
  64         lst = ['A', 'test2', 'test_nonexistent']
  65         dircache.annotate(self.tempdir, lst)
  66         self.assertEquals(lst, ['A/', 'test2', 'test_nonexistent'])
  67
  68
  69 def test_main():
  70     try:
  71         run_unittest(DircacheTests)
  72     finally:
  73         dircache.reset()
  74
  75
  76 if __name__ == "__main__":
  77     test_main()