test file added & mongo tested

Services/MongoDb/Models/action_models/domain.py

@@ -0,0 +1,78 @@
+"""
+MongoDB Domain Models.
+
+This module provides Pydantic models for domain management,
+including domain history and access details.
+"""
+
+from datetime import datetime
+from typing import Any, Dict, List, Optional
+from pydantic import Field, model_validator
+
+from ApiLibrary import system_arrow
+from Services.MongoDb.Models.action_models.base import MongoBaseModel, MongoDocument
+
+
+class DomainData(MongoBaseModel):
+    """Model for domain data.
+    
+    Attributes:
+        user_uu_id: Unique identifier of the user
+        main_domain: Primary domain
+        other_domains_list: List of additional domains
+        extra_data: Additional domain-related data
+    """
+    
+    user_uu_id: str = Field(..., description="User's unique identifier")
+    main_domain: str = Field(..., description="Primary domain")
+    other_domains_list: List[str] = Field(
+        default_factory=list,
+        description="List of additional domains"
+    )
+    extra_data: Optional[Dict[str, Any]] = Field(
+        default_factory=dict,
+        alias="extraData",
+        description="Additional domain-related data"
+    )
+
+    class Config:
+        from_attributes = True
+        populate_by_name = True
+        validate_assignment = True
+
+
+class DomainDocument(MongoDocument):
+    """Model for domain-related documents."""
+    
+    data: DomainData = Field(..., description="Domain data")
+    
+    def update_main_domain(self, new_domain: str) -> None:
+        """Update the main domain and move current to history.
+        
+        Args:
+            new_domain: New main domain to set
+        """
+        if self.data.main_domain and self.data.main_domain != new_domain:
+            if self.data.main_domain not in self.data.other_domains_list:
+                self.data.other_domains_list.append(self.data.main_domain)
+        self.data.main_domain = new_domain
+
+
+class DomainDocumentCreate(MongoDocument):
+    """Model for creating new domain documents."""
+    data: DomainData = Field(..., description="Initial domain data")
+
+    class Config:
+        from_attributes = True
+        populate_by_name = True
+        validate_assignment = True
+
+
+class DomainDocumentUpdate(MongoDocument):
+    """Model for updating existing domain documents."""
+    data: DomainData = Field(..., description="Updated domain data")
+
+    class Config:
+        from_attributes = True
+        populate_by_name = True
+        validate_assignment = True