Initial commit: library-desk service extraction from portainer-core

tests/test_multi_tenancy.py

@@ -0,0 +1,115 @@
+"""Tests for multi-tenancy helpers."""
+
+import pytest
+from src.core.multi_tenancy import (
+    sanitize_user_id,
+    get_qdrant_collection_name,
+    get_wikijs_namespace,
+    get_neo4j_user_label,
+    validate_user_id,
+    is_path_in_user_namespace,
+    DEFAULT_USER
+)
+
+
+class TestSanitizeUserId:
+    """Test user ID sanitization."""
+
+    def test_lowercase_conversion(self):
+        assert sanitize_user_id("JohnDoe") == "johndoe"
+
+    def test_email_conversion(self):
+        assert sanitize_user_id("john@example.com") == "john_at_example_com"
+
+    def test_dot_conversion(self):
+        assert sanitize_user_id("john.doe") == "john_doe"
+
+    def test_space_conversion(self):
+        assert sanitize_user_id("John Doe") == "john_doe"
+
+    def test_special_chars_removal(self):
+        assert sanitize_user_id("john-doe!") == "john_doe"
+
+    def test_consecutive_underscores(self):
+        assert sanitize_user_id("john__doe") == "john_doe"
+
+    def test_leading_trailing_underscores(self):
+        assert sanitize_user_id("_john_") == "john"
+
+
+class TestQdrantCollectionName:
+    """Test Qdrant collection name generation."""
+
+    def test_simple_user(self):
+        assert get_qdrant_collection_name("jpmschweitzer") == "library_desk_jpmschweitzer"
+
+    def test_email_user(self):
+        assert get_qdrant_collection_name("john@example.com") == "library_desk_john_at_example_com"
+
+    def test_default_user(self):
+        assert get_qdrant_collection_name(DEFAULT_USER) == f"library_desk_{DEFAULT_USER}"
+
+
+class TestWikijsNamespace:
+    """Test Wiki.js namespace generation."""
+
+    def test_simple_user(self):
+        assert get_wikijs_namespace("jpmschweitzer") == "/users/jpmschweitzer"
+
+    def test_email_user(self):
+        assert get_wikijs_namespace("john@example.com") == "/users/john_at_example_com"
+
+    def test_starts_with_slash(self):
+        namespace = get_wikijs_namespace("testuser")
+        assert namespace.startswith("/")
+
+
+class TestNeo4jUserLabel:
+    """Test Neo4j user label generation."""
+
+    def test_simple_user(self):
+        assert get_neo4j_user_label("jpmschweitzer") == "User_Jpmschweitzer_Document"
+
+    def test_email_user(self):
+        result = get_neo4j_user_label("john@example.com")
+        # Should be title case
+        assert result == "User_John_At_Example_Com_Document"
+
+    def test_title_case(self):
+        result = get_neo4j_user_label("john_doe")
+        assert result == "User_John_Doe_Document"
+
+
+class TestValidateUserId:
+    """Test user ID validation."""
+
+    def test_valid_simple(self):
+        assert validate_user_id("jpmschweitzer") is True
+
+    def test_valid_email(self):
+        assert validate_user_id("john@example.com") is True
+
+    def test_empty_invalid(self):
+        assert validate_user_id("") is False
+
+    def test_too_long_invalid(self):
+        assert validate_user_id("a" * 101) is False
+
+    def test_no_alphanumeric_invalid(self):
+        assert validate_user_id("___") is False
+
+
+class TestPathInNamespace:
+    """Test path namespace checking."""
+
+    def test_path_in_namespace(self):
+        assert is_path_in_user_namespace("/users/jpmschweitzer/projects", "jpmschweitzer") is True
+
+    def test_path_not_in_namespace(self):
+        assert is_path_in_user_namespace("/users/other/projects", "jpmschweitzer") is False
+
+    def test_public_path_not_in_namespace(self):
+        assert is_path_in_user_namespace("/public/docs", "jpmschweitzer") is False
+
+    def test_root_path(self):
+        assert is_path_in_user_namespace("/users/test", "test") is True