add missing await and fix typing errors

Author: sleepyStick

pymongo/asynchronous/monitor.py

@@ -250,7 +250,7 @@ async def _check_server(self) -> ServerDescription:
             except (OperationFailure, NotPrimaryError) as exc:
                 # Update max cluster time even when hello fails.
                 details = cast(Mapping[str, Any], exc.details)
-                self._topology.receive_cluster_time(details.get("$clusterTime"))
+                await self._topology.receive_cluster_time(details.get("$clusterTime"))
                 raise
         except ReferenceError:
             raise

pymongo/asynchronous/periodic_executor.py

@@ -81,7 +81,7 @@ def close(self, dummy: Any = None) -> None:
     async def join(self, timeout: Optional[int] = None) -> None:
         if self._task is not None:
             try:
-                await asyncio.wait_for(self._task, timeout=timeout)
+                await asyncio.wait_for(self._task, timeout=timeout)  # type-ignore: [arg-type]
             except asyncio.TimeoutError:
                 # Task timed out
                 pass
@@ -274,7 +274,7 @@ def _on_executor_deleted(ref: weakref.ReferenceType[PeriodicExecutor]) -> None:
     _EXECUTORS.remove(ref)
 
 
-async def _shutdown_executors() -> None:
+def _shutdown_executors() -> None:
     if _EXECUTORS is None:
         return
 
@@ -291,6 +291,6 @@ async def _shutdown_executors() -> None:
     for ref in executors:
         executor = ref()
         if executor:
-            await executor.join(1)
+            executor.join(1)
 
     executor = None

pymongo/asynchronous/topology.py

@@ -359,7 +359,9 @@ async def _select_servers_loop(
                 await self._condition.wait(common.MIN_HEARTBEAT_INTERVAL)
             else:
                 try:
-                    await asyncio.wait_for(self._condition.wait(), common.MIN_HEARTBEAT_INTERVAL)
+                    await asyncio.wait_for(
+                        self._condition.wait(), common.MIN_HEARTBEAT_INTERVAL
+                    )  # type-ignore: [arg-type]
                 except asyncio.TimeoutError:
                     pass
             self._description.check_compatible()
@@ -665,8 +667,10 @@ async def request_check_all(self, wait_time: int = 5) -> None:
                 await self._condition.wait(wait_time)
             else:
                 try:
-                    await asyncio.wait_for(self._condition.wait(), wait_time)
-                except TimeoutError:
+                    await asyncio.wait_for(
+                        self._condition.wait(), wait_time
+                    )  # type-ignore: [arg-type]
+                except asyncio.TimeoutError:
                     pass
 
     def data_bearing_servers(self) -> list[ServerDescription]:

pymongo/synchronous/periodic_executor.py

@@ -81,7 +81,7 @@ def close(self, dummy: Any = None) -> None:
     def join(self, timeout: Optional[int] = None) -> None:
         if self._task is not None:
             try:
-                asyncio.wait_for(self._task, timeout=timeout)
+                asyncio.wait_for(self._task, timeout=timeout)  # type-ignore: [arg-type]
             except asyncio.TimeoutError:
                 # Task timed out
                 pass

pymongo/synchronous/topology.py

@@ -359,7 +359,9 @@ def _select_servers_loop(
                 self._condition.wait(common.MIN_HEARTBEAT_INTERVAL)
             else:
                 try:
-                    asyncio.wait_for(self._condition.wait(), common.MIN_HEARTBEAT_INTERVAL)
+                    asyncio.wait_for(
+                        self._condition.wait(), common.MIN_HEARTBEAT_INTERVAL
+                    )  # type-ignore: [arg-type]
                 except asyncio.TimeoutError:
                     pass
             self._description.check_compatible()
@@ -663,8 +665,8 @@ def request_check_all(self, wait_time: int = 5) -> None:
                 self._condition.wait(wait_time)
             else:
                 try:
-                    asyncio.wait_for(self._condition.wait(), wait_time)
-                except TimeoutError:
+                    asyncio.wait_for(self._condition.wait(), wait_time)  # type-ignore: [arg-type]
+                except asyncio.TimeoutError:
                     pass
 
     def data_bearing_servers(self) -> list[ServerDescription]: