"""
Тесты для проверки изоляции фотографий между тенантами.
Убеждаемся что файлы товаров одного тенанта не видны другому.
"""
import os
import tempfile
from io import BytesIO
from pathlib import Path

from django.test import TestCase, TransactionTestCase
from django.core.files.uploadedfile import SimpleUploadedFile
from django.db import connection
from django_tenants.test.cases import TenantTestCase
from PIL import Image

from products.models.products import Product, ProductCategory
from products.models.photos import ProductPhoto


class MultiTenantPhotoIsolationTest(TenantTestCase):
    """
    Тест для проверки что фотографии товаров изолированы между тенантами.
    Гарантирует что товары с одинаковыми ID в разных тенантах имеют разные файлы.
    """

    @staticmethod
    def create_test_image(name="test.jpg", size=(800, 600)):
        """
        Создает тестовое изображение PIL.

        Args:
            name: Имя файла
            size: Размер (width, height)

        Returns:
            SimpleUploadedFile: Загруженный файл для использования в тесте
        """
        img = Image.new('RGB', size, color='red')
        img_io = BytesIO()
        img.save(img_io, format='JPEG')
        img_io.seek(0)
        return SimpleUploadedFile(name, img_io.getvalue(), content_type='image/jpeg')

    def setUp(self):
        """Подготовка для каждого теста"""
        super().setUp()

        # Создаем товар с ID=1 в ТЕКУЩЕМ тенанте
        self.product = Product.objects.create(
            name="Test Product",
            slug="test-product",
            cost_price=100.0,
            price=200.0
        )

    def test_photo_file_path_includes_tenant_id(self):
        """
        Тест: Фото сохраняется с tenant_id в пути.

        На диске файл должен быть в: tenants/{tenant_id}/products/{product_id}/{photo_id}/
        """
        # Загружаем фото
        test_image = self.create_test_image()
        photo = ProductPhoto.objects.create(
            product=self.product,
            image=test_image,
            order=0
        )

        # Получаем tenant_id из текущей схемы
        current_schema = connection.schema_name

        # Проверяем что путь в БД не содержит tenant_id (экономия)
        self.assertFalse(
            photo.image.name.startswith('tenants/'),
            f"Path in DB should not include tenant_id, got: {photo.image.name}"
        )

        # Проверяем что базовая структура правильная
        self.assertTrue(
            photo.image.name.startswith('products/'),
            f"Path should start with 'products/', got: {photo.image.name}"
        )

    def test_different_tenants_same_product_id_different_files(self):
        """
        Тест: Товары с одинаковым ID в разных тенантах имеют разные файлы.

        Ключевой тест для проверки что баг исправлен!
        """
        # Загружаем фото для товара ID=1 в текущем тенанте
        test_image_1 = self.create_test_image(name="test_1.jpg", size=(800, 600))
        photo_1 = ProductPhoto.objects.create(
            product=self.product,
            image=test_image_1,
            order=0
        )

        # После загрузки фото находится во временной папке
        # Путь в БД сохраняется БЕЗ tenant_id (для мобильности)
        path_1 = photo_1.image.name

        # Получаем текущий tenant_id (из schema_name)
        current_schema = connection.schema_name

        # Проверяем что путь в БД не содержит tenant_id
        self.assertFalse(
            path_1.startswith('tenants/'),
            f"Path in DB should not include tenant_id, got: {path_1}"
        )

        # Проверяем что путь начинается с 'products/'
        self.assertTrue(
            path_1.startswith('products/'),
            f"Photo path should start with 'products/', got: {path_1}"
        )

    def test_photo_quality_level_set(self):
        """
        Тест: После загрузки фото устанавливается уровень качества.
        """
        test_image = self.create_test_image(size=(1200, 900))
        photo = ProductPhoto.objects.create(
            product=self.product,
            image=test_image,
            order=0
        )
        # Передаем use_async через save()
        photo.save(use_async=False)

        # Проверяем что качество установлено
        self.assertIn(
            photo.quality_level,
            dict(ProductPhoto.QUALITY_LEVEL_CHOICES).keys(),
            f"Quality level should be one of the choices, got: {photo.quality_level}"
        )

    def test_storage_class_configured(self):
        """
        Тест: TenantAwareFileSystemStorage настроен в settings.
        """
        from django.conf import settings

        storage_setting = getattr(settings, 'DEFAULT_FILE_STORAGE', None)

        self.assertEqual(
            storage_setting,
            'products.utils.storage.TenantAwareFileSystemStorage',
            f"DEFAULT_FILE_STORAGE should be configured for tenant-aware storage, got: {storage_setting}"
        )


class PhotoSecurityTest(TenantTestCase):
    """
    Тесты для проверки безопасности - что TenantAwareFileSystemStorage
    предотвращает доступ к файлам других тенантов.
    """

    def test_storage_rejects_cross_tenant_access(self):
        """
        Тест: Storage отказывает в доступе к файлам других тенантов.
        """
        from products.utils.storage import TenantAwareFileSystemStorage

        storage = TenantAwareFileSystemStorage()

        # Текущий tenant_id из schema
        current_schema = connection.schema_name

        # Пытаемся получить URL файла из "другого" тенанта
        # Это должно не пройти проверку безопасности
        # Формируем путь с другим tenant_id
        other_tenant_path = "tenants/999/products/1/1/original.jpg"

        # Пытаемся получить URL - должно выбросить исключение
        with self.assertRaises(RuntimeError) as cm:
            storage.url(other_tenant_path)

        self.assertIn(
            "different tenant",
            str(cm.exception).lower(),
            "Storage should reject access to files from different tenant"
        )
