From de6f19c43f1a92b888a69269b9d4955e42d497ee Mon Sep 17 00:00:00 2001 From: Chris Veilleux Date: Tue, 11 Jun 2019 14:46:50 -0500 Subject: [PATCH] added call to add_membership --- shared/selene/testing/account.py | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/shared/selene/testing/account.py b/shared/selene/testing/account.py index 239a6e66..e43fa313 100644 --- a/shared/selene/testing/account.py +++ b/shared/selene/testing/account.py @@ -24,10 +24,21 @@ def build_test_account(**overrides): AccountAgreement(type=PRIVACY_POLICY, accept_date=date.today()), AccountAgreement(type=TERMS_OF_USE, accept_date=date.today()) ] + if 'membership' in overrides: + membership_overrides = overrides['membership'] + if membership_overrides is None: + # if the membership override is set to None, the assumed intent + # is to add an account that does not have a membership + test_membership = None + else: + test_membership = build_test_membership(**membership_overrides) + else: + test_membership = build_test_membership() + return Account( email_address=overrides.get('email_address') or 'foo@mycroft.ai', username=overrides.get('username') or 'foobar', - membership=build_test_membership(**overrides), + membership=test_membership, agreements=overrides.get('agreements') or test_agreements ) @@ -36,6 +47,7 @@ def add_account(db, **overrides): acct_repository = AccountRepository(db) account = build_test_account(**overrides) account.id = acct_repository.add(account, 'foo') + acct_repository.add_membership(account.id, account.membership) return account