Django

Code

Show
Ignore:
Timestamp:
10/10/08 16:41:12 (1 month ago)
Author:
jacob
Message:

More be-nice-to-the-buildbot: be better about cleaning up files created by the cache/session tests.

Files:

Legend:

Unmodified
Added
Removed
Modified
Copied
Moved

  • django/trunk/tests/regressiontests/cache/tests.py

    r8278 r9224  
    1010import unittest 
    1111 
    12 from django.core.cache import cache 
     12from django.core.cache import cache, get_cache 
    1313from django.core.cache.backends.filebased import CacheClass as FileCache 
    1414from django.http import HttpResponse 
    …… 
    2424 
    2525class Cache(unittest.TestCase): 
     26    def setUp(self): 
     27        # Special-case the file cache so we can clean up after ourselves. 
     28        if isinstance(cache, FileCache): 
     29            self.cache_dir = tempfile.mkdtemp() 
     30            self.cache = get_cache("file:///%s" % self.cache_dir) 
     31        else: 
     32            self.cache_dir = None 
     33            self.cache = cache 
     34             
     35    def tearDown(self): 
     36        if self.cache_dir is not None: 
     37            shutil.rmtree(self.cache_dir) 
     38     
    2639    def test_simple(self): 
    2740        # simple set/get 
    28         cache.set("key", "value") 
    29         self.assertEqual(cache.get("key"), "value") 
     41        self.cache.set("key", "value") 
     42        self.assertEqual(self.cache.get("key"), "value") 
    3043 
    3144    def test_add(self): 
    3245        # test add (only add if key isn't already in cache) 
    33         cache.add("addkey1", "value") 
    34         result = cache.add("addkey1", "newvalue") 
     46        self.cache.add("addkey1", "value") 
     47        result = self.cache.add("addkey1", "newvalue") 
    3548        self.assertEqual(result, False) 
    36         self.assertEqual(cache.get("addkey1"), "value") 
     49        self.assertEqual(self.cache.get("addkey1"), "value") 
    3750 
    3851    def test_non_existent(self): 
    3952        # get with non-existent keys 
    40         self.assertEqual(cache.get("does_not_exist"), None) 
    41         self.assertEqual(cache.get("does_not_exist", "bang!"), "bang!") 
     53        self.assertEqual(self.cache.get("does_not_exist"), None) 
     54        self.assertEqual(self.cache.get("does_not_exist", "bang!"), "bang!") 
    4255 
    4356    def test_get_many(self): 
    4457        # get_many 
    45         cache.set('a', 'a') 
    46         cache.set('b', 'b') 
    47         cache.set('c', 'c') 
    48         cache.set('d', 'd') 
    49         self.assertEqual(cache.get_many(['a', 'c', 'd']), {'a' : 'a', 'c' : 'c', 'd' : 'd'}) 
    50         self.assertEqual(cache.get_many(['a', 'b', 'e']), {'a' : 'a', 'b' : 'b'}) 
     58        self.cache.set('a', 'a') 
     59        self.cache.set('b', 'b') 
     60        self.cache.set('c', 'c') 
     61        self.cache.set('d', 'd') 
     62        self.assertEqual(self.cache.get_many(['a', 'c', 'd']), {'a' : 'a', 'c' : 'c', 'd' : 'd'}) 
     63        self.assertEqual(self.cache.get_many(['a', 'b', 'e']), {'a' : 'a', 'b' : 'b'}) 
    5164 
    5265    def test_delete(self): 
    5366        # delete 
    54         cache.set("key1", "spam") 
    55         cache.set("key2", "eggs") 
    56         self.assertEqual(cache.get("key1"), "spam") 
    57         cache.delete("key1") 
    58         self.assertEqual(cache.get("key1"), None) 
    59         self.assertEqual(cache.get("key2"), "eggs") 
     67        self.cache.set("key1", "spam") 
     68        self.cache.set("key2", "eggs") 
     69        self.assertEqual(self.cache.get("key1"), "spam") 
     70        self.cache.delete("key1") 
     71        self.assertEqual(self.cache.get("key1"), None) 
     72        self.assertEqual(self.cache.get("key2"), "eggs") 
    6073 
    6174    def test_has_key(self): 
    6275        # has_key 
    63         cache.set("hello1", "goodbye1") 
    64         self.assertEqual(cache.has_key("hello1"), True) 
    65         self.assertEqual(cache.has_key("goodbye1"), False) 
     76        self.cache.set("hello1", "goodbye1") 
     77        self.assertEqual(self.cache.has_key("hello1"), True) 
     78        self.assertEqual(self.cache.has_key("goodbye1"), False) 
    6679 
    6780    def test_in(self): 
    68         cache.set("hello2", "goodbye2") 
    69         self.assertEqual("hello2" in cache, True) 
    70         self.assertEqual("goodbye2" in cache, False) 
     81        self.cache.set("hello2", "goodbye2") 
     82        self.assertEqual("hello2" in self.cache, True) 
     83        self.assertEqual("goodbye2" in self.cache, False) 
    7184 
    7285    def test_data_types(self): 
    …… 
    8093            'class'     : C, 
    8194        } 
    82         cache.set("stuff", stuff) 
    83         self.assertEqual(cache.get("stuff"), stuff) 
     95        self.cache.set("stuff", stuff) 
     96        self.assertEqual(self.cache.get("stuff"), stuff) 
    8497 
    8598    def test_expiration(self): 
    86         cache.set('expire1', 'very quickly', 1) 
    87         cache.set('expire2', 'very quickly', 1) 
    88         cache.set('expire3', 'very quickly', 1) 
     99        self.cache.set('expire1', 'very quickly', 1) 
     100        self.cache.set('expire2', 'very quickly', 1) 
     101        self.cache.set('expire3', 'very quickly', 1) 
    89102 
    90103        time.sleep(2) 
    91         self.assertEqual(cache.get("expire1"), None) 
     104        self.assertEqual(self.cache.get("expire1"), None) 
    92105 
    93         cache.add("expire2", "newvalue") 
    94         self.assertEqual(cache.get("expire2"), "newvalue") 
    95         self.assertEqual(cache.has_key("expire3"), False) 
     106        self.cache.add("expire2", "newvalue") 
     107        self.assertEqual(self.cache.get("expire2"), "newvalue") 
     108        self.assertEqual(self.cache.has_key("expire3"), False) 
    96109 
    97110    def test_unicode(self): 
    …… 
    103116            } 
    104117        for (key, value) in stuff.items(): 
    105             cache.set(key, value) 
    106             self.assertEqual(cache.get(key), value) 
     118            self.cache.set(key, value) 
     119            self.assertEqual(self.cache.get(key), value) 
    107120 
    108121 
    …… 
    112125    """ 
    113126    def setUp(self): 
    114         self.dirname = tempfile.mktemp() 
    115         os.mkdir(self.dirname) 
     127        self.dirname = tempfile.mkdtemp() 
    116128        self.cache = FileCache(self.dirname, {}) 
    117129