From be9b242f2180f5b795dfb3a168a97af1682999fd Mon Sep 17 00:00:00 2001
From: Tim Swast <swast@google.com>
Date: Thu, 22 Jul 2021 10:21:02 -0500
Subject: [PATCH] docs: add sample to delete job metadata (#798)

Planned to be included in https://cloud.google.com/bigquery/docs/managing-jobs
---
 samples/snippets/conftest.py        | 25 ++++++++++++++++
 samples/snippets/delete_job.py      | 44 +++++++++++++++++++++++++++++
 samples/snippets/delete_job_test.py | 33 ++++++++++++++++++++++
 tests/system/test_client.py         | 17 -----------
 4 files changed, 102 insertions(+), 17 deletions(-)
 create mode 100644 samples/snippets/delete_job.py
 create mode 100644 samples/snippets/delete_job_test.py

diff --git a/samples/snippets/conftest.py b/samples/snippets/conftest.py
index 000e5f85c..74984f902 100644
--- a/samples/snippets/conftest.py
+++ b/samples/snippets/conftest.py
@@ -50,6 +50,31 @@ def dataset_id(bigquery_client: bigquery.Client, project_id: str):
     bigquery_client.delete_dataset(dataset, delete_contents=True, not_found_ok=True)
 
 
+@pytest.fixture(scope="session")
+def dataset_id_us_east1(bigquery_client: bigquery.Client, project_id: str):
+    dataset_id = prefixer.create_prefix()
+    full_dataset_id = f"{project_id}.{dataset_id}"
+    dataset = bigquery.Dataset(full_dataset_id)
+    dataset.location = "us-east1"
+    bigquery_client.create_dataset(dataset)
+    yield dataset_id
+    bigquery_client.delete_dataset(dataset, delete_contents=True, not_found_ok=True)
+
+
+@pytest.fixture(scope="session")
+def table_id_us_east1(
+    bigquery_client: bigquery.Client, project_id: str, dataset_id_us_east1: str
+):
+    table_id = prefixer.create_prefix()
+    full_table_id = f"{project_id}.{dataset_id_us_east1}.{table_id}"
+    table = bigquery.Table(
+        full_table_id, schema=[bigquery.SchemaField("string_col", "STRING")]
+    )
+    bigquery_client.create_table(table)
+    yield full_table_id
+    bigquery_client.delete_table(table, not_found_ok=True)
+
+
 @pytest.fixture
 def random_table_id(bigquery_client: bigquery.Client, project_id: str, dataset_id: str):
     """Create a new table ID each time, so random_table_id can be used as
diff --git a/samples/snippets/delete_job.py b/samples/snippets/delete_job.py
new file mode 100644
index 000000000..abed0c90d
--- /dev/null
+++ b/samples/snippets/delete_job.py
@@ -0,0 +1,44 @@
+# Copyright 2021 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
+#
+#     https://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.
+
+
+def delete_job_metadata(job_id: str, location: str):
+    orig_job_id = job_id
+    orig_location = location
+    # [START bigquery_delete_job]
+    from google.cloud import bigquery
+    from google.api_core import exceptions
+
+    # TODO(developer): Set the job ID to the ID of the job whose metadata you
+    #                  wish to delete.
+    job_id = "abcd-efgh-ijkl-mnop"
+
+    # TODO(developer): Set the location to the region or multi-region
+    #                  containing the job.
+    location = "us-east1"
+
+    # [END bigquery_delete_job]
+    job_id = orig_job_id
+    location = orig_location
+
+    # [START bigquery_delete_job]
+    client = bigquery.Client()
+
+    client.delete_job_metadata(job_id, location=location)
+
+    try:
+        client.get_job(job_id, location=location)
+    except exceptions.NotFound:
+        print(f"Job metadata for job {location}:{job_id} was deleted.")
+    # [END bigquery_delete_job]
diff --git a/samples/snippets/delete_job_test.py b/samples/snippets/delete_job_test.py
new file mode 100644
index 000000000..c9baa817d
--- /dev/null
+++ b/samples/snippets/delete_job_test.py
@@ -0,0 +1,33 @@
+# Copyright 2021 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
+#
+#     https://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 delete_job
+
+
+def test_delete_job_metadata(
+    capsys, bigquery_client: bigquery.Client, table_id_us_east1: str
+):
+    query_job: bigquery.QueryJob = bigquery_client.query(
+        f"SELECT COUNT(*) FROM `{table_id_us_east1}`", location="us-east1",
+    )
+    query_job.result()
+    assert query_job.job_id is not None
+
+    delete_job.delete_job_metadata(query_job.job_id, "us-east1")
+
+    out, _ = capsys.readouterr()
+    assert "deleted" in out
+    assert f"us-east1:{query_job.job_id}" in out
diff --git a/tests/system/test_client.py b/tests/system/test_client.py
index ceb62b8cd..2536af9fc 100644
--- a/tests/system/test_client.py
+++ b/tests/system/test_client.py
@@ -63,7 +63,6 @@
 from google.cloud import bigquery_v2
 from google.cloud.bigquery.dataset import Dataset
 from google.cloud.bigquery.dataset import DatasetReference
-from google.cloud.bigquery.schema import SchemaField
 from google.cloud.bigquery.table import Table
 from google.cloud._helpers import UTC
 from google.cloud.bigquery import dbapi, enums
@@ -506,22 +505,6 @@ def test_delete_dataset_delete_contents_false(self):
         with self.assertRaises(exceptions.BadRequest):
             Config.CLIENT.delete_dataset(dataset)
 
-    def test_delete_job_metadata(self):
-        dataset_id = _make_dataset_id("us_east1")
-        self.temp_dataset(dataset_id, location="us-east1")
-        full_table_id = f"{Config.CLIENT.project}.{dataset_id}.test_delete_job_metadata"
-        table = Table(full_table_id, schema=[SchemaField("col", "STRING")])
-        Config.CLIENT.create_table(table)
-        query_job: bigquery.QueryJob = Config.CLIENT.query(
-            f"SELECT COUNT(*) FROM `{full_table_id}`", location="us-east1",
-        )
-        query_job.result()
-        self.assertIsNotNone(Config.CLIENT.get_job(query_job))
-
-        Config.CLIENT.delete_job_metadata(query_job)
-        with self.assertRaises(NotFound):
-            Config.CLIENT.get_job(query_job)
-
     def test_get_table_w_public_dataset(self):
         public = "bigquery-public-data"
         dataset_id = "samples"