assert_called_once -> assert_called_once_with