test: fix PITR-lite system tests

googleapis · Oct 7, 2020 · d77bae8 · d77bae8
1 parent c1b407d
commit d77bae8
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/tests/system/test_system.py b/tests/system/test_system.py
@@ -339,7 +339,7 @@ def test_create_database_pitr_invalid_retention_period(self):
         retention_period = "0d"
         ddl_statements = [
             "ALTER DATABASE {}"
-            " SET OPTIONS (version_retention_period = {})".format(
+            " SET OPTIONS (version_retention_period = '{}')".format(
                 temp_db_id, retention_period
             )
         ]
@@ -358,7 +358,7 @@ def test_create_database_pitr_success(self):
         retention_period = "7d"
         ddl_statements = [
             "ALTER DATABASE {}"
-            " SET OPTIONS (version_retention_period = {})".format(
+            " SET OPTIONS (version_retention_period = '{}')".format(
                 temp_db_id, retention_period
             )
         ]
@@ -386,7 +386,7 @@ def test_create_database_pitr_success(self):
                 "WHERE SCHEMA_NAME = '' AND OPTION_NAME = 'version_retention_period'"
             )
             for result in results:
-                self.assertEqual(result, retention_period)
+                self.assertEqual(result[0], retention_period)
 
     def test_table_not_found(self):
         temp_db_id = "temp_db" + unique_resource_id("_")
@@ -454,12 +454,11 @@ def test_update_database_ddl_pitr_invalid(self):
         # We want to make sure the operation completes.
         create_op.result(240)  # raises on failure / timeout.
 
-        temp_db.reload()
         self.assertIsNone(temp_db.version_retention_period)
 
         ddl_statements = DDL_STATEMENTS + [
             "ALTER DATABASE {}"
-            " SET OPTIONS (version_retention_period = {})".format(
+            " SET OPTIONS (version_retention_period = '{}')".format(
                 temp_db_id, retention_period
             )
         ]
@@ -480,12 +479,11 @@ def test_update_database_ddl_pitr_success(self):
         # We want to make sure the operation completes.
         create_op.result(240)  # raises on failure / timeout.
 
-        temp_db.reload()
         self.assertIsNone(temp_db.version_retention_period)
 
         ddl_statements = DDL_STATEMENTS + [
             "ALTER DATABASE {}"
-            " SET OPTIONS (version_retention_period = {})".format(
+            " SET OPTIONS (version_retention_period = '{}')".format(
                 temp_db_id, retention_period
             )
         ]