adding test_delete_bookmark_yes, test_delete_bookmark_no, fix caplog.records method call

This commit is contained in:
Paul 2016-08-26 11:54:57 -05:00
parent 0e0dc452c7
commit 197b6c4f31
2 changed files with 18 additions and 1 deletions

2
buku
View File

@ -744,6 +744,8 @@ class BukuDb:
except IndexError:
logger.error('Index out of bound')
return True
def delete_all_bookmarks(self):
"""Should only be called inside of delete_bookmark
Drops the bookmark table if it exists

View File

@ -6,6 +6,7 @@ import imp
import os
from tempfile import TemporaryDirectory
import unittest, pytest
from unittest import mock
from os.path import join, expanduser
import sqlite3
@ -212,6 +213,20 @@ class TestBukuDb(unittest.TestCase):
from_db = bdb.get_bookmark_by_index(index)
self.assertIsNone(from_db)
# @unittest.skip('skipping')
def test_delete_bookmark_yes(self):
bdb = BukuDb()
# checking that "y" response causes delete_bookmark to return True
with mock.patch('builtins.input', return_value='y'):
self.assertTrue(bdb.delete_bookmark(0))
# @unittest.skip('skipping')
def test_delete_bookmark_no(self):
bdb = BukuDb()
# checking that non-"y" response causes delete_bookmark to return None
with mock.patch('builtins.input', return_value='n'):
self.assertIsNone(bdb.delete_bookmark(0))
# @unittest.skip('skipping')
def test_delete_all_bookmarks(self):
bdb = BukuDb()
@ -284,7 +299,7 @@ def test_print_bookmark(capsys, caplog, setup):
bdb.print_bookmark(1)
out, err = capsys.readouterr()
for record in caplog.records:
for record in caplog.records():
assert record.levelname == "ERROR"
assert record.getMessage() == "No matching index"
assert (out, err) == ('', '')