Store timezone aware snapshot time on snapshots

Author: plamut

google/cloud/bigquery/table.py

@@ -1306,10 +1306,9 @@ def __init__(self, resource: Dict[str, Any]):
 
         self.snapshot_time = None
         if "snapshotTime" in resource:
-            dt = google.cloud._helpers._rfc3339_to_datetime(resource["snapshotTime"])
-            # The helper returns a timezone *aware* datetime object (with UTC tzinfo),
-            # we need to make it naive.
-            self.snapshot_time = dt.replace(tzinfo=None)
+            self.snapshot_time = google.cloud._helpers._rfc3339_to_datetime(
+                resource["snapshotTime"]
+            )
 
 
 class Row(object):

tests/unit/test_table.py

@@ -692,6 +692,7 @@ def test_snapshot_definition_not_set(self):
         assert table.snapshot_definition is None
 
     def test_snapshot_definition_set(self):
+        from google.cloud._helpers import UTC
         from google.cloud.bigquery.table import SnapshotDefinition
 
         dataset = DatasetReference(self.PROJECT, self.DS_ID)
@@ -714,7 +715,7 @@ def test_snapshot_definition_set(self):
             "/projects/project_x/datasets/dataset_y/tables/table_z"
         )
         assert snapshot.snapshot_time == datetime.datetime(
-            2010, 9, 28, 10, 20, 30, 123000
+            2010, 9, 28, 10, 20, 30, 123000, tzinfo=UTC
         )
 
     def test_description_setter_bad_value(self):