Fix tests

backend/app/models/request.py

@@ -1,31 +1,41 @@
 """Request model"""
 from enum import Enum
-from sqlalchemy import Column, Integer, String, ForeignKey, DateTime
-from sqlalchemy.sql import func
+from sqlalchemy import Column, Integer, String, Enum as SQLEnum, ForeignKey, DateTime
 from sqlalchemy.orm import relationship
-from .base import Base
+from datetime import datetime
+
+from ..database import Base
 
 class RequestStatus(str, Enum):
+    """Request status enum"""
     NEW = "new"
     IN_PROGRESS = "in_progress"
     COMPLETED = "completed"
     REJECTED = "rejected"
 
 class RequestPriority(str, Enum):
+    """Request priority enum"""
     LOW = "low"
     MEDIUM = "medium"
     HIGH = "high"
 
+class RequestType(str, Enum):
+    """Request type enum"""
+    VACATION = "vacation"
+    SICK_LEAVE = "sick_leave"
+    EQUIPMENT = "equipment"
+    OTHER = "other"
+
 class Request(Base):
+    """Request model"""
     __tablename__ = "requests"
 
     id = Column(Integer, primary_key=True, 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())
+    request_type = Column(SQLEnum(RequestType), nullable=False)
+    description = Column(String, nullable=False)
+    priority = Column(SQLEnum(RequestPriority), nullable=False, default=RequestPriority.MEDIUM)
+    status = Column(SQLEnum(RequestStatus), nullable=False, default=RequestStatus.NEW)
+    created_at = Column(DateTime, nullable=False, default=datetime.utcnow)
+    employee_id = Column(Integer, ForeignKey("employees.id"), nullable=False)
 
-    # Определяем отношение к Employee
     employee = relationship("Employee", back_populates="requests")