diff --git a/hc/payments/models.py b/hc/payments/models.py index 45b095ca..97ab1c4c 100644 --- a/hc/payments/models.py +++ b/hc/payments/models.py @@ -1,8 +1,14 @@ -import braintree - +from django.conf import settings from django.contrib.auth.models import User from django.db import models +if settings.USE_PAYMENTS: + import braintree +else: + # hc.payments tests mock this object, so tests should + # still be able to run: + braintree = None + class SubscriptionManager(models.Manager):