import os
import django
import unittest
from antifroud.check_fraud import run_reservation_check

# Установка переменной окружения для Django settings
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "your_project.settings")
django.setup()

class FraudCheckTests(unittest.TestCase):
    def test_run_reservation_check(self):
        """Тест проверки бронирования на мошенничество"""
        result = run_reservation_check()
        self.assertIsNotNone(result)  # Проверяем, что результат не None
        self.assertIsInstance(result, dict)  # Проверяем, что возвращается словарь

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