mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Format tests

Browse Source
This commit is contained in:
Noah Stapp 2026-02-19 15:23:02 -05:00
parent 003796fdbe
commit 77464f09c7
2 changed files with 32 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
test/asynchronous/test_retryable_reads.py
View File

@ -263,17 +263,17 @@ class TestRetryableReads(AsyncIntegrationTest):
@async_client_context.require_replica_set
@async_client_context.require_failCommand_fail_point
async def test_retryable_reads_caused_by_overload_errors_are_retried_on_a_different_replicaset_server_when_one_is_available(
async def test_03_01_retryable_reads_caused_by_overload_errors_are_retried_on_a_different_replicaset_server_when_one_is_available(
self
):
listener = OvertCommandListener()
# Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
# 1. Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
client = await self.async_rs_or_single_client(
event_listeners=[listener], retryReads=True, readPreference="primaryPreferred"
)
# Configure a fail point with the RetryableError and SystemOverloadedError error labels.
# 2. Configure a fail point with the RetryableError and SystemOverloadedError error labels.
command_args = {
"configureFailPoint": "failCommand",
"mode": {"times": 1},
@ -285,30 +285,32 @@ class TestRetryableReads(AsyncIntegrationTest):
}
await async_set_fail_point(client, command_args)
# Reset the command event monitor to clear the fail point command from its stored events.
# 3. Reset the command event monitor to clear the fail point command from its stored events.
listener.reset()
# Execute a `find` command with `client`.
# 4. Execute a `find` command with `client`.
await client.t.t.find_one({})
# Assert that one failed command event and one successful command event occurred.
# 5. Assert that one failed command event and one successful command event occurred.
self.assertEqual(len(listener.failed_events), 1)
self.assertEqual(len(listener.succeeded_events), 1)
# Assert that both events occurred on different servers.
# 6. Assert that both events occurred on different servers.
assert listener.failed_events[0].connection_id != listener.succeeded_events[0].connection_id
@async_client_context.require_replica_set
@async_client_context.require_failCommand_fail_point
async def test_retryable_reads_error_are_retried_on_same_replicaset_server(self):
async def test_03_02_retryable_reads_caused_by_non_overload_errors_are_retried_on_the_same_replicaset_server(
self
):
listener = OvertCommandListener()
# Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
# 1. Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
client = await self.async_rs_or_single_client(
event_listeners=[listener], retryReads=True, readPreference="primaryPreferred"
)
# Configure a fail point with the RetryableError error label.
# 2. Configure a fail point with the RetryableError error label.
command_args = {
"configureFailPoint": "failCommand",
"mode": {"times": 1},
@ -320,17 +322,17 @@ class TestRetryableReads(AsyncIntegrationTest):
}
await async_set_fail_point(client, command_args)
# Reset the command event monitor to clear the fail point command from its stored events.
# 3. Reset the command event monitor to clear the fail point command from its stored events.
listener.reset()
# Execute a `find` command with `client`.
# 4. Execute a `find` command with `client`.
await client.t.t.find_one({})
# Assert that one failed command event and one successful command event occurred.
# 5. Assert that one failed command event and one successful command event occurred.
self.assertEqual(len(listener.failed_events), 1)
self.assertEqual(len(listener.succeeded_events), 1)
# Assert that both events occurred the same server.
# 6. Assert that both events occurred the same server.
assert listener.failed_events[0].connection_id == listener.succeeded_events[0].connection_id

30
test/test_retryable_reads.py
View File

@ -261,17 +261,17 @@ class TestRetryableReads(IntegrationTest):
@client_context.require_replica_set
@client_context.require_failCommand_fail_point
def test_retryable_reads_caused_by_overload_errors_are_retried_on_a_different_replicaset_server_when_one_is_available(
def test_03_01_retryable_reads_caused_by_overload_errors_are_retried_on_a_different_replicaset_server_when_one_is_available(
self
):
listener = OvertCommandListener()
# Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
# 1. Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
client = self.rs_or_single_client(
event_listeners=[listener], retryReads=True, readPreference="primaryPreferred"
)
# Configure a fail point with the RetryableError and SystemOverloadedError error labels.
# 2. Configure a fail point with the RetryableError and SystemOverloadedError error labels.
command_args = {
"configureFailPoint": "failCommand",
"mode": {"times": 1},
@ -283,30 +283,32 @@ class TestRetryableReads(IntegrationTest):
}
set_fail_point(client, command_args)
# Reset the command event monitor to clear the fail point command from its stored events.
# 3. Reset the command event monitor to clear the fail point command from its stored events.
listener.reset()
# Execute a `find` command with `client`.
# 4. Execute a `find` command with `client`.
client.t.t.find_one({})
# Assert that one failed command event and one successful command event occurred.
# 5. Assert that one failed command event and one successful command event occurred.
self.assertEqual(len(listener.failed_events), 1)
self.assertEqual(len(listener.succeeded_events), 1)
# Assert that both events occurred on different servers.
# 6. Assert that both events occurred on different servers.
assert listener.failed_events[0].connection_id != listener.succeeded_events[0].connection_id
@client_context.require_replica_set
@client_context.require_failCommand_fail_point
def test_retryable_reads_error_are_retried_on_same_replicaset_server(self):
def test_03_02_retryable_reads_caused_by_non_overload_errors_are_retried_on_the_same_replicaset_server(
self
):
listener = OvertCommandListener()
# Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
# 1. Create a client `client` with `retryReads=true`, `readPreference=primaryPreferred`, and command event monitoring enabled.
client = self.rs_or_single_client(
event_listeners=[listener], retryReads=True, readPreference="primaryPreferred"
)
# Configure a fail point with the RetryableError error label.
# 2. Configure a fail point with the RetryableError error label.
command_args = {
"configureFailPoint": "failCommand",
"mode": {"times": 1},
@ -318,17 +320,17 @@ class TestRetryableReads(IntegrationTest):
}
set_fail_point(client, command_args)
# Reset the command event monitor to clear the fail point command from its stored events.
# 3. Reset the command event monitor to clear the fail point command from its stored events.
listener.reset()
# Execute a `find` command with `client`.
# 4. Execute a `find` command with `client`.
client.t.t.find_one({})
# Assert that one failed command event and one successful command event occurred.
# 5. Assert that one failed command event and one successful command event occurred.
self.assertEqual(len(listener.failed_events), 1)
self.assertEqual(len(listener.succeeded_events), 1)
# Assert that both events occurred the same server.
# 6. Assert that both events occurred the same server.
assert listener.failed_events[0].connection_id == listener.succeeded_events[0].connection_id