Robottest

backends/advanced/src/advanced_omi_backend/controllers/user_controller.py

@@ -17,7 +17,7 @@
 from advanced_omi_backend.client_manager import get_user_clients_all
 from advanced_omi_backend.database import chunks_col, db, users_col
 from advanced_omi_backend.memory import get_memory_service
-from advanced_omi_backend.users import User, UserCreate
+from advanced_omi_backend.users import User, UserCreate, UserUpdate
 
 logger = logging.getLogger(__name__)
 
@@ -78,8 +78,9 @@ async def create_user(user_data: UserCreate):
         )
 
 
-async def update_user(user_id: str, user_data: UserCreate):
+async def update_user(user_id: str, user_data: UserUpdate):
     """Update an existing user."""
+    print("DEBUG: New update_user function is being called!")
     try:
         # Validate ObjectId format
         try:
@@ -106,23 +107,9 @@ async def update_user(user_id: str, user_data: UserCreate):
 
         # Convert to User object for the manager
         user_obj = User(**existing_user)
-
-        # Prepare update data - only include non-None fields
-        update_data = {}
-        if user_data.email:
-            update_data["email"] = user_data.email
-        if user_data.display_name is not None:
-            update_data["display_name"] = user_data.display_name
-        if hasattr(user_data, 'is_superuser'):
-            update_data["is_superuser"] = user_data.is_superuser
-        if hasattr(user_data, 'is_active'):
-            update_data["is_active"] = user_data.is_active
-        if user_data.password:
-            # Hash the password if provided
-            update_data["hashed_password"] = user_manager.password_helper.hash(user_data.password)
-
-        # Update the user
-        updated_user = await user_manager.update(user_obj, update_data)
+
+        # Update the user using the fastapi-users manager (now with fix for missing method)
+        updated_user = await user_manager.update(user_obj, user_data)
 
         return JSONResponse(
             status_code=200,

backends/advanced/src/advanced_omi_backend/routers/modules/user_routes.py

@@ -10,7 +10,7 @@
 
 from advanced_omi_backend.auth import current_superuser
 from advanced_omi_backend.controllers import user_controller
-from advanced_omi_backend.users import User, UserCreate
+from advanced_omi_backend.users import User, UserCreate, UserUpdate
 
 logger = logging.getLogger(__name__)
 
@@ -30,7 +30,7 @@ async def create_user(user_data: UserCreate, current_user: User = Depends(curren
 
 
 @router.put("/{user_id}")
-async def update_user(user_id: str, user_data: UserCreate, current_user: User = Depends(current_superuser)):
+async def update_user(user_id: str, user_data: UserUpdate, current_user: User = Depends(current_superuser)):
     """Update a user. Admin only."""
     return await user_controller.update_user(user_id, user_data)
 

backends/advanced/src/advanced_omi_backend/users.py

@@ -16,6 +16,7 @@ class UserCreate(BaseUserCreate):
     """Schema for creating new users."""
 
     display_name: Optional[str] = None
+    is_superuser: Optional[bool] = False
 
 
 class UserRead(BaseUser[PydanticObjectId]):
@@ -28,8 +29,26 @@ class UserRead(BaseUser[PydanticObjectId]):
 
 class UserUpdate(BaseUserUpdate):
     """Schema for updating user data."""
-    
+
     display_name: Optional[str] = None
+    is_superuser: Optional[bool] = None
+
+    def create_update_dict_superuser(self):
+        """Create update dictionary for superuser operations."""
+        update_dict = {}
+        if self.email is not None:
+            update_dict["email"] = self.email
+        if self.password is not None:
+            update_dict["password"] = self.password
+        if self.is_active is not None:
+            update_dict["is_active"] = self.is_active
+        if self.is_verified is not None:
+            update_dict["is_verified"] = self.is_verified
+        if self.is_superuser is not None:
+            update_dict["is_superuser"] = self.is_superuser
+        if self.display_name is not None:
+            update_dict["display_name"] = self.display_name
+        return update_dict
 
 
 class User(BeanieBaseUser, Document):

tests/README.md

@@ -0,0 +1,230 @@
+# Friend-Lite API Tests
+
+Comprehensive Robot Framework test suite for the Friend-Lite advanced backend API endpoints.
+
+## Test Structure
+
+### Test Files
+- **`auth_tests.robot`** - Authentication and user management tests
+- **`memory_tests.robot`** - Memory management and search tests
+- **`conversation_tests.robot`** - Conversation management and versioning tests
+- **`health_tests.robot`** - Health check and status endpoint tests
+- **`chat_tests.robot`** - Chat service and session management tests
+- **`client_queue_tests.robot`** - Client management and queue monitoring tests
+- **`system_admin_tests.robot`** - System administration and configuration tests
+- **`all_api_tests.robot`** - Master test suite runner
+
+### Resource Files
+- **`resources/auth_keywords.robot`** - Authentication helper keywords
+- **`resources/memory_keywords.robot`** - Memory management keywords
+- **`resources/conversation_keywords.robot`** - Conversation management keywords
+- **`resources/chat_keywords.robot`** - Chat service keywords
+- **`resources/setup_resources.robot`** - Basic setup and health check keywords
+- **`resources/login_resources.robot`** - Login-specific utilities
+
+### Configuration
+- **`test_env.py`** - Environment configuration and test data
+- **`.env`** - Environment variables (create from template)
+
+## Running Tests
+
+### Prerequisites
+1. Friend-Lite backend running at `http://localhost:8001` (or set `API_URL` in `.env`)
+2. Admin user credentials configured in `.env`
+3. Robot Framework and RequestsLibrary installed
+
+### Environment Setup
+```bash
+# Copy environment template
+cp .env.template .env
+
+# Edit .env with your configuration
+API_URL=http://localhost:8001
+ADMIN_EMAIL=admin@example.com
+ADMIN_PASSWORD=your-secure-admin-password
+```
+
+### Running Individual Test Suites
+```bash
+# Authentication and user tests
+robot auth_tests.robot
+
+# Memory management tests
+robot memory_tests.robot
+
+# Conversation management tests
+robot conversation_tests.robot
+
+# Health and status tests
+robot health_tests.robot
+
+# Chat service tests
+robot chat_tests.robot
+
+# Client and queue tests
+robot client_queue_tests.robot
+
+# System administration tests
+robot system_admin_tests.robot
+```
+
+### Running All Tests
+```bash
+# Run complete test suite
+robot *.robot
+
+# Run with specific tags
+robot --include auth *.robot
+robot --include positive *.robot
+robot --include admin *.robot
+```
+
+### Test Output
+```bash
+# Custom output directory
+robot --outputdir results *.robot
+
+# Verbose logging
+robot --loglevel DEBUG *.robot
+
+# Parallel execution
+pabot --processes 4 *.robot
+```
+
+## Test Coverage
+
+### Authentication & Users (`/api/users`, `/auth`)
+- ✅ Login with valid/invalid credentials
+- ✅ Get current user information
+- ✅ Create/update/delete users (admin only)
+- ✅ User authorization and access control
+- ✅ Admin privilege enforcement
+
+### Memory Management (`/api/memories`)
+- ✅ Get user memories with pagination
+- ✅ Search memories with similarity thresholds
+- ✅ Get memories with transcripts
+- ✅ Delete specific memories
+- ✅ Admin memory access across users
+- ✅ Unfiltered memory access for debugging
+
+### Conversation Management (`/api/conversations`)
+- ✅ List and retrieve conversations
+- ✅ Conversation version history
+- ✅ Transcript reprocessing
+- ✅ Memory reprocessing with version selection
+- ✅ Version activation (transcript/memory)
+- ✅ Conversation deletion and cleanup
+- ✅ User data isolation
+
+### Health & Status (`/health`, `/readiness`)
+- ✅ Main health check with service details
+- ✅ Readiness check for orchestration
+- ✅ Authentication service health
+- ✅ Queue system health status
+- ✅ Chat service health check
+- ✅ System metrics (admin only)
+
+### Chat Service (`/api/chat`)
+- ✅ Session creation and management
+- ✅ Session title updates
+- ✅ Message retrieval
+- ✅ Chat statistics
+- ✅ Memory extraction from sessions
+- ✅ Session deletion and cleanup
+
+### Client & Queue Management
+- ✅ Active client monitoring
+- ✅ Queue job listing with pagination
+- ✅ Queue statistics and health
+- ✅ User job isolation
+- ✅ Processing task monitoring (admin only)
+
+### System Administration
+- ✅ Authentication configuration
+- ✅ Diarization settings management
+- ✅ Speaker configuration
+- ✅ Memory configuration (YAML)
+- ✅ Configuration validation and reload
+- ✅ Bulk memory deletion
+
+## Test Categories
+
+### By Access Level
+- **Public**: Health checks, auth config
+- **User**: Memories, conversations, chat sessions
+- **Admin**: User management, system config, metrics
+
+### By Test Type
+- **Positive**: Valid operations and expected responses
+- **Negative**: Invalid inputs, unauthorized access
+- **Security**: Authentication, authorization, data isolation
+- **Integration**: Cross-service functionality
+
+### By Component
+- **Auth**: Authentication and authorization
+- **Memory**: Memory storage and retrieval
+- **Conversation**: Audio processing and transcription
+- **Chat**: Interactive chat functionality
+- **System**: Configuration and administration
+
+## Key Features Tested
+
+### Security
+- JWT token authentication
+- Role-based access control (admin vs user)
+- Data isolation between users
+- Unauthorized access prevention
+
+### Data Management
+- CRUD operations for all entities
+- Pagination and filtering
+- Search functionality with thresholds
+- Versioning and history tracking
+
+### System Integration
+- Service health monitoring
+- Configuration management
+- Queue system monitoring
+- Cross-service communication
+
+### Error Handling
+- Invalid input validation
+- Non-existent resource handling
+- Permission denied scenarios
+- Service unavailability graceful degradation
+
+## Maintenance
+
+### Adding New Tests
+1. Create test file or add to existing suite
+2. Use appropriate resource keywords
+3. Follow naming conventions (`Test Name Test`)
+4. Include proper tags and documentation
+5. Add cleanup in teardown if needed
+
+### Updating Keywords
+1. Modify resource files for reusable functionality
+2. Keep keywords focused and single-purpose
+3. Use proper argument handling
+4. Include documentation strings
+
+### Environment Variables
+Update `test_env.py` when adding new configuration options or test data.
+
+## Troubleshooting
+
+### Common Issues
+- **401 Unauthorized**: Check admin credentials in `.env`
+- **Connection Refused**: Ensure backend is running
+- **Test Failures**: Check service health endpoints first
+- **Timeout Errors**: Increase timeouts in test configuration
+
+### Debug Mode
+```bash
+# Run with detailed logging
+robot --loglevel TRACE auth_tests.robot
+
+# Stop on first failure
+robot --exitonfailure *.robot
+```

tests/all_api_tests.robot

@@ -0,0 +1,58 @@
+*** Settings ***
+Documentation    Master Test Suite for All Friend-Lite API Endpoints
+Suite Setup      Master Suite Setup
+Suite Teardown   Master Suite Teardown
+
+*** Test Cases ***
+
+Run Auth Tests
+    [Documentation]    Execute authentication and user management tests
+    [Tags]             auth    users    suite
+    Run Tests    auth_tests.robot
+
+Run Memory Tests
+    [Documentation]    Execute memory management tests
+    [Tags]             memory    suite
+    Run Tests    memory_tests.robot
+
+Run Conversation Tests
+    [Documentation]    Execute conversation management tests
+    [Tags]             conversation    suite
+    Run Tests    conversation_tests.robot
+
+Run Health Tests
+    [Documentation]    Execute health and status tests
+    [Tags]             health    status    suite
+    Run Tests    health_tests.robot
+
+Run Chat Tests
+    [Documentation]    Execute chat service tests
+    [Tags]             chat    suite
+    Run Tests    chat_tests.robot
+
+Run Client Queue Tests
+    [Documentation]    Execute client and queue management tests
+    [Tags]             client    queue    suite
+    Run Tests    client_queue_tests.robot
+
+Run System Admin Tests
+    [Documentation]    Execute system and admin tests
+    [Tags]             system    admin    suite
+    Run Tests    system_admin_tests.robot
+
+*** Keywords ***
+
+Master Suite Setup
+    [Documentation]    Setup for the entire test suite
+    Log    Starting Friend-Lite API Test Suite
+    Log    Testing against: ${API_URL}
+
+Master Suite Teardown
+    [Documentation]    Cleanup for the entire test suite
+    Log    Friend-Lite API Test Suite completed
+
+Run Tests
+    [Documentation]    Run a specific test file
+    [Arguments]    ${test_file}
+    Log    Executing: ${test_file}
+    # Note: This is a placeholder - actual test execution handled by robot command