3737from test_utils .system import unique_resource_id
3838from test_utils .retry import RetryErrors
3939
40- from google .cloud ._helpers import UTC
4140from google .cloud .bigtable import Client
4241from google .cloud .bigtable import enums
4342from google .cloud .bigtable import column_family
5453STORAGE_TYPE = enums .StorageType .SSD
5554LABEL_KEY = "python-snippet"
5655LABEL_STAMP = (
57- datetime .datetime .now (datetime .UTC )
58- .replace (microsecond = 0 , tzinfo = UTC )
56+ datetime .datetime .now (datetime .timezone . utc )
57+ .replace (microsecond = 0 )
5958 .strftime ("%Y-%m-%dt%H-%M-%S" )
6059)
6160LABELS = {LABEL_KEY : str (LABEL_STAMP )}
@@ -179,7 +178,7 @@ def test_bigtable_write_read_drop_truncate():
179178 value = "value_{}" .format (i ).encode ()
180179 row = table .row (row_key )
181180 row .set_cell (
182- COLUMN_FAMILY_ID , col_name , value , timestamp = datetime .datetime .now (datetime .UTC )
181+ COLUMN_FAMILY_ID , col_name , value , timestamp = datetime .datetime .now (datetime .timezone . utc )
183182 )
184183 rows .append (row )
185184 response = table .mutate_rows (rows )
@@ -270,7 +269,7 @@ def test_bigtable_mutations_batcher():
270269 row_key = row_keys [0 ]
271270 row = table .row (row_key )
272271 row .set_cell (
273- COLUMN_FAMILY_ID , column_name , "value-0" , timestamp = datetime .datetime .now (datetime .UTC )
272+ COLUMN_FAMILY_ID , column_name , "value-0" , timestamp = datetime .datetime .now (datetime .timezone . utc )
274273 )
275274 batcher .mutate (row )
276275 # Add a collections of rows
@@ -279,7 +278,7 @@ def test_bigtable_mutations_batcher():
279278 row = table .row (row_keys [i ])
280279 value = "value_{}" .format (i ).encode ()
281280 row .set_cell (
282- COLUMN_FAMILY_ID , column_name , value , timestamp = datetime .datetime .now (datetime .UTC )
281+ COLUMN_FAMILY_ID , column_name , value , timestamp = datetime .datetime .now (datetime .timezone . utc )
283282 )
284283 rows .append (row )
285284 batcher .mutate_rows (rows )
@@ -759,7 +758,7 @@ def test_bigtable_batcher_mutate_flush_mutate_rows():
759758 row_key = b"row_key_1"
760759 row = table .row (row_key )
761760 row .set_cell (
762- COLUMN_FAMILY_ID , COL_NAME1 , "value-0" , timestamp = datetime .datetime .now (datetime .UTC )
761+ COLUMN_FAMILY_ID , COL_NAME1 , "value-0" , timestamp = datetime .datetime .now (datetime .timezone . utc )
763762 )
764763
765764 # In batcher, mutate will flush current batch if it
@@ -967,12 +966,12 @@ def test_bigtable_row_data_cells_cell_value_cell_values():
967966 value = b"value_in_col1"
968967 row = Config .TABLE .row (b"row_key_1" )
969968 row .set_cell (
970- COLUMN_FAMILY_ID , COL_NAME1 , value , timestamp = datetime .datetime .now (datetime .UTC )
969+ COLUMN_FAMILY_ID , COL_NAME1 , value , timestamp = datetime .datetime .now (datetime .timezone . utc )
971970 )
972971 row .commit ()
973972
974973 row .set_cell (
975- COLUMN_FAMILY_ID , COL_NAME1 , value , timestamp = datetime .datetime .now (datetime .UTC )
974+ COLUMN_FAMILY_ID , COL_NAME1 , value , timestamp = datetime .datetime .now (datetime .timezone . utc )
976975 )
977976 row .commit ()
978977
@@ -1050,7 +1049,7 @@ def test_bigtable_row_setcell_rowkey():
10501049
10511050 cell_val = b"cell-val"
10521051 row .set_cell (
1053- COLUMN_FAMILY_ID , COL_NAME1 , cell_val , timestamp = datetime .datetime .now (datetime .UTC )
1052+ COLUMN_FAMILY_ID , COL_NAME1 , cell_val , timestamp = datetime .datetime .now (datetime .timezone . utc )
10541053 )
10551054 # [END bigtable_api_row_set_cell]
10561055
0 commit comments