Enhance Test Coverage

[hrosicka]

• Review the current test suite and identify untested modules or features.
  • Expand tests to cover edge cases, error handling, and user interactions.
  • Add integration tests for backend and frontend flows.