home *** CD-ROM | disk | FTP | other *** search
/ Maximum CD 2010 November / maximum-cd-2010-11.iso / DiscContents / calibre-0.7.13.msi / file_2044 (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2010-08-06  |  2.1 KB  |  45 lines

  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.6)
  3.  
  4. import shutil
  5. import tempfile
  6. import unittest
  7.  
  8. class TestCase(unittest.TestCase):
  9.     
  10.     def setUp(self):
  11.         super(TestCase, self).setUp()
  12.         self._on_teardown = []
  13.  
  14.     
  15.     def make_temp_dir(self):
  16.         temp_dir = tempfile.mkdtemp(prefix = 'tmp-%s-' % self.__class__.__name__)
  17.         
  18.         def tear_down():
  19.             shutil.rmtree(temp_dir)
  20.  
  21.         self._on_teardown.append(tear_down)
  22.         return temp_dir
  23.  
  24.     
  25.     def monkey_patch(self, obj, name, value):
  26.         orig_value = getattr(obj, name)
  27.         setattr(obj, name, value)
  28.         
  29.         def reverse_patch():
  30.             setattr(obj, name, orig_value)
  31.  
  32.         self._on_teardown.append(reverse_patch)
  33.  
  34.     
  35.     def assert_contains(self, container, containee):
  36.         self.assertTrue(containee in container, '%r not in %r' % (containee, container))
  37.  
  38.     
  39.     def tearDown(self):
  40.         for func in reversed(self._on_teardown):
  41.             func()
  42.         
  43.  
  44.  
  45.