From 84073bfdb0f202198b0fc2e3a537185a50d2e9c6 Mon Sep 17 00:00:00 2001
From: Fred Drake <fdrake@acm.org>
Date: Thu, 19 Jul 2001 22:59:09 +0000
Subject: [PATCH] Regression test for the dircache module, contributed by Nick
 Mathewson. Modified by Fred Drake. This closes SF patch #440827.

---
 Lib/test/test_dircache.py | 62 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 Lib/test/test_dircache.py

diff --git a/Lib/test/test_dircache.py b/Lib/test/test_dircache.py
new file mode 100644
index 00000000000..d6e3c24f595
--- /dev/null
+++ b/Lib/test/test_dircache.py
@@ -0,0 +1,62 @@
+"""
+  Test cases for the dircache module
+  Nick Mathewson
+"""
+
+import unittest
+from test_support import run_unittest, TESTFN
+import dircache, os, time
+
+
+class DircacheTests(unittest.TestCase):
+    def setUp(self):
+        self.tempdir = TESTFN+"_dir"
+        os.mkdir(self.tempdir)
+
+    def tearDown(self):
+        for fname in os.listdir(self.tempdir):
+            self.delTemp(fname)
+        os.rmdir(self.tempdir)
+
+    def writeTemp(self, fname):
+        f = open(os.path.join(self.tempdir, fname), 'w')
+        f.close()
+
+    def mkdirTemp(self, fname):
+        os.mkdir(os.path.join(self.tempdir, fname))
+
+    def delTemp(self, fname):
+        fname = os.path.join(self.tempdir, fname)
+        if os.path.isdir(fname):
+            os.rmdir(fname)
+        else:
+            os.unlink(fname)
+    
+    def test_listdir(self):
+        ## SUCCESSFUL CASES
+        entries = dircache.listdir(self.tempdir)
+        self.assertEquals(entries, [])
+
+        # Check that cache is actually caching, not just passing through.
+        self.assert_(dircache.listdir(self.tempdir) is entries) 
+        
+        # Sadly, dircache has the same granularity as stat.mtime, and so
+        # can't notice any changes that occured within 1 sec of the last
+        # time it examined a directory.
+        time.sleep(1)
+        self.writeTemp("test1")
+        entries = dircache.listdir(self.tempdir)
+        self.assertEquals(entries, ['test1'])
+        self.assert_(dircache.listdir(self.tempdir) is entries)
+        
+        ## UNSUCCESSFUL CASES
+        self.assertEquals(dircache.listdir(self.tempdir+"_nonexistent"), [])
+
+    def test_annotate(self):
+        self.writeTemp("test2")
+        self.mkdirTemp("A")
+        lst = ['A', 'test2', 'test_nonexistent']
+        dircache.annotate(self.tempdir, lst)
+        self.assertEquals(lst, ['A/', 'test2', 'test_nonexistent'])
+
+run_unittest(DircacheTests)