Skip to content

Commit 9316023

Browse files
committed
fix failover2 to failover_v2
1 parent fb1be5e commit 9316023

3 files changed

Lines changed: 12 additions & 12 deletions

File tree

docs/examples/MySQLDjangoFailover.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
'HOST': 'database.cluster-xyz.us-east-1.rds.amazonaws.com',
4141
'PORT': 3306,
4242
'OPTIONS': {
43-
'plugins': 'failover2',
43+
'plugins': 'failover_v2',
4444
'connect_timeout': 10,
4545
'autocommit': True,
4646
},

tests/integration/container/django/test_django_basic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ def django_setup(self, conn_utils):
157157
'HOST': conn_utils.writer_cluster_host,
158158
'PORT': conn_utils.port,
159159
'OPTIONS': {
160-
'plugins': 'failover2,aurora_connection_tracker',
160+
'plugins': 'failover_v2,aurora_connection_tracker',
161161
'connect_timeout': 10,
162162
'autocommit': True,
163163
},

tests/integration/container/django/test_django_plugins.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def django_setup(self, conn_utils, create_secret, request, create_custom_endpoin
313313
# Get configuration from test parameter or use defaults
314314
if hasattr(request, 'param') and isinstance(request.param, dict):
315315
config = request.param
316-
plugins_config = config.get('plugins', 'aurora_connection_tracker,failover2')
316+
plugins_config = config.get('plugins', 'aurora_connection_tracker,failover_v2')
317317
extra_options = config.get('options', {})
318318

319319
# Check if we need to use custom endpoint
@@ -347,7 +347,7 @@ def django_setup(self, conn_utils, create_secret, request, create_custom_endpoin
347347
else:
348348
host = config.get('host', conn_utils.writer_cluster_host)
349349
else:
350-
plugins_config = 'aurora_connection_tracker,failover2'
350+
plugins_config = 'aurora_connection_tracker,failover_v2'
351351
extra_options = {}
352352
user = conn_utils.user
353353
password = conn_utils.password
@@ -448,14 +448,14 @@ def test_django_with_no_plugins(self, test_environment: TestEnvironment, django_
448448
# Clean up
449449
TestModel.objects.all().delete()
450450

451-
@pytest.mark.parametrize('django_setup', [{'plugins': 'failover'}], indirect=True)
451+
@pytest.mark.parametrize('django_setup', [{'plugins': 'failover_v2'}], indirect=True)
452452
def test_django_with_failover_only(self, test_environment: TestEnvironment, django_models, django_setup):
453453
"""Test Django with only failover plugin"""
454454
TestModel = self.TestModel
455455
config = django_setup # Get the config from fixture
456456

457457
# Verify only failover plugin is configured
458-
assert config['plugins'] == 'failover'
458+
assert config['plugins'] == 'failover_v2'
459459

460460
# Test basic functionality works with failover plugin
461461
test_obj = TestModel.objects.create(
@@ -470,14 +470,14 @@ def test_django_with_failover_only(self, test_environment: TestEnvironment, djan
470470
# Clean up
471471
TestModel.objects.all().delete()
472472

473-
@pytest.mark.parametrize('django_setup', [{'plugins': 'aurora_connection_tracker,failover2'}], indirect=True)
473+
@pytest.mark.parametrize('django_setup', [{'plugins': 'aurora_connection_tracker,failover_v2'}], indirect=True)
474474
def test_django_with_multiple_plugins(self, test_environment: TestEnvironment, django_models, django_setup):
475475
"""Test Django with multiple plugins enabled"""
476476
TestModel = self.TestModel
477477
config = django_setup # Get the config from fixture
478478

479479
# Verify multiple plugins are configured
480-
assert config['plugins'] == 'aurora_connection_tracker,failover2'
480+
assert config['plugins'] == 'aurora_connection_tracker,failover_v2'
481481

482482
# Test basic functionality works with multiple plugins
483483
test_obj = TestModel.objects.create(
@@ -559,7 +559,7 @@ def test_django_with_iam_plugin(self, test_environment: TestEnvironment, django_
559559
TestModel.objects.all().delete()
560560

561561
@pytest.mark.parametrize('django_setup', [{
562-
'plugins': 'failover',
562+
'plugins': 'failover_v2',
563563
'options': {
564564
'socket_timeout': 10,
565565
'connect_timeout': 10,
@@ -576,7 +576,7 @@ def test_django_failover_during_query(self, test_environment: TestEnvironment, d
576576
config = django_setup
577577

578578
# Verify failover plugin is configured
579-
assert 'failover' in config['plugins']
579+
assert 'failover_v2' in config['plugins']
580580

581581
# Get initial writer ID
582582
initial_writer_id = rds_utils.get_cluster_writer_instance_id()
@@ -616,7 +616,7 @@ def test_django_failover_during_query(self, test_environment: TestEnvironment, d
616616
TestModel.objects.all().delete()
617617

618618
@pytest.mark.parametrize('django_setup', [{
619-
'plugins': 'custom_endpoint,failover2',
619+
'plugins': 'custom_endpoint,failover_v2',
620620
'use_custom_endpoint': True,
621621
'options': {
622622
'socket_timeout': 10,
@@ -637,7 +637,7 @@ def test_django_custom_endpoint_failover_during_query(
637637

638638
# Verify custom_endpoint and failover plugins are configured
639639
assert 'custom_endpoint' in config['plugins']
640-
assert 'failover' in config['plugins']
640+
assert 'failover_v2' in config['plugins']
641641

642642
# Get initial writer ID
643643
initial_writer_id = rds_utils.get_cluster_writer_instance_id()

0 commit comments

Comments
 (0)