Add controller test helpers