Add controller's tests

/cc @josh @rafaelfranca