added call to add_membership

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