Add tests for focus navigation