Return to ExtFileTestCase.py CVS log

Up to [Repository] / ExtFile / tests

Annotation of ExtFile/tests/ExtFileTestCase.py, revision 1.1

1.1     ! dwinter     1: 
        !             2: from Testing import ZopeTestCase
        !             3: 
        !             4: import os, sys, tempfile
        !             5: 
        !             6: from Products.ExtFile import transaction
        !             7: 
        !             8: # Repository configuration
        !             9: from Products.ExtFile import ExtFile, Config
        !            10: ExtFile.ExtFile._repository = ['reposit']
        !            11: ExtFile.REPOSITORY = Config.FLAT
        !            12: ExtFile.NORMALIZE_CASE = Config.KEEP
        !            13: ExtFile.ZODB_PATH = Config.VIRTUAL
        !            14: ExtFile.REPOSITORY_EXTENSIONS = Config.MIMETYPE_REPLACE
        !            15: ExtFile.UNDO_POLICY = Config.BACKUP_ON_DELETE
        !            16:             
        !            17: # File names of test data
        !            18: from Globals import package_home
        !            19: here = package_home(globals())
        !            20: gifImage = os.path.join(here, 'data', 'Folder_icon.gif')
        !            21: jpegImage = os.path.join(here, 'data', 'Teneriffa_small.jpg')
        !            22: tiffImage = os.path.join(here, 'data', 'Mountain_cmyk.tif')
        !            23: notImage = os.path.join(here, 'data', 'Binary.foo')
        !            24: 
        !            25: # Define some permission sets
        !            26: standard_perms = ZopeTestCase.standard_permissions
        !            27: copymove_perms = ['View management screens', 'Add ExtImages', 'Add ExtFiles', 'Delete objects']
        !            28: access_perms = ['View management screens']
        !            29: change_perms = ['Change ExtFile/ExtImage']
        !            30: 
        !            31: # Put some Zope objects into the test ZODB
        !            32: app = ZopeTestCase.app()
        !            33: factory = app.manage_addProduct['OFSP']
        !            34: factory.manage_addImage('GifImage', file=open(gifImage, 'rb'))
        !            35: factory.manage_addImage('JpegImage', file=open(jpegImage, 'rb'), content_type='image/jpeg')
        !            36: factory.manage_addImage('TiffImage', file=open(tiffImage, 'rb'), content_type='image/tiff')
        !            37: factory.manage_addFile('NotImage', file=open(notImage, 'rb'))
        !            38: transaction.commit()
        !            39: ZopeTestCase.close(app)
        !            40: 
        !            41: # Load ZCML to get events configured
        !            42: try:
        !            43:     import Products.Five
        !            44:     import OFS.subscribers
        !            45: except ImportError:
        !            46:     pass
        !            47: else:
        !            48:     from Products.Five import zcml
        !            49:     zcml.load_config('configure.zcml', Products.Five)
        !            50:             
        !            51:         
        !            52: class LocalInstanceHome:
        !            53:             
        !            54:     local_home = here #tempfile.gettempdir()
        !            55:             
        !            56:     def afterSetUp(self):
        !            57:         try:
        !            58:             import App.config
        !            59:         except ImportError:
        !            60:             # Modify builtins
        !            61:             b = getattr(__builtins__, '__dict__', __builtins__)
        !            62:             self._ih = INSTANCE_HOME
        !            63:             b['INSTANCE_HOME'] = self.local_home
        !            64:         else:
        !            65:             # Zope 2.7+
        !            66:             cfg = App.config.getConfiguration()
        !            67:             self._ih = cfg.instancehome
        !            68:             cfg.instancehome = self.local_home
        !            69:             App.config.setConfiguration(cfg)
        !            70: 
        !            71:     def afterClear(self):
        !            72:         try:
        !            73:             import App.config
        !            74:         except ImportError:
        !            75:             # Restore builtins
        !            76:             b = getattr(__builtins__, '__dict__', __builtins__)
        !            77:             if hasattr(self, '_ih'):
        !            78:                 b['INSTANCE_HOME'] = self._ih
        !            79:         else:
        !            80:             # Zope 2.7+
        !            81:             cfg = App.config.getConfiguration()
        !            82:             if hasattr(self, '_ih'):
        !            83:                 cfg.instancehome = self._ih
        !            84:             App.config.setConfiguration(cfg)
        !            85: 
        !            86: 
        !            87: class ExtFileTestCase(LocalInstanceHome, ZopeTestCase.ZopeTestCase):
        !            88:         
        !            89:     def afterSetUp(self):
        !            90:         LocalInstanceHome.afterSetUp(self)
        !            91:         self._nuke_reposit = 1
        !            92: 
        !            93:     def afterClear(self):
        !            94:         if getattr(self, '_nuke_reposit', 0):
        !            95:             # Remove repository
        !            96:             repository = os.path.join(INSTANCE_HOME, 'reposit')
        !            97:             if os.path.isdir(repository): 
        !            98:                 import shutil
        !            99:                 shutil.rmtree(repository, 1)
        !           100:             del self._nuke_reposit
        !           101:         LocalInstanceHome.afterClear(self)
        !           102:         
        !           103:     def _fsname(self, id):
        !           104:         return os.path.join(INSTANCE_HOME, 'reposit', id)
        !           105:         
        !           106:     def _exists(self, id):
        !           107:         return os.path.isfile(self._fsname(id))
        !           108:         
        !           109:     def _listdir(self):
        !           110:         return os.listdir(os.path.join(INSTANCE_HOME, 'reposit'))
        !           111:         
        !           112:     def _fsize(self, id):
        !           113:         return os.stat(id)[6]
        !           114:         
        !           115:     def addExtFile(self, id, file, content_type='', folder=None):
        !           116:         # Add an ExtFile
        !           117:         if folder is None:
        !           118:             folder = self.folder
        !           119:         id = folder.manage_addProduct['ExtFile'].manage_addExtFile(id=id, file=file, content_type=content_type)
        !           120:         self.file = folder[id]
        !           121:         return self.file
        !           122:         
        !           123:     def addExtImage(self, id, file, content_type='', folder=None):
        !           124:         # Add an ExtImage
        !           125:         if folder is None:
        !           126:             folder = self.folder
        !           127:         id = folder.manage_addProduct['ExtFile'].manage_addExtImage(id=id, file=file, content_type=content_type)
        !           128:         self.image = folder[id]
        !           129:         return self.image
        !           130: 
        !           131: 
        !           132: # FileUpload factory
        !           133: from ZPublisher.HTTPRequest import FileUpload
        !           134: 
        !           135: class DummyFieldStorage:
        !           136:     '''Quacks like a FieldStorage'''
        !           137: 
        !           138:     def __init__(self, file, filename, headers):
        !           139:         self.file = file
        !           140:         self.filename = filename
        !           141:         self.headers = headers
        !           142: 
        !           143: def makeFileUpload(file, content_type='', filename=''):
        !           144:     headers = {}
        !           145:     if type(file) == type(''):
        !           146:         file = open(file, 'rb')
        !           147:     if content_type:
        !           148:         headers['content-type'] = content_type
        !           149:     fs = DummyFieldStorage(file, filename, headers)
        !           150:     return FileUpload(fs)
        !           151: