def create_invocation(
        self,
        id: str,
        start_time_ms: int,
        end_time_ms: int,
        completion_status: str = None,
    ):
        """Helper to build an invocation dict."""
        inv = {
            "id": id,
            "input_records_count": 1000,
            "start_time_ms": start_time_ms,
            "end_time_ms": end_time_ms,
        }
        if completion_status is not None:
            inv["completion_status"] = {
                "status": completion_status,
                "record_count": 1000,
            }
        else:
            inv["completion_status"] = None
        return inv
Helper to build an invocation dict.
tests/unit/test_snowflake_run_status.py