test_directory.py - master - Mativ/otus-python-dz-3

Репозиторий помещен в архив 24 мая 2026. Вся функциональность доступна только для чтения.
test_directory.py
200 строк8 KB
mativ
comments
20 апр 2026, 11:03
20 апр 2026, 11:0331cf55e
"""Тесты телефонного справочника (ДЗ №3)."""
import json
import os
import unittest

from model.book import Book
from model.contact import Contact
from service.bookstorage import BookStorage
from service.filestorage import FileStorage


# 1. Базовые функции: добавление, поиск, изменение, удаление

class TestBasicOperations(unittest.TestCase):
    """Тесты базовых операций справочника."""

    def setUp(self):
        """Создаёт справочник с одним контактом перед каждым тестом."""
        self.book = Book()
        self.book.add(Contact("Иван Иванов", "+7 900 111 22 33", "работа"))

    def test_add_contact(self):
        """Добавление контакта увеличивает размер справочника и назначает ID."""
        book = Book()
        c = Contact("Иван Иванов", "+7 900 000 00 00", "тест")
        book.add(c)
        self.assertEqual(len(book), 1)
        self.assertEqual(c.id, 1)

    def test_find_by_name(self):
        """Поиск по имени возвращает контакт с совпадающим именем."""
        result = self.book.get_by_suggestion("Иван")
        self.assertIsNotNone(result)
        if result is not None:
            self.assertEqual(result.name, "Иван Иванов")

    def test_find_by_phone(self):
        """Поиск по номеру телефона возвращает совпадающий контакт."""
        result = self.book.get_by_suggestion("111 22 33")
        self.assertIsNotNone(result)

    def test_find_by_general_query(self):
        """Поиск по произвольной строке находит контакт в любом поле."""
        result = self.book.get_by_suggestion("работа")
        self.assertIsNotNone(result)

    def test_edit_contact(self):
        """Изменение имени контакта сохраняется в справочнике."""
        contact = self.book.get_by_id(1)
        contact.name = "Пётр Петров"
        self.book.update(contact)
        self.assertEqual(self.book.get_by_id(1).name, "Пётр Петров")

    def test_delete_contact(self):
        """Удаление контакта уменьшает размер справочника до нуля."""
        self.book.delete(1)
        self.assertEqual(len(self.book), 0)


# 2. Загрузка и сохранение файла

class TestFileIO(unittest.TestCase):
    """Тесты загрузки и сохранения справочника в файл."""

    TEST_FILE = "test_contacts.json"

    def setUp(self):
        """Задаёт путь к тестовому файлу перед каждым тестом."""
        self.tmppath = self.TEST_FILE

    def tearDown(self):
        """Удаляет тестовый файл после каждого теста."""
        if os.path.exists(self.tmppath):
            os.remove(self.tmppath)

    def test_save_file(self):
        """Сохранение справочника создаёт корректный JSON-файл."""
        book = Book()
        book.add(Contact("Иван", "+7 900 000 00 00", "тест"))
        BookStorage(FileStorage(self.tmppath)).save(book)
        with open(self.tmppath, "r", encoding="utf-8") as f:
            data = json.load(f)
        self.assertEqual(len(data), 1)
        self.assertEqual(data[0]["name"], "Иван")

    def test_load_file(self):
        """Загрузка файла создаёт справочник с правильными данными."""
        data = [{
            "name": "Иван",
            "phone": "+7 900 000 00 00",
            "comment": "тест"
        }]
        with open(self.tmppath, "w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False)
        book = BookStorage(FileStorage(self.tmppath)).load()
        self.assertEqual(len(book), 1)
        self.assertEqual(book.get_all()[0].name, "Иван")

    def test_save_load_roundtrip(self):
        """Сохранение и повторная загрузка сохраняют все контакты."""
        book = Book()
        book.add(Contact("Иван", "+7 900 000 00 01", "тест"))
        book.add(Contact("Пётр", "+7 900 000 00 02", "тест"))
        bs = BookStorage(FileStorage(self.tmppath))
        bs.save(book)
        loaded = bs.load()
        names = []
        for c in loaded.get_all():
            names.append(c.name)
        self.assertIn("Иван", names)
        self.assertIn("Пётр", names)


# 3. Обработка исключений и граничные условия

class TestExceptions(unittest.TestCase):
    """Тесты обработки исключений и граничных условий."""

    def setUp(self):
        """Создаёт пустой справочник перед каждым тестом."""
        self.book = Book()

    def test_add_empty_contact_raises(self):
        """Добавление контакта с пустым именем вызывает исключение."""
        with self.assertRaises(Exception):
            self.book.add(Contact("", "+7 900 000 00 00", "тест"))

    def test_find_nonexistent_contact_returns_none(self):
        """Поиск несуществующего контакта возвращает None."""
        self.book.add(Contact("Иван Иванов", "+7 900 000 00 00", "тест"))
        result = self.book.get_by_suggestion("несуществующий запрос")
        self.assertIsNone(result)

    def test_delete_wrong_id_raises(self):
        """Удаление по несуществующему ID вызывает исключение."""
        with self.assertRaises(Exception):
            self.book.delete(999)

    def test_load_nonexistent_file_raises(self):
        """Загрузка несуществующего файла вызывает исключение."""
        with self.assertRaises(Exception):
            FileStorage("/nonexistent/file.json").loadList()

    def test_invalid_phone_raises(self):
        """Телефон с недопустимыми символами вызывает исключение."""
        with self.assertRaises(Exception):
            Contact.requirePhoneValid("abc")


# 4. Параметризованные тесты

class TestParametrized(unittest.TestCase):
    """Параметризованные тесты для разных форматов имён и телефонов."""

    def test_phone_formats_valid(self):
        """Допустимые форматы телефона проходят валидацию без ошибок."""
        valid = [
            "+7 900 700 50 30",
            "8-800-555-35-35",
            "+7900",
            "1234567890",
            "123 456 789",
        ]
        for phone in valid:
            Contact.requirePhoneValid(phone)

    def test_phone_formats_invalid(self):
        """Недопустимые форматы телефона вызывают исключение."""
        invalid = [
            "",
            "abc",
            "+7(900)700-50-30",
            "телефон",
        ]
        for phone in invalid:
            with self.assertRaises(Exception):
                Contact.requirePhoneValid(phone)

    def test_name_formats_valid(self):
        """Допустимые форматы имени проходят валидацию без ошибок."""
        valid = [
            "Иван",
            "John Smith",
            "Мария-Антуанетта",
            "O'Brien",
            "А",
        ]
        for name in valid:
            Contact.requireNameValid(name)

    def test_name_formats_invalid(self):
        """Пустое имя вызывает исключение."""
        invalid = [""]
        for name in invalid:
            with self.assertRaises(Exception):
                Contact.requireNameValid(name)


if __name__ == "__main__":
    unittest.main()