@@ -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