Django

Code

Changeset 9224

Show
Ignore:
Timestamp:
10/10/08 16:41:12 (3 months 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/django/contrib/sessions/tests.py

    r8459 r9224  
    5555>>> DatabaseSession('1').get('cat') 
    5656 
     57# Do file session tests in an isolated directory, and kill it after we're done. 
     58>>> original_session_file_path = settings.SESSION_FILE_PATH 
     59>>> import tempfile 
     60>>> temp_session_store = settings.SESSION_FILE_PATH = tempfile.mkdtemp() 
     61 
    5762>>> file_session = FileSession() 
    5863>>> file_session.modified 
    …… 
    106111ImproperlyConfigured: The session storage path '/if/this/directory/exists/you/have/a/weird/computer' doesn't exist. Please set your SESSION_FILE_PATH setting to an existing directory in which Django can store session data. 
    107112 
     113# Clean up after the file tests 
     114>>> settings.SESSION_FILE_PATH = original_session_file_path 
     115>>> import shutil 
     116>>> shutil.rmtree(temp_session_store) 
     117 
     118# 
     119# Cache-based tests 
     120# NB: be careful to delete any sessions created; stale sessions fill up the 
     121# /tmp and eventually overwhelm it after lots of runs (think buildbots) 
     122# 
     123 
    108124>>> cache_session = CacheSession() 
    109125>>> cache_session.modified 
    …… 
    145161>>> cache_session = CacheSession(cache_session.session_key) 
    146162>>> cache_session.save() 
    147 >>> CacheSession('1').get('cat') 
     163>>> cache_session.delete(cache_session.session_key) 
    148164 
    149165>>> s = SessionBase() 

  • 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