Generated tests for organization and role using pytest.

Author: AkanshuSrivastav

tests/test_organization.py

@@ -0,0 +1,61 @@
+# test_organization.py
+
+import pytest
+from utils.models import Organization
+
+def test_organization_creation():
+    """Test basic organization creation"""
+    org = Organization("Test Org", "Test Description")
+    assert org.name == "Test Org"
+    assert org.description == "Test Description"
+    assert org.roles == []
+    assert org.members == []
+
+def test_organization_add_role():
+    """Test adding a role to organization"""
+    org = Organization("Test Org", "Test Description")
+    role = {"name": "Admin", "permissions": ["read", "write"]}
+    org.add_role(role)
+    assert len(org.roles) == 1
+    assert org.roles[0] == role
+
+def test_organization_add_member():
+    """Test adding a member to organization"""
+    org = Organization("Test Org", "Test Description")
+    member = {"id": 1, "name": "John Doe"}
+    org.add_member(member)
+    assert len(org.members) == 1
+    assert org.members[0] == member
+
+def test_organization_remove_member():
+    """Test removing a member from organization"""
+    org = Organization("Test Org", "Test Description")
+    member = {"id": 1, "name": "John Doe"}
+    org.add_member(member)
+    org.remove_member(1)
+    assert len(org.members) == 0
+
+def test_organization_get_member():
+    """Test getting a member from organization"""
+    org = Organization("Test Org", "Test Description")
+    member = {"id": 1, "name": "John Doe"}
+    org.add_member(member)
+    retrieved_member = org.get_member(1)
+    assert retrieved_member == member
+
+def test_organization_get_nonexistent_member():
+    """Test getting a non-existent member"""
+    org = Organization("Test Org", "Test Description")
+    assert org.get_member(1) is None
+
+# Additional tests for organization.py
+
+def test_organization_invalid_name():
+    """Test organization creation with invalid name"""
+    with pytest.raises(ValueError):
+        Organization("", "Description")
+
+def test_organization_none_name():
+    """Test organization creation with None name"""
+    with pytest.raises(ValueError):
+        Organization(None, "Description")

tests/test_role.py

@@ -0,0 +1,60 @@
+# test_role.py
+
+import pytest
+from utils.models import Role
+
+def test_role_creation():
+    """Test basic role creation"""
+    role = Role("Admin", ["read", "write"])
+    assert role.name == "Admin"
+    assert role.permissions == ["read", "write"]
+
+def test_role_add_permission():
+    """Test adding a permission to role"""
+    role = Role("Admin", ["read"])
+    role.add_permission("write")
+    assert "write" in role.permissions
+    assert len(role.permissions) == 2
+
+def test_role_remove_permission():
+    """Test removing a permission from role"""
+    role = Role("Admin", ["read", "write"])
+    role.remove_permission("write")
+    assert "write" not in role.permissions
+    assert len(role.permissions) == 1
+
+def test_role_has_permission():
+    """Test checking if role has specific permission"""
+    role = Role("Admin", ["read", "write"])
+    assert role.has_permission("read") is True
+    assert role.has_permission("delete") is False
+
+def test_role_add_existing_permission():
+    """Test adding a permission that already exists"""
+    role = Role("Admin", ["read"])
+    role.add_permission("read")
+    assert len(role.permissions) == 1
+
+def test_role_remove_nonexistent_permission():
+    """Test removing a permission that doesn't exist"""
+    role = Role("Admin", ["read"])
+    role.remove_permission("write")
+    assert len(role.permissions) == 1
+    assert role.permissions == ["read"]
+
+# Additional tests for role.py
+
+def test_role_invalid_name():
+    """Test role creation with invalid name"""
+    with pytest.raises(ValueError):
+        Role("", ["read"])
+
+def test_role_none_permissions():
+    """Test role creation with None permissions"""
+    with pytest.raises(ValueError):
+        Role("Admin", None)
+
+def test_role_empty_permissions():
+    """Test role creation with empty permissions list"""
+    role = Role("Admin", [])
+    assert len(role.permissions) == 0