Sprint_4 – Adding tests for the BooksCollector class

README.md

@@ -1 +1,15 @@
-# qa_python
+# qa_python
+Тесты для BooksCollector
+
+test_initial_state - проверка первоначального состояния
+test_add_new_book_correct_add_book_successful_add - проверка добавления книги в словарь без жанра
+test_add_new_book_incorrect_add_book_unsuccessful_add - проверка добавления книги в словарь
+test_set_book_genre_correct_genre_success - проверка на успешную установку жанра из списка
+test_set_book_genre_incorrect_genre_unsuccess - проверку жанр не входящий в список не устанавливается
+test_get_book_genre - получение книг по жанрам
+test_get_books_with_specific_genre - проверка выведения списка книг с определенным жанром
+test_get_books_genre - проверка получения списка книг
+test_get_books_for_children - проверка возврата детских книг
+test_add_book_in_favorites - проверка добавления книги в избранное
+test_delete_book_from_favorites - проверка удаления книги из избранного
+test_get_list_of_favorites_books - проверка получения списка избранных книг

test_books_collector.py

@@ -0,0 +1,135 @@
+import pytest
+from main import BooksCollector
+
+
+class TestBooksCollector:
+
+    # фикстура для пустого коллектора
+    @pytest.fixture
+    def collector(self):
+        return BooksCollector()
+
+    # фикстура с несколькими книгами и жанрами
+    @pytest.fixture
+    def collector_with_books(self):
+        collector = BooksCollector()
+        # добавляем книги разных жанров
+        collector.add_new_book('Гарри Поттер')       # фантастика
+        collector.add_new_book('Дюна')               # фантастика
+        collector.add_new_book('Оно')                # ужасы
+        collector.add_new_book('Дракула')            # ужасы
+        collector.add_new_book('Малыш и Карлсон')    # детская / мультфильм
+        collector.add_new_book('Незнайка на Луне')   # детская / мультфильм
+        collector.add_new_book('Ну, погоди!')        # мультфильм
+
+        # устанавливаем жанры
+        collector.set_book_genre('Гарри Поттер', 'Фантастика')
+        collector.set_book_genre('Дюна', 'Фантастика')
+        collector.set_book_genre('Оно', 'Ужасы')
+        collector.set_book_genre('Дракула', 'Ужасы')
+        collector.set_book_genre('Малыш и Карлсон', 'Мультфильмы')
+        collector.set_book_genre('Незнайка на Луне', 'Мультфильмы')
+        collector.set_book_genre('Ну, погоди!', 'Мультфильмы')
+        return collector
+
+    # проверка добавления одной книги
+    def test_add_new_book_add_one_book(self, collector):
+        collector.add_new_book('Том Сойер')
+        # проверяем, что книга добавилась
+        assert 'Том Сойер' in collector.get_books_genre()
+        # проверяем, что жанр пустой
+        assert collector.get_book_genre('Том Сойер') == ''
+
+    # проверка разных вариантов добавления книги
+    @pytest.mark.parametrize(
+        "name, expected_count",
+        [
+            ('Том Сойер', 1),
+            ('', 0),
+            ('А'*41, 0)  # название слишком длинное
+        ],
+        ids=["valid_name", "empty_name", "too_long_name"]
+    )
+    def test_add_new_book_various_cases(self, collector, name, expected_count):
+        collector.add_new_book(name)
+        assert len(collector.get_books_genre()) == expected_count
+
+    # проверка установки жанра книги корректного
+    def test_set_book_genre_correct(self, collector):
+        collector.add_new_book('Том Сойер')
+        collector.set_book_genre('Том Сойер', 'Фантастика')
+        assert collector.get_book_genre('Том Сойер') == 'Фантастика'
+
+    # проверка установки жанра книги некорректного
+    def test_set_book_genre_wrong_genre(self, collector):
+        collector.add_new_book('Том Сойер')
+        collector.set_book_genre('Том Сойер', 'НеверныйЖанр')
+        assert collector.get_book_genre('Том Сойер') == ''
+
+    # проверка получения жанра книги
+    def test_get_book_genre_returns_right_genre(self, collector_with_books):
+        assert collector_with_books.get_book_genre('Гарри Поттер') == 'Фантастика'
+        assert collector_with_books.get_book_genre('Оно') == 'Ужасы'
+
+    # проверка получения списка книг по жанру
+    def test_get_books_with_specific_genre(self, collector_with_books):
+        books = collector_with_books.get_books_with_specific_genre('Фантастика')
+        assert 'Гарри Поттер' in books
+        assert 'Дюна' in books
+        assert len(books) == 2
+
+    # проверка получения словаря всех книг
+    def test_get_books_genre_returns_dict(self, collector_with_books):
+        books_dict = collector_with_books.get_books_genre()
+        assert 'Гарри Поттер' in books_dict
+        assert isinstance(books_dict, dict)
+
+    # проверка получения книг для детей
+    def test_get_books_for_children_excludes_rated(self, collector_with_books):
+        books_for_children = collector_with_books.get_books_for_children()
+        # книги без возрастного рейтинга должны быть
+        assert 'Малыш и Карлсон' in books_for_children
+        assert 'Незнайка на Луне' in books_for_children
+        assert 'Ну, погоди!' in books_for_children
+        # книги с возрастным рейтингом не должны быть
+        assert 'Оно' not in books_for_children
+        assert 'Дракула' not in books_for_children
+
+    # проверка добавления книги в избранное
+    def test_add_book_in_favorites_adds_book(self, collector):
+        collector.add_new_book('Том Сойер')
+        collector.add_book_in_favorites('Том Сойер')
+        assert collector.get_list_of_favorites_books() == ['Том Сойер']
+
+    # проверка, что дубликаты не добавляются в избранное
+    def test_add_book_in_favorites_no_duplicates(self, collector):
+        collector.add_new_book('Том Сойер')
+        collector.add_book_in_favorites('Том Сойер')
+        collector.add_book_in_favorites('Том Сойер')
+        assert collector.get_list_of_favorites_books() == ['Том Сойер']
+
+    # проверка удаления книги из избранного
+    def test_delete_book_from_favorites(self, collector):
+        # создаём книгу
+        collector.add_new_book('Том Сойер')
+
+        # добавляем в избранное
+        collector.add_book_in_favorites('Том Сойер')
+        # проверяем, что книга в избранном
+        assert collector.get_list_of_favorites_books() == ['Том Сойер']
+
+        # удаляем из избранного
+        collector.delete_book_from_favorites('Том Сойер')
+        # проверяем, что её больше нет
+        assert collector.get_list_of_favorites_books() == []
+
+    # проверка получения списка всех избранных книг
+    def test_get_list_of_favorites_books_returns_correct_list(self, collector):
+        collector.add_new_book('Том Сойер')
+        collector.add_new_book('Дюна')
+        collector.add_book_in_favorites('Том Сойер')
+        collector.add_book_in_favorites('Дюна')
+        favorites = collector.get_list_of_favorites_books()
+        assert favorites == ['Том Сойер', 'Дюна']
+
+

