def test_keymgr_get_with_external(self, client_for_admin): self.override_config('use_external_key_manager', True) actual_key = 'test_key_super_secret' keyref = 'test_key_reference' secret = mock.Mock(payload=actual_key) secrets_manager = mock.Mock() secrets_manager.get = mock.Mock(return_value=secret) client = mock.Mock(secrets=secrets_manager) client_for_admin.return_value = client # with external key manager, get should return a key from a reference key = keymgr.get(keyref) secrets_manager.get.assert_called_with(keyref) self.assertEqual(actual_key, key)
def test_keymgr_get_no_external(self): actual_key = 'test_key_super_secret' # with no external key manager, get should return the argument keyref = keymgr.get(actual_key) self.assertEqual(actual_key, keyref)