Got basic UT for controller working