Update Python files format

Author: sinclert-canonical

tests/integration/backups.py

@@ -171,9 +171,9 @@ async def pitr_operations(
         "restore", **{"backup-id": backup_id, "restore-to-time": "bad"}
     )
     await action.wait()
-    assert (
-        action.status == "failed"
-    ), "restore should fail with bad restore-to-time parameter, but it succeeded"
+    assert action.status == "failed", (
+        "restore should fail with bad restore-to-time parameter, but it succeeded"
+    )
 
     logger.info(f"Restoring backup {backup_id} with year_before restore-to-time parameter")
     await juju_.run_action(
@@ -183,9 +183,9 @@ async def pitr_operations(
         apps=[MYSQL_APPLICATION_NAME, S3_INTEGRATOR],
         timeout=TIMEOUT,
     )
-    assert await check_test_data_existence(
-        first_mysql_ip, should_not_exist=[td1, td2]
-    ), "test data should not exist"
+    assert await check_test_data_existence(first_mysql_ip, should_not_exist=[td1, td2]), (
+        "test data should not exist"
+    )
 
     logger.info(f"Restoring backup {backup_id} with year_after restore-to-time parameter")
     await juju_.run_action(
@@ -195,9 +195,9 @@ async def pitr_operations(
         apps=[MYSQL_APPLICATION_NAME, S3_INTEGRATOR],
         timeout=TIMEOUT,
     )
-    assert await check_test_data_existence(
-        first_mysql_ip, should_exist=[td1, td2]
-    ), "both test data should exist"
+    assert await check_test_data_existence(first_mysql_ip, should_exist=[td1, td2]), (
+        "both test data should exist"
+    )
 
     logger.info(f"Restoring backup {backup_id} with actual restore-to-time parameter")
     await juju_.run_action(
@@ -219,9 +219,9 @@ async def pitr_operations(
         apps=[MYSQL_APPLICATION_NAME, S3_INTEGRATOR],
         timeout=TIMEOUT,
     )
-    assert await check_test_data_existence(
-        first_mysql_ip, should_exist=[td1, td2]
-    ), "both test data should exist"
+    assert await check_test_data_existence(first_mysql_ip, should_exist=[td1, td2]), (
+        "both test data should exist"
+    )
     clean_backups_from_buckets(cloud_configs, cloud_credentials)
 
 

tests/integration/helpers.py

@@ -120,9 +120,9 @@ async def get_relation_data(
     data = yaml.safe_load(raw_data)
     # Filter the data based on the relation name.
     relation_data = [v for v in data[unit_name]["relation-info"] if v["endpoint"] == relation_name]
-    assert (
-        relation_data
-    ), f"no relation data could be grabbed on relation with endpoint {relation_name}"
+    assert relation_data, (
+        f"no relation data could be grabbed on relation with endpoint {relation_name}"
+    )
 
     return relation_data
 
@@ -408,9 +408,9 @@ async def get_process_pid(
     if return_code == 1:
         return None
 
-    assert (
-        return_code == 0
-    ), f"Failed getting pid, unit={unit_name}, container={container_name}, process={process}"
+    assert return_code == 0, (
+        f"Failed getting pid, unit={unit_name}, container={container_name}, process={process}"
+    )
 
     stripped_pid = pid.strip()
     if not stripped_pid:

tests/integration/high_availability/high_availability_helpers.py

@@ -337,9 +337,9 @@ async def send_signal_to_pod_container_process(
     )
     response.run_forever(timeout=5)
 
-    assert (
-        response.returncode == 0
-    ), f"Failed to send {signal_code} signal, unit={unit_name}, container={container_name}, process={process}"
+    assert response.returncode == 0, (
+        f"Failed to send {signal_code} signal, unit={unit_name}, container={container_name}, process={process}"
+    )
 
 
 async def get_process_stat(
@@ -362,9 +362,9 @@ async def get_process_stat(
     ]
     return_code, stat, _ = await ops_test.juju(*get_stat_commands)
 
-    assert (
-        return_code == 0
-    ), f"Failed to get STAT, unit_name={unit_name}, container_name={container_name}, process={process}"
+    assert return_code == 0, (
+        f"Failed to get STAT, unit_name={unit_name}, container_name={container_name}, process={process}"
+    )
 
     return stat
 
@@ -498,9 +498,9 @@ async def ensure_all_units_continuous_writes_incrementing(
                         ops_test, unit, credentials
                     )
                     logger.info(f"{max_written_value=} on unit {unit.name}")
-                    assert (
-                        max_written_value > last_max_written_value
-                    ), "Continuous writes not incrementing"
+                    assert max_written_value > last_max_written_value, (
+                        "Continuous writes not incrementing"
+                    )
 
                     last_max_written_value = max_written_value
 
@@ -573,9 +573,9 @@ async def ensure_process_not_running(
     get_pid_commands = ["ssh", "--container", container_name, unit_name, "pgrep", "-x", process]
     return_code, pid, _ = await ops_test.juju(*get_pid_commands)
 
-    assert (
-        return_code != 0
-    ), f"Process {process} is still running with pid {pid} on unit {unit_name}, container {container_name}"
+    assert return_code != 0, (
+        f"Process {process} is still running with pid {pid} on unit {unit_name}, container {container_name}"
+    )
 
 
 def get_sts_partition(ops_test: OpsTest, app_name: str) -> int:

tests/integration/high_availability/test_async_replication.py

@@ -250,9 +250,9 @@ async def test_standby_promotion(
     assert results[0] > 1, "No data was written to the database"
 
     cluster_set_status = await get_cluster_status(leader_unit, cluster_set=True)
-    assert (
-        cluster_set_status["clusters"]["cuzco"]["clusterrole"] == "primary"
-    ), "standby not promoted to primary"
+    assert cluster_set_status["clusters"]["cuzco"]["clusterrole"] == "primary", (
+        "standby not promoted to primary"
+    )
 
 
 @markers.juju3
@@ -278,12 +278,12 @@ async def test_failover(ops_test: OpsTest, first_model: Model, second_model: Mod
     )
 
     cluster_set_status = await get_cluster_status(leader_unit, cluster_set=True)
-    assert (
-        cluster_set_status["clusters"]["lima"]["clusterrole"] == "primary"
-    ), "standby not promoted to primary"
-    assert (
-        cluster_set_status["clusters"]["cuzco"]["globalstatus"] == "invalidated"
-    ), "old primary not invalidated"
+    assert cluster_set_status["clusters"]["lima"]["clusterrole"] == "primary", (
+        "standby not promoted to primary"
+    )
+    assert cluster_set_status["clusters"]["cuzco"]["globalstatus"] == "invalidated", (
+        "old primary not invalidated"
+    )
 
     # restore mysqld process
     for unit in second_model_units:

tests/integration/high_availability/test_k8s_endpoints.py

@@ -85,9 +85,9 @@ async def test_labeling_of_k8s_endpoints(ops_test: OpsTest, charm):
     )
 
     for primary in cluster_one_primary_addresses:
-        assert (
-            primary in cluster_one_ips
-        ), f"{primary} (not belonging to cluster 1) should not be in cluster one addresses"
+        assert primary in cluster_one_ips, (
+            f"{primary} (not belonging to cluster 1) should not be in cluster one addresses"
+        )
 
     assert set(cluster_one_primary_addresses + cluster_one_replica_addresses) == set(
         cluster_one_ips
@@ -106,9 +106,9 @@ async def test_labeling_of_k8s_endpoints(ops_test: OpsTest, charm):
     )
 
     for primary in cluster_two_primary_addresses:
-        assert (
-            primary in cluster_two_ips
-        ), f"{primary} (not belonging to cluster w) should not be in cluster two addresses"
+        assert primary in cluster_two_ips, (
+            f"{primary} (not belonging to cluster w) should not be in cluster two addresses"
+        )
 
     assert set(cluster_two_primary_addresses + cluster_two_replica_addresses) == set(
         cluster_two_ips

tests/integration/high_availability/test_log_rotation.py

@@ -86,9 +86,9 @@ async def test_log_rotation(
 
     for file in log_files:
         # audit.log can be rotated and new file not created until access to db
-        assert (
-            file in ls_output or file == "audit.log"
-        ), f"❌ files other than log files exist {ls_output}"
+        assert file in ls_output or file == "audit.log", (
+            f"❌ files other than log files exist {ls_output}"
+        )
 
     logger.info("Dispatching custom event to rotate logs")
     await dispatch_custom_event_for_logrotate(ops_test, unit.name)

tests/integration/high_availability/test_node_drain.py

@@ -36,9 +36,9 @@ async def test_pod_eviction_and_pvc_deletion(
 
     logger.info("Waiting until 3 mysql instances are online")
     # ensure all units in the cluster are online
-    assert await ensure_n_online_mysql_members(
-        ops_test, 3
-    ), "The deployed mysql application is not fully online"
+    assert await ensure_n_online_mysql_members(ops_test, 3), (
+        "The deployed mysql application is not fully online"
+    )
 
     logger.info("Ensuring all units have continuous writes incrementing")
     await ensure_all_units_continuous_writes_incrementing(ops_test, credentials=credentials)
@@ -65,9 +65,9 @@ async def test_pod_eviction_and_pvc_deletion(
         )
 
     logger.info("Waiting until 3 mysql instances are online")
-    assert await ensure_n_online_mysql_members(
-        ops_test, 3
-    ), "The deployed mysql application is not fully online after primary pod eviction"
+    assert await ensure_n_online_mysql_members(ops_test, 3), (
+        "The deployed mysql application is not fully online after primary pod eviction"
+    )
 
     logger.info("Ensuring all units have continuous writes incrementing")
     await ensure_all_units_continuous_writes_incrementing(ops_test, credentials=credentials)

tests/integration/high_availability/test_replication_reelection.py

@@ -61,9 +61,9 @@ async def test_kill_primary_check_reelection(
         assert primary_name != new_primary_name
 
         # wait (and retry) until the killed pod is back online in the mysql cluster
-        assert await ensure_n_online_mysql_members(
-            ops_test, 3
-        ), "Old primary has not come back online after being killed"
+        assert await ensure_n_online_mysql_members(ops_test, 3), (
+            "Old primary has not come back online after being killed"
+        )
 
     await ensure_all_units_continuous_writes_incrementing(ops_test, credentials=credentials)
 

tests/integration/high_availability/test_replication_scaling.py

@@ -54,9 +54,9 @@ async def test_scaling_without_data_loss(
 
     # scale up the mysql application
     await scale_application(ops_test, mysql_application_name, 4)
-    assert await ensure_n_online_mysql_members(
-        ops_test, 4
-    ), "The cluster is not fully online after scaling up"
+    assert await ensure_n_online_mysql_members(ops_test, 4), (
+        "The cluster is not fully online after scaling up"
+    )
 
     # ensure value inserted before scale exists in all units
     for attempt in Retrying(stop=stop_after_delay(10), wait=wait_fixed(2)):
@@ -84,9 +84,9 @@ async def test_scaling_without_data_loss(
 
     # scale down the mysql application
     await scale_application(ops_test, mysql_application_name, 3)
-    assert await ensure_n_online_mysql_members(
-        ops_test, 3
-    ), "The cluster is not fully online after scaling down"
+    assert await ensure_n_online_mysql_members(ops_test, 3), (
+        "The cluster is not fully online after scaling down"
+    )
 
     # ensure data written before scale down is persisted
     for unit in ops_test.model.applications[mysql_application_name].units:

tests/integration/high_availability/test_self_healing_network_cut.py

@@ -33,9 +33,9 @@ async def test_network_cut_affecting_an_instance(
     assert mysql_application_name, "mysql application name is not set"
 
     logger.info("Ensuring that there are 3 online mysql members")
-    assert await ensure_n_online_mysql_members(
-        ops_test, 3
-    ), "The deployed mysql application does not have three online nodes"
+    assert await ensure_n_online_mysql_members(ops_test, 3), (
+        "The deployed mysql application does not have three online nodes"
+    )
 
     logger.info("Ensuring that all instances have incrementing continuous writes")
     await ensure_all_units_continuous_writes_incrementing(ops_test, credentials=credentials)
@@ -102,9 +102,9 @@ async def test_network_cut_affecting_an_instance(
     assert isolated_primary_memberrole == "secondary"
 
     logger.info("Ensure there are 3 online mysql members")
-    assert await ensure_n_online_mysql_members(
-        ops_test, 3
-    ), "The deployed mysql application does not have three online nodes"
+    assert await ensure_n_online_mysql_members(ops_test, 3), (
+        "The deployed mysql application does not have three online nodes"
+    )
 
     logger.info("Ensure all units have incrementing continuous writes")
     await ensure_all_units_continuous_writes_incrementing(ops_test, credentials=credentials)