Add unit test for route