Add units controller tests.