inital

tests/unit/test_models.py

@@ -0,0 +1,76 @@
+"""Unit tests for models."""
+import pytest
+from datetime import datetime
+from app.models import User, Organization
+from app.utils.constants import UserStatus
+
+
+@pytest.mark.unit
+class TestUserModel:
+    """Tests for User model."""
+
+    def test_create_user(self, db):
+        """Test creating a user."""
+        user = User(
+            email="test@example.com",
+            full_name="Test User",
+            status=UserStatus.ACTIVE,
+        )
+        user.save()
+
+        assert user.id is not None
+        assert user.email == "test@example.com"
+        assert user.full_name == "Test User"
+        assert user.status == UserStatus.ACTIVE
+        assert user.created_at is not None
+        assert user.deleted_at is None
+
+    def test_user_to_dict(self, test_user):
+        """Test user to_dict method."""
+        user_dict = test_user.to_dict()
+
+        assert "id" in user_dict
+        assert "email" in user_dict
+        assert user_dict["email"] == test_user.email
+        assert "created_at" in user_dict
+
+    def test_user_soft_delete(self, test_user):
+        """Test soft deleting a user."""
+        test_user.delete(soft=True)
+
+        assert test_user.deleted_at is not None
+        assert isinstance(test_user.deleted_at, datetime)
+
+
+@pytest.mark.unit
+class TestOrganizationModel:
+    """Tests for Organization model."""
+
+    def test_create_organization(self, db):
+        """Test creating an organization."""
+        org = Organization(
+            name="Test Org",
+            slug="test-org",
+            description="Test organization",
+        )
+        org.save()
+
+        assert org.id is not None
+        assert org.name == "Test Org"
+        assert org.slug == "test-org"
+        assert org.is_active is True
+        assert org.created_at is not None
+
+    def test_organization_to_dict(self, test_organization):
+        """Test organization to_dict method."""
+        org_dict = test_organization.to_dict()
+
+        assert "id" in org_dict
+        assert "name" in org_dict
+        assert org_dict["name"] == test_organization.name
+        assert "slug" in org_dict
+
+    def test_get_member_count(self, test_organization):
+        """Test getting member count."""
+        count = test_organization.get_member_count()
+        assert count == 1  # Only the owner