From a699f284dcb11db3dccd7c4f7a848acad2f1696a Mon Sep 17 00:00:00 2001 From: karimpwnz Date: Sat, 2 Jan 2021 04:14:38 +0200 Subject: [PATCH] Support types in set_keyring step --- features/steps/core.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/features/steps/core.py b/features/steps/core.py index 56d285f2..9cd2ffc5 100644 --- a/features/steps/core.py +++ b/features/steps/core.py @@ -164,10 +164,14 @@ def use_password(context, password, num=1): context.password = iter([password] * num) -@given("we have a keyring") -def set_keyring(context): - keyring.set_keyring(TestKeyring()) - +@given("we have a {type} keyring") +def set_keyring(context, type=None): + if type == "locked": + keyring.set_keyring(LockedKeyring()) + elif type == "failed": + keyring.set_keyring(FailedKeyring()) + else: + keyring.set_keyring(TestKeyring()) @given("we do not have a keyring") def disable_keyring(context):