def testRegisterFileFactory(self):
class FooFile(files.File):
pass
class BarFile(files.File):
pass
def TitanFileFactory(path, **unused_kwargs):
if path.startswith('/foo/files/'):
return FooFile
elif path.startswith('/bar/files/'):
return BarFile
return files.File
files.register_file_factory(TitanFileFactory)
foo_file = files.File('/foo/files/a')
bar_file = files.File('/bar/files/b')
normal_file = files.File('/c')
self.assertTrue(isinstance(foo_file, FooFile))
self.assertTrue(isinstance(bar_file, BarFile))
self.assertTrue(isinstance(normal_file, files.File))
def testRegisterFileFactory(self):
class FooFile(files.File):
pass
class BarFile(files.File):
pass
def TitanFileFactory(path, **unused_kwargs):
if path.startswith('/foo/files/'):
return FooFile
elif path.startswith('/bar/files/'):
return BarFile
return files.File
files.register_file_factory(TitanFileFactory)
foo_file = files.File('/foo/files/a')
bar_file = files.File('/bar/files/b')
normal_file = files.File('/c')
self.assertTrue(isinstance(foo_file, FooFile))
self.assertTrue(isinstance(bar_file, BarFile))
self.assertTrue(isinstance(normal_file, files.File))
def setUp(self): super(VersionsTest, self).setUp() self.vcs = versions.VersionControlService() files.register_file_factory(lambda *args, **kwargs: VersionedFile)
