docs: show common job properties in get_job and cancel_job samples (#1137)

* docs: show common job properties in `get_job` and `cancel_job` samples

* flake8

Author: tswast

docs/snippets.py

@@ -757,6 +757,7 @@ def test_client_query_total_rows(client, capsys):
 
 
 def test_manage_job(client):
+    # TODO(b/199162556): delete after migrating docs
     sql = """
         SELECT corpus
         FROM `bigquery-public-data.samples.shakespeare`

samples/snippets/manage_job_cancel.py

@@ -0,0 +1,26 @@
+# Copyright 2016-2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# [START bigquery_cancel_job]
+from google.cloud import bigquery
+
+
+def cancel_job(
+    client: bigquery.Client, location: str = "us", job_id: str = "abcd-efgh-ijkl-mnop",
+):
+    job = client.cancel_job(job_id, location=location)
+    print(f"{job.location}:{job.job_id} cancelled")
+
+
+# [END bigquery_cancel_job]

samples/snippets/manage_job_get.py

@@ -0,0 +1,33 @@
+# Copyright 2016-2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# [START bigquery_get_job]
+from google.cloud import bigquery
+
+
+def get_job(
+    client: bigquery.Client, location: str = "us", job_id: str = "abcd-efgh-ijkl-mnop",
+):
+    job = client.get_job(job_id, location=location)
+
+    # All job classes have "location" and "job_id" string properties.
+    # Use these properties for job operations such as "cancel_job" and
+    # "delete_job".
+    print(f"{job.location}:{job.job_id}")
+    print(f"Type: {job.job_type}")
+    print(f"State: {job.state}")
+    print(f"Created: {job.created.isoformat()}")
+
+
+# [END bigquery_get_job]

samples/snippets/manage_job_test.py

@@ -0,0 +1,39 @@
+# Copyright 2022 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from google.cloud import bigquery
+import pytest
+
+import manage_job_cancel
+import manage_job_get
+
+
+def test_manage_job(capsys: pytest.CaptureFixture):
+    client = bigquery.Client()
+    sql = """
+        SELECT corpus
+        FROM `bigquery-public-data.samples.shakespeare`
+        GROUP BY corpus;
+    """
+    location = "us"
+    job = client.query(sql, location=location)
+
+    manage_job_cancel.cancel_job(client, location=location, job_id=job.job_id)
+    out, _ = capsys.readouterr()
+    assert f"{job.location}:{job.job_id} cancelled" in out
+
+    manage_job_get.get_job(client, location=location, job_id=job.job_id)
+    out, _ = capsys.readouterr()
+    assert f"{job.location}:{job.job_id}" in out
+    assert "Type: query" in out