Add STI tests to hosts_controller