Add missing controller tests