Added controller unit tests