tests.py

@@ -1,24 +1,89 @@
+import pytest
 from main import BooksCollector
 
-# класс TestBooksCollector объединяет набор тестов, которыми мы покрываем наше приложение BooksCollector
-# обязательно указывать префикс Test
+
 class TestBooksCollector:
 
-    # пример теста:
-    # обязательно указывать префикс test_
-    # дальше идет название метода, который тестируем add_new_book_
-    # затем, что тестируем add_two_books - добавление двух книг
-    def test_add_new_book_add_two_books(self):
-        # создаем экземпляр (объект) класса BooksCollector
+    def test_initial_state(self):
+        collector = BooksCollector()
+        assert collector.get_books_genre() == {} and collector.get_list_of_favorites_books() == []
+
+    def test_add_new_book_correct_add_book_successful_add(self):
+        collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        assert collector.get_book_genre('Азбука') == ''
+
+    @pytest.mark.parametrize("name, expected_count", [
+        ('Азбука', 1),
+        ('', 0),
+        ('Азбука' * 10, 0)
+    ])
+    def test_add_new_book_incorrect_add_book_unsuccessful_add(self, name, expected_count):
+        collector = BooksCollector()
+        collector.add_new_book(name)
+        assert len(collector.get_books_genre()) == expected_count
+
+    def test_set_book_genre_correct_genre_success(self):
+        collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        collector.set_book_genre('Азбука', 'Ужасы')
+        assert collector.books_genre['Азбука'] == 'Ужасы'
+
+    def test_set_book_genre_incorrect_genre_unsuccess(self):
+        collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        collector.set_book_genre('Азбука', 'FFFFFF')
+        assert collector.books_genre['Азбука'] == ''
+
+    def test_get_book_genre(self):
         collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        collector.set_book_genre('Азбука', 'Ужасы')
+        assert collector.get_book_genre('Азбука') == 'Ужасы'
 
-        # добавляем две книги
-        collector.add_new_book('Гордость и предубеждение и зомби')
-        collector.add_new_book('Что делать, если ваш кот хочет вас убить')
+    def test_get_books_with_specific_genre(self):
+        collector = BooksCollector()
+        books = ['Азбука', 'Алгебра', 'Маленький принц']
+        for book in books:
+            collector.add_new_book(book)
+            collector.set_book_genre(book, 'Ужасы')
+
+        collector.add_new_book('Ну погоди')
+        collector.set_book_genre('Ну погоди', 'Мультфильмы')
+
+        assert collector.get_books_with_specific_genre('Ужасы') == books
+
+    def test_get_books_genre(self):
+        collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        assert collector.get_books_genre() == {'Азбука': ''}
+
+    def test_get_books_for_children(self):
+        collector = BooksCollector()
+        collector.add_new_book('Детская')
+        collector.add_new_book('Взрослая')
+        collector.set_book_genre('Детская', 'Фантастика')
+        collector.set_book_genre('Взрослая', 'Ужасы')
 
-        # проверяем, что добавилось именно две
-        # словарь books_rating, который нам возвращает метод get_books_rating, имеет длину 2
-        assert len(collector.get_books_rating()) == 2
+        assert collector.get_books_for_children() == ['Детская']
 
-    # напиши свои тесты ниже
-    # чтобы тесты были независимыми в каждом из них создавай отдельный экземпляр класса BooksCollector()
+    def test_add_book_in_favorites(self):
+        collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        collector.add_book_in_favorites('Азбука')
+        assert collector.get_list_of_favorites_books() == ['Азбука']
+
+    def test_delete_book_from_favorites(self):
+        collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        collector.add_book_in_favorites('Азбука')
+        collector.delete_book_from_favorites('Азбука')
+        assert collector.get_list_of_favorites_books() == []
+
+    def test_get_list_of_favorites_books(self):
+        collector = BooksCollector()
+        collector.add_new_book('Азбука')
+        collector.add_new_book('Алгебра')
+        collector.add_book_in_favorites('Азбука')
+        collector.add_book_in_favorites('Алгебра')
+        assert collector.get_list_of_favorites_books() == ['Азбука', 'Алгебра']