def runTest(self):
dir = Dir(self.create_tmp_dir())
notebook = Notebook.new_from_dir(dir)
page = notebook.get_page(Path('SomePage'))
orig_store_page_1 = notebook.store_page
orig_store_page_2 = notebook.store_page_async
store_page_counter = tests.Counter()
def wrapper1(page):
store_page_counter()
orig_store_page_1(page)
def wrapper2(page, tree):
store_page_counter()
return orig_store_page_2(page, tree)
notebook.store_page = wrapper1
notebook.store_page_async = wrapper2
pageview = tests.MockObject()
pageview.readonly = False
handler = SavePageHandler(pageview, notebook, lambda: page)
# Normal operation
self.assertFalse(page.modified)
handler.try_save_page()
self.assertEqual(store_page_counter.count, 0)
self.assertFalse(page.modified)
handler.save_page_now()
self.assertEqual(store_page_counter.count, 1)
page.modified = True
handler.try_save_page()
self.assertEqual(store_page_counter.count, 2)
ongoing_operation(notebook)() # effectively a join
self.assertFalse(page.modified)
page.modified = True
handler.save_page_now()
self.assertEqual(store_page_counter.count, 3)
self.assertFalse(page.modified)
# With errors
def wrapper3(page):
raise AssertionError
def wrapper4(page, tree):
def error_cb():
raise AssertionError
return orig_store_page_2(page, error_cb)
notebook.store_page = wrapper3
notebook.store_page_async = wrapper4
page.modified = True
def catch_dialog(dialog):
assert isinstance(dialog, SavePageErrorDialog)
with tests.LoggingFilter('zim'):
with tests.DialogContext(catch_dialog):
handler.save_page_now()
self.assertTrue(page.modified)
# For autosave first error is ignore, 2nd results in dialog
self.assertFalse(handler._error_event
and handler._error_event.is_set())
with tests.LoggingFilter('zim'):
handler.try_save_page()
ongoing_operation(notebook)() # effectively a join
self.assertTrue(handler._error_event and handler._error_event.is_set())
with tests.LoggingFilter('zim'):
with tests.DialogContext(catch_dialog):
handler.try_save_page()
self.assertTrue(page.modified)
class TestDialog(tests.TestCase):
def setUp(self):
notebook = self.setUpNotebook()
pageview = setUpPageView(notebook)
def raise_error(*a):
raise AssertionError
notebook.store_page = raise_error
pageview.textview.get_buffer().set_text('Editing ...\n')
assert pageview.page.modified
self.page = pageview.page
self.pageview = pageview
self.handler = SavePageHandler(pageview, notebook,
lambda: pageview.page)
def testCancel(self):
def cancel(dialog):
self.assertIsInstance(dialog, SavePageErrorDialog)
self.assertTrue(self.page.modified)
self.assertEqual(self.page.dump('wiki'), ['Editing ...\n'])
dialog.response(Gtk.ResponseType.CANCEL)
self.assertTrue(self.page.modified)
self.assertEqual(self.page.dump('wiki'), ['Editing ...\n'])
with tests.LoggingFilter('zim'):
with tests.DialogContext(cancel):
self.handler.save_page_now()
def testDiscard(self):
def discard(dialog):
self.assertIsInstance(dialog, SavePageErrorDialog)
self.assertTrue(self.page.modified)
self.assertEqual(self.page.dump('wiki'), ['Editing ...\n'])
dialog.discard()
self.assertFalse(self.page.modified)
self.assertNotEqual(self.page.dump('wiki'), ['Editing ...\n'])
with tests.LoggingFilter('zim'):
with tests.DialogContext(discard):
self.handler.save_page_now()
def testSaveCopy(self):
folder = self.setUpFolder(mock=tests.MOCK_ALWAYS_REAL)
file = folder.file('copy.txt')
def save_copy(dialog):
self.assertIsInstance(dialog, SavePageErrorDialog)
self.assertTrue(self.page.modified)
self.assertEqual(self.page.dump('wiki'), ['Editing ...\n'])
dialog.save_copy()
self.assertFalse(self.page.modified)
self.assertNotEqual(self.page.dump('wiki'), ['Editing ...\n'])
def save_copy_dialog(dialog):
dialog.set_file(file)
dialog.do_response_ok()
with tests.LoggingFilter('zim'):
with tests.DialogContext(save_copy, save_copy_dialog):
self.handler.save_page_now()
self.assertEqual(file.read(), 'Editing ...\n')