def get_tmp_dir(name): if 'REAL_TMP' in os.environ: # Set in tests/__init__.py dir = LocalFolder(os.environ['REAL_TMP']) else: dir = LocalFolder(tempfile.gettempdir()) #~ print("TMPDIR:", dir) dir = dir.folder('test_versioncontrol').folder(name) if dir.exists(): dir.remove_children() dir.remove() assert not dir.exists() dir.touch() return dir
def setUp(self): folder = LocalFolder(XDG_DATA_HOME.subdir('zim/templates').path) assert 'tests/tmp' in folder.path.replace('\\', '/') if folder.exists(): folder.remove_children() for name, content in ( ('html/foo_test.html', 'Test 123\n'), ('html/bar_test.html', 'Test 123\n'), ): folder.file(name).write(content)
def setUp(self): folder = LocalFolder(XDG_DATA_HOME.subdir('zim/templates').path) assert 'tests/tmp' in folder.path.replace('\\', '/') if folder.exists(): folder.remove_children() for name, content in ( ('html/foo_test.html', 'Test 123\n'), ('html/bar_test.html', 'Test 123\n'), ): folder.file(name).write(content) manager = ApplicationManager() entry = manager.create('text/plain', 'test', 'test') manager.set_default_application('text/plain', entry)
def export_iter(self, pages): basename = encode_filename(pages.name) folder = LocalFolder(get_tmpdir().subdir('mhtml_export_tmp_dir').path) # XXX if folder.exists(): folder.remove_children() else: folder.touch() file = folder.file(basename + '.html') layout = SingleFileLayout(file, pages.prefix) exporter = SingleFileExporter(layout, self.template, 'html', document_root_url=self.document_root_url) for p in exporter.export_iter(pages): yield p encoder = MHTMLEncoder() linker = ExportLinker(pages.notebook, layout, output=file, usebase=True) self.file.write(encoder(layout, linker))
def testError(self): def creator_with_failure(*a): raise ThumbnailCreatorFailure def creator_with_error(*a): raise ValueError file = LocalFolder(tests.ZIM_DATADIR).file('zim.png') self.assertTrue(file.exists()) self.assertTrue(file.isimage()) for creator in creator_with_failure, creator_with_error: #~ print(">>", creator.__name__) queue = ThumbnailQueue(creator) queue.queue_thumbnail_request(file, 64) with tests.LoggingFilter('zim.plugins.attachmentbrowser', 'Exception'): queue.start() while not queue.queue_empty(): r = queue.get_ready_thumbnail() self.assertIsNone(r[0], None)