update test charm

tests/integration/application/src/charm.py

@@ -34,7 +34,7 @@ def __init__(self, *args):
         # relation events for mongos client
         self.database = DatabaseRequires(
             self,
-            relation_name="mongos_proxy",
+            relation_name="mongos",
             database_name="my-test-db",
             extra_user_roles=EXTRA_USER_ROLES,
         )

tests/integration/helpers.py

@@ -26,14 +26,12 @@ async def check_mongos(
     ops_test: OpsTest,
     unit: ops.model.Unit,
     auth: bool,
-    app_name: Optional[str],
+    app_name: Optional[str] = None,
     uri: str = None,
     external: bool = False,
 ) -> bool:
     """Returns whether mongos is running on the provided unit."""
-    mongos_check = await generate_mongos_command(
-        ops_test, auth, app_name, uri, external
-    )
+    mongos_check = await generate_mongos_command(ops_test, auth, app_name, uri, external)
 
     # since mongos is communicating only via the unix domain socket, we cannot connect to it via
     # traditional pymongo methods
@@ -69,16 +67,12 @@ async def generate_mongos_uri(
 ) -> str:
     """Generates a URI for accessing mongos."""
     host = (
-        MONGOS_SOCKET
-        if not external
-        else await get_ip_address(ops_test, app_name=MONGOS_APP_NAME)
+        MONGOS_SOCKET if not external else await get_ip_address(ops_test, app_name=MONGOS_APP_NAME)
     )
     if not auth:
         return f"mongodb://{host}:27018"
 
-    secret_uri = await get_application_relation_data(
-        ops_test, app_name, "mongos", "secret-user"
-    )
+    secret_uri = await get_application_relation_data(ops_test, app_name, "mongos", "secret-user")
 
     secret_data = await get_secret_data(ops_test, secret_uri)
     return secret_data.get("uris")
@@ -125,9 +119,7 @@ async def get_application_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
-    ]
+    relation_data = [v for v in data[unit.name]["relation-info"] if v["endpoint"] == relation_name]
     if relation_id:
         # Filter the data based on the relation id.
         relation_data = [v for v in relation_data if v["relation-id"] == relation_id]