Add tests for existing functionality
Add assertions for error logging