Add extension controller unit tests