Support types in set_keyring step

This commit is contained in:
karimpwnz 2021-01-02 04:14:38 +02:00
parent cce978b4ab
commit a699f284dc

View file

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