def test_add_entry(): title = "avi" content = "How are you doing today?" password = "******" add_entry(content, title, password) entry = m.Note.get(m.Note.title == title) print(entry) assert entry.content == content
def test_upload_drive_invalid(self, upload_drive_function): title = "avi" content = "How are you doing today?" password = "******" sync = True add_entry(content, title, password, sync) entry = m.Note.get(m.Note.title == title) p_1 = upload_drive(entry.title, entry.content) assert not p_1
def test_view_entry(): notes.input = lambda t: 'q' crypto = Crypto.Crypto() title = "found in this world" content = "Batman is found!!!" password = "******" content_1 = crypto.encrypt(content, password) password_to_store = crypto.key_to_store(password) add_entry(content_1, title, password_to_store) entry = m.Note.get(m.Note.title == title) flag = view_entry(entry, password) assert not flag
def test_download_drive_valid(self, upload_drive_function): title = "avi" content = "How are you doing today?" password = "******" sync = True add_entry(content, title, password, sync) entry = m.Note.get(m.Note.title == title) dir1 = os.getcwd() f = open(os.path.join(os.path.join(dir1, "sync"), title + ".txt"), "w+") # pylint: disable=invalid-name f.write(content) f.close() p_1 = download_drive(entry, entry.title, entry.content, password) assert p_1