Add comprehensive unit tests

[pnilan]

Description

Add comprehensive unit tests to improve code coverage and ensure reliability.

Tasks

• Write unit tests for core functionality
• Set up testing framework if not already configured
• Achieve minimum 80% code coverage
• Integrate tests into CI/CD pipeline
• Document testing procedures

Acceptance Criteria

• All critical paths have test coverage
• Tests pass in CI/CD
• Code coverage metrics are tracked