Тесты для бекенда

backend/app/models/request.py

@@ -1,17 +1,17 @@
 """Request model"""
+from enum import Enum
 from sqlalchemy import Column, Integer, String, ForeignKey, DateTime
 from sqlalchemy.sql import func
 from sqlalchemy.orm import relationship
-from enum import Enum as PyEnum
-from ..database import Base
+from ..db.base import Base
 
-class RequestStatus(str, PyEnum):
+class RequestStatus(str, Enum):
     NEW = "new"
     IN_PROGRESS = "in_progress"
     COMPLETED = "completed"
     REJECTED = "rejected"
 
-class RequestPriority(str, PyEnum):
+class RequestPriority(str, Enum):
     LOW = "low"
     MEDIUM = "medium"
     HIGH = "high"
@@ -20,13 +20,13 @@ class Request(Base):
     __tablename__ = "requests"
 
     id = Column(Integer, primary_key=True, index=True)
-    title = Column(String, nullable=False)
-    description = Column(String, nullable=False)
-    status = Column(String, nullable=False, default=RequestStatus.NEW)
-    priority = Column(String, nullable=False)
+    department = Column(String, index=True)
+    request_type = Column(String, index=True)
+    description = Column(String)
+    priority = Column(String)
+    status = Column(String, default=RequestStatus.NEW)
     employee_id = Column(Integer, ForeignKey("employees.id"))
     created_at = Column(DateTime(timezone=True), server_default=func.now())
-    updated_at = Column(DateTime(timezone=True), onupdate=func.now())
 
     # Определяем отношение к Employee
     employee = relationship("Employee", back_populates="requests")