class TestLibrary(TestCase):
def setUp(self) -> None:
self.CuT = Library()
self.CuT.db.close_db()
self.patron = Mock()
self.book_title_there = 'Adventures of Elvis'
self.book_title_not_there = 'Adventures of Dino'
self.book_title_language = 'Harry Potter'
self.book_author = 'Dummy Name'
self.first_name = 'Sam'
self.last_name = 'Wheeler'
self.age = 27
self.member_id = 100001
def test_library_db_created(self):
# Assert
self.assertIsInstance(self.CuT.db, Library_DB)
def test_book_api_created(self):
# Assert
self.assertIsInstance(self.CuT.api, Books_API)
def test_book_is_ebook(self):
# Action
self.CuT.api = Mock()
self.CuT.api.get_ebooks.return_value = dummy_book_list_json
# Assert
self.assertTrue(self.CuT.is_ebook(self.book_title_there))
def test_book_is_not_ebook(self):
# Action
self.CuT.api = Mock()
self.CuT.api.get_ebooks.return_value = dummy_book_list_json
# Assert
self.assertFalse(self.CuT.is_ebook(self.book_title_not_there))
def test_gets_book_count_for_more_than_zero(self):
# Action
self.CuT.api = Mock()
self.CuT.api.get_ebooks.return_value = dummy_similar_book_list_json
# Assert
self.assertEqual(self.CuT.get_ebooks_count(self.book_title_there), 10)
def test_gets_book_count_for_zero_books(self):
# Action
self.CuT.api = Mock()
self.CuT.api.get_ebooks.return_value = []
# Assert
self.assertEqual(self.CuT.get_ebooks_count(self.book_title_not_there),
0)
def test_gets_book_count_if_no_book_is_there(self):
# Action
self.CuT.api = Mock()
self.CuT.api.get_ebooks.return_value = [
{
'title': 'Adventures of Dino',
'ebook_count': 0
},
]
# Assert
self.assertEqual(self.CuT.get_ebooks_count(self.book_title_not_there),
0)
def test_author_entered_for_book_is_true(self):
# Action
self.CuT.api = Mock()
self.CuT.api.books_by_author.return_value = dummy_author_book_list_json
# Assert
self.assertTrue(
self.CuT.is_book_by_author(self.book_author,
self.book_title_there))
def test_author_entered_for_book_is_false(self):
# Action
self.CuT.api = Mock()
self.CuT.api.books_by_author.return_value = dummy_author_book_list_json
# Assert
self.assertFalse(
self.CuT.is_book_by_author(self.book_author,
self.book_title_not_there))
def test_author_entered_for_book_has_no_book_in_library(self):
# Action
self.CuT.api = Mock()
self.CuT.api.books_by_author.return_value = []
# Assert
self.assertFalse(
self.CuT.is_book_by_author(self.book_author,
self.book_title_not_there))
def test_gets_languages_for_the_book(self):
# Action
self.CuT.api = Mock()
self.CuT.api.get_book_info.return_value = book_language_list
languages = self.CuT.get_languages_for_book(self.book_title_language)
# Assert
self.assertEqual(len(languages), len(book_language_list))
def test_gets_one_language_for_the_book(self):
language_list = [{'language': ['eng']}]
# Action
self.CuT.api = Mock()
self.CuT.api.get_book_info.return_value = language_list
languages = self.CuT.get_languages_for_book(self.book_title_language)
# Assert
self.assertEqual(len(languages), 1)
def test_gets_zero_language_for_no_book(self):
# Assume
language_list = []
# Action
self.CuT.api = Mock()
self.CuT.api.get_book_info.return_value = language_list
languages = self.CuT.get_languages_for_book(self.book_title_language)
# Assert
self.assertEqual(len(languages), 0)
def test_gets_the_correct_languages_for_the_book(self):
# Action
self.CuT.api = Mock()
self.CuT.api.get_book_info.return_value = book_language_list
languages = self.CuT.get_languages_for_book(self.book_title_language)
all_languages = list()
for ebook in book_language_list:
all_languages.append(ebook['language'][0])
is_same = True
for lang in languages:
if lang not in all_languages:
is_same = False
break
# Assert
self.assertTrue(is_same)
def test_patron_is_registered_in_db(self):
# Assume
self.CuT.db = Mock()
self.CuT.db.insert_patron.return_value = 1
self.CuT.db.retrieve_patron.return_value = True
patron = Patron(self.first_name, self.last_name, self.age,
self.member_id)
patron.get_memberID = Mock(return_value=1)
# Action
self.CuT.register_patron(self.first_name, self.last_name, self.age,
self.member_id)
is_patron_registered = self.CuT.is_patron_registered(patron)
# Removes the instance of the db.json created
if path.exists('db.json'):
os.remove('db.json')
# Assert
self.assertTrue(is_patron_registered)
def test_patron_is_not_registered_in_db(self):
# Assume
self.CuT.db = Mock()
self.CuT.db.insert_patron.return_value = 1
self.CuT.db.retrieve_patron.return_value = False
patron = Patron(self.first_name, self.last_name, self.age,
self.member_id)
patron.get_memberID = Mock(return_value=1)
# Action
self.CuT.register_patron(self.first_name, self.last_name, self.age,
self.member_id)
is_patron_registered = self.CuT.is_patron_registered(patron)
# Removes the instance of the db.json created
if path.exists('db.json'):
os.remove('db.json')
# Assert
self.assertFalse(is_patron_registered)
def test_has_patron_borrowed_book_true(self):
# Assume
self.CuT.db = Mock()
self.patron.fname.return_value = self.first_name
self.patron.lname.return_value = self.last_name
self.patron.age.return_value = self.age
self.patron.memberID.return_value = self.member_id
self.patron.borrowed_books.return_value = []
self.CuT.db.update_patron.return_value = True
self.patron.add_borrowed_book.return_value = True
self.patron.get_borrowed_books.return_value = [
self.book_title_there.lower()
]
# Action
self.CuT.borrow_book(self.book_title_there, self.patron)
is_book_borrowed = self.CuT.is_book_borrowed(self.book_title_there,
self.patron)
# Removes the instance of the db.json created
if path.exists('db.json'):
os.remove('db.json')
# Assert
self.assertTrue(is_book_borrowed)
def test_has_patron_borrowed_book_false(self):
# Assume
self.CuT.db = Mock()
self.patron.fname.return_value = self.first_name
self.patron.lname.return_value = self.last_name
self.patron.age.return_value = self.age
self.patron.memberID.return_value = self.member_id
self.patron.borrowed_books.return_value = []
self.CuT.db.update_patron.return_value = True
self.patron.add_borrowed_book.return_value = True
self.patron.get_borrowed_books.return_value = ['Some other book title']
# Action
self.CuT.borrow_book(self.book_title_there, self.patron)
is_book_borrowed = self.CuT.is_book_borrowed(self.book_title_there,
self.patron)
# Removes the instance of the db.json created
if path.exists('db.json'):
os.remove('db.json')
# Assert
self.assertFalse(is_book_borrowed)
def test_patron_has_returned_book(self):
# Assume
self.CuT.db = Mock()
self.patron.fname.return_value = self.first_name
self.patron.lname.return_value = self.last_name
self.patron.age.return_value = self.age
self.patron.memberID.return_value = self.member_id
self.patron.borrowed_books.return_value = [
self.book_title_there.lower()
]
self.CuT.db.update_patron.return_value = True
self.patron.return_borrowed_book.return_value = True
self.patron.get_borrowed_books.return_value = []
# Action
self.CuT.borrow_book(self.book_title_there, self.patron)
self.CuT.return_borrowed_book(self.book_title_there, self.patron)
is_book_borrowed = self.CuT.is_book_borrowed(self.book_title_there,
self.patron)
# Removes the instance of the db.json created
if path.exists('db.json'):
os.remove('db.json')
# Assert
self.assertFalse(is_book_borrowed)
class TestLibrary(unittest.TestCase):
@patch('library.ext_api_interface.Books_API')
@patch('library.library_db_interface.Library_DB')
def setUp(self, mockBooks_API, mockLibrary_DB):
self.library = Library()
self.patron = Patron("bob", "bobson", 41, 1)
self.library.api = mockBooks_API
self.library.db = mockLibrary_DB
mockBooks_API.get_ebooks.return_value = [{
'title': 'The Count of Monte Cristo',
"ebook_count": 1
}, {
'title': 'A Tale of Two Cities',
"ebook_count": 1
}, {
'title': '1984',
"ebook_count": 1
}, {
'title': 'Slaughterhouse 5',
"ebook_count": 1
}, {
'title': 'Breakfast of Champions',
"ebook_count": 1
}]
mockBooks_API.books_by_author.return_value = [
'Slaughterhouse 5', 'Breakfast of Champions'
]
mockBooks_API.get_book_info.return_value = [{
'title': 'The Count of Monte Cristo',
"ebook_count": 1,
'publisher': 'Penguin',
'publish_year': '1844',
'language': "french"
}]
mockLibrary_DB.patrons = [self.patron]
mockLibrary_DB.update_patron = self.mock_update_patron
mockLibrary_DB.insert_patron.return_value = 1
mockLibrary_DB.retrieve_patron = self.mock_retrieve_patron
def mock_update_patron(self, patron):
self.library.db.patrons[0] = patron
def mock_retrieve_patron(self, patron):
if patron == 1:
return self.library.db.patrons[0]
else:
return None
def test_is_ebook(self):
library = self.library
self.assertFalse(library.is_ebook('title'))
self.assertTrue(library.is_ebook("1984"))
def test_get_ebooks_count(self):
library = self.library
self.assertEqual(library.get_ebooks_count('title'), 5)
def test_is_book_by_author(self):
library = self.library
self.assertTrue(
library.is_book_by_author("Kurt Vonnegut", 'Slaughterhouse 5'))
self.assertFalse(library.is_book_by_author("Kurt Vonnegut", "1984"))
def test_get_languages_for_book(self):
library = self.library
result = set('french')
self.assertEqual(
library.get_languages_for_book('The Count of Monte Cristo'),
result)
def test_register_patron(self):
library = self.library
self.assertEqual(library.register_patron("bob", "bobson", 41, 1), 1)
def test_is_patron_registered(self):
library = self.library
patron = self.patron
p2 = Patron("bob", "bobson", 41, 2)
self.assertTrue(library.is_patron_registered(patron))
self.assertFalse(library.is_patron_registered(p2))
def test_borrow_book(self):
library = self.library
patron = self.patron
library.borrow_book("1984", patron)
self.assertEqual(
self.library.db.retrieve_patron(1).get_borrowed_books(), ["1984"])
def test_return_borrowed_book(self):
library = self.library
patron = self.patron
library.borrow_book("1984", patron)
library.return_borrowed_book("1984", patron)
self.assertEqual(
self.library.db.retrieve_patron(1).get_borrowed_books(), [])
def test_is_book_borrowed(self):
library = self.library
patron = self.patron
library.borrow_book("1984", patron)
self.assertTrue(self.library.is_book_borrowed("1984", patron))