def local_store_persists_data(pickle_dump):
note_store = get_note_store()
note_store.listNotebooks.__name__ = 'listNotebooks'
def listNotebooks(*args, **kwargs):
return []
note_store.listNotebooks.func_code = listNotebooks.func_code
local_note_store = LocalNoteStore(note_store, Mock())
local_note_store.listNotebooks()
local_note_store = LocalNoteStore(note_store, Mock())
local_note_store.listNotebooks()
assert_equal(note_store.listNotebooks.call_count, 1)
def local_store_wraps_edam_note_store(pickle_dump, marshal_dump):
note_store = get_note_store()
changed_function = Mock(return_value=1)
local_note_store = LocalNoteStore(note_store, changed_function)
local_note_store.listNotebooks()
local_note_store.listNotebooks()
assert_equal(note_store.listNotebooks.call_count, 1)
changed_function.return_value = 2
local_note_store.listNotebooks()
local_note_store.listNotebooks()
assert_equal(note_store.listNotebooks.call_count, 2)
