Add tests for custom routes