test: wait for grain call before checking cancellation (#10319)