Add tests for the menu controller