@@ -501,16 +501,16 @@ def test_s3_relation_credentials_changed(self, mock_add_s3_credentials):
501501 },
502502 )
503503
504+ expected_credentials = {
505+ "access_key" : "ak" ,
506+ "secret_key" : "sk" ,
507+ "endpoint" : "https://s3.example" ,
508+ }
504509 self .assertEqual (
505510 harness .charm ._stored .s3_credentials ,
506- {
507- "access-key" : "ak" ,
508- "secret-key" : "sk" ,
509- "bucket" : "test-bucket" ,
510- "endpoint" : "https://s3.example" ,
511- },
511+ expected_credentials ,
512512 )
513- mock_add_s3_credentials .assert_called_once_with (harness . charm . _stored . s3_credentials )
513+ mock_add_s3_credentials .assert_called_once_with (expected_credentials )
514514 self .assertIsInstance (harness .charm .unit .status , MaintenanceStatus )
515515
516516 @patch (
@@ -532,7 +532,7 @@ def test_s3_relation_credentials_changed_failure_sets_blocked(self, _mock_add):
532532
533533 self .assertEqual (
534534 harness .charm ._stored .s3_credentials ,
535- {"access-key " : "ak" , "secret-key " : "sk" , "bucket " : "test-bucket" },
535+ {"access_key " : "ak" , "secret_key " : "sk" , "endpoint " : None },
536536 )
537537 self .assertIsInstance (harness .charm .unit .status , BlockedStatus )
538538 self .assertIn (
@@ -570,7 +570,10 @@ def test_s3_relation_credentials_updated(self, mock_add_s3_credentials):
570570 "s3-integrator" ,
571571 {"access-key" : "ak" , "secret-key" : "sk" , "bucket" : "test-bucket" },
572572 )
573- self .assertEqual (harness .charm ._stored .s3_credentials ["access-key" ], "ak" )
573+ self .assertEqual (
574+ harness .charm ._stored .s3_credentials ,
575+ {"access_key" : "ak" , "secret_key" : "sk" , "endpoint" : None },
576+ )
574577
575578 harness .update_relation_data (
576579 relation_id ,
@@ -579,10 +582,10 @@ def test_s3_relation_credentials_updated(self, mock_add_s3_credentials):
579582 )
580583 self .assertEqual (
581584 harness .charm ._stored .s3_credentials ,
582- {"access-key " : "ak2" , "secret-key " : "sk2" , "bucket " : "test-bucket" },
585+ {"access_key " : "ak2" , "secret_key " : "sk2" , "endpoint " : None },
583586 )
584587 mock_add_s3_credentials .assert_called_with (
585- {"access-key " : "ak2" , "secret-key " : "sk2" , "bucket " : "test-bucket" }
588+ {"access_key " : "ak2" , "secret_key " : "sk2" , "endpoint " : None }
586589 )
587590 self .assertIsInstance (harness .charm .unit .status , MaintenanceStatus )
588591
@@ -612,7 +615,8 @@ def test_s3_relation_credentials_gone(self, *_):
612615 "s3-integrator" ,
613616 {"access-key" : "ak" , "secret-key" : "sk" },
614617 )
615- self .assertIn ("access-key" , harness .charm ._stored .s3_credentials )
618+ expected_credentials = {"access_key" : "ak" , "secret_key" : "sk" , "endpoint" : None }
619+ self .assertEqual (harness .charm ._stored .s3_credentials , expected_credentials )
616620
617621 harness .remove_relation (relation_id )
618622 self .assertEqual (harness .charm ._stored .s3_credentials , {})
@@ -654,7 +658,10 @@ def test_s3_relation_credentials_gone_failure_sets_blocked(self, _mock_remove):
654658
655659 harness .remove_relation (relation_id )
656660
657- self .assertEqual (harness .charm ._stored .s3_credentials , {})
661+ self .assertEqual (
662+ harness .charm ._stored .s3_credentials ,
663+ {"access_key" : "ak" , "secret_key" : "sk" , "endpoint" : None },
664+ )
658665 self .assertIsInstance (harness .charm .unit .status , BlockedStatus )
659666 self .assertIn ("failed to remove s3 credentials" , harness .charm .unit .status .message )
660667
0 commit comments