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)