diff --git a/src/reactive/barbican_vault_handlers.py b/src/reactive/barbican_vault_handlers.py index fcd199b..df5b236 100644 --- a/src/reactive/barbican_vault_handlers.py +++ b/src/reactive/barbican_vault_handlers.py @@ -77,9 +77,8 @@ def plugin_info_barbican_publish(): # fetch current secret-id, if any, from relation with barbican principle current_secret_id = None - secrets = reactive.endpoint_from_flag('secrets.available') - if secrets: - for relation in secrets.relations: + if barbican: + for relation in barbican.relations: data = relation.to_publish.get('data') if data and data.get('approle_secret_id'): current_secret_id = data.get('approle_secret_id') diff --git a/unit_tests/test_barbican_vault_handlers.py b/unit_tests/test_barbican_vault_handlers.py index 9827bd7..944fff4 100644 --- a/unit_tests/test_barbican_vault_handlers.py +++ b/unit_tests/test_barbican_vault_handlers.py @@ -119,8 +119,7 @@ class TestBarbicanVaultHandlers(test_utils.PatchHelper): handlers.plugin_info_barbican_publish() self.endpoint_from_flag.assert_has_calls([ mock.call('endpoint.secrets.joined'), - mock.call('secrets-storage.available'), - mock.call('secrets.available'), + mock.call('secrets-storage.available') ]) vault_data = { 'approle_role_id': secrets_storage.unit_role_id,