Implement missing tests

There are several missing test cases that were already defined in the codebase but left unimplemented and marked with `// TODO` comments.
These tests should be completed and integrated to improve the overall test coverage and strengthen confidence in existing functionality.