Added tests to ui-controller