def __init__(self): self.book_category_database = bcb.BookCategoryDataBase( 'book_category_db.json') self.user_database = ub.UserDataBase('users_db.json') self.author_database = ab.AuthorDataBase('author_db.json') self.book_database = bb.BookDataBase('book_db.json') self.history = self.load_history('history.json')
def test_add_author(self): prepare_db_file() db = author_database.AuthorDataBase('adb_test.json') self.assertFalse(db.is_author_in_db("7")) aut = author.Author("7", "Test", "User") self.assertTrue(db.add_author(aut)) self.assertTrue(db.is_author_in_db("7")) remove_db_file()
def test_author_database_init(self): prepare_db_file() db = author_database.AuthorDataBase('adb_test.json') dic = ret_test_dic() for i in dic: self.assertEqual(db.author_dict[i].uid, i) self.assertEqual(db.author_dict[i].aut_name, dic[i]['name']) self.assertEqual(db.author_dict[i].aut_surname, dic[i]['surname']) remove_db_file()
def test_return_author_list(self): prepare_db_file() db = author_database.AuthorDataBase('adb_test.json') aut_list = db.return_list() temp_list = [ db.author_dict["0"].aut_name + " " + db.author_dict["0"].aut_surname, db.author_dict["1"].aut_name + " " + db.author_dict["1"].aut_surname ] self.assertEqual(aut_list, temp_list) remove_db_file()
def test_return_max_id(self): prepare_db_file() db = author_database.AuthorDataBase('adb_test.json') self.assertEqual("1", db.get_max_id()) remove_db_file()
def test_is_author_in_db(self): prepare_db_file() db = author_database.AuthorDataBase('adb_test.json') self.assertTrue(db.is_author_in_db("0")) self.assertFalse(db.is_author_in_db("7")) remove_db_file()
def test_author_database_init_no_db_file(self): remove_db_file() db = author_database.AuthorDataBase('adb_test.json') self.assertEqual(db.author_dict, dict())