backend.tests.unit.database.test_models

  1import pytest
  2from datetime import datetime
  3from sqlalchemy.exc import IntegrityError
  4from backend.database.models import Worker, Entry
  5
  6
  7def test_create_worker_success(db_session, sample_worker_data):
  8    """
  9    Testuje poprawne utworzenie i zapisanie pracownika (Worker).
 10    """
 11    worker = Worker(
 12        name=sample_worker_data["name"],
 13        face_embedding=sample_worker_data["face_embedding"],
 14        expiration_date=sample_worker_data["expiration_date"],
 15        secret=sample_worker_data["secret"]
 16    )
 17
 18    db_session.session.add(worker)
 19    db_session.session.commit()
 20
 21    assert worker.id is not None
 22    assert worker.name == "Jan Kowalski"
 23    assert worker.secret == "super_secret_qr_code_content"
 24    assert isinstance(worker.face_embedding, bytes)
 25
 26
 27def test_create_worker_missing_fields(db_session):
 28    """
 29    Testuje, czy baza danych rzuca błąd (IntegrityError),
 30    gdy próbujemy zapisać pracownika bez wymaganych pól (nullable=False).
 31    """
 32    # Brak pola 'name' i 'secret'
 33    worker = Worker(
 34        face_embedding=b'some_data',
 35        expiration_date=datetime.now()
 36    )
 37
 38    db_session.session.add(worker)
 39
 40    with pytest.raises(IntegrityError):
 41        db_session.session.commit()
 42
 43
 44def test_create_entry_linked_to_worker(db_session, created_worker):
 45    """
 46    Testuje utworzenie wpisu (Entry) powiązanego z istniejącym pracownikiem.
 47    """
 48    entry = Entry(
 49        worker_id=created_worker.id,
 50        code=200,
 51        message="Access Granted",
 52        face_image=b'snapshot_blob'
 53    )
 54
 55    db_session.session.add(entry)
 56    db_session.session.commit()
 57
 58    assert entry.id is not None
 59    assert entry.worker_id == created_worker.id
 60    assert entry.message == "Access Granted"
 61
 62
 63def test_create_entry_without_worker(db_session):
 64    """
 65    Testuje utworzenie wpisu (Entry) bez powiązania z pracownikiem (worker_id jest nullable).
 66    Np. w przypadku próby wejścia osoby nierozpoznanej.
 67    """
 68    entry = Entry(
 69        worker_id=None,
 70        code=404,
 71        message="User not found",
 72        face_image=b'unknown_face_blob'
 73    )
 74
 75    db_session.session.add(entry)
 76    db_session.session.commit()
 77
 78    assert entry.id is not None
 79    assert entry.worker_id is None
 80
 81
 82def test_entry_default_date(db_session):
 83    """
 84    Testuje, czy pole 'date' w modelu Entry otrzymuje domyślną wartość (server_default),
 85    jeśli nie zostanie podane przy tworzeniu obiektu.
 86    """
 87    entry = Entry(
 88        code=200,
 89        message="Test Date",
 90        worker_id=None
 91    )
 92
 93    db_session.session.add(entry)
 94    db_session.session.commit()
 95
 96    # Po commicie musimy odświeżyć obiekt, aby pobrać wartość wygenerowaną przez serwer bazy danych
 97    db_session.session.refresh(entry)
 98
 99    assert entry.date is not None
100    assert isinstance(entry.date, datetime)
101
102
103def test_entry_missing_required_fields(db_session):
104    """
105    Testuje brak wymaganych pól w modelu Entry (code, message).
106    """
107    entry = Entry(
108        worker_id=None
109        # Brak code i message
110    )
111
112    db_session.session.add(entry)
113
114    with pytest.raises(IntegrityError):
115        db_session.session.commit()