19 lines
750 B
Python
19 lines
750 B
Python
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()
|