def availability_policy(self, product, stockrecord):
""" A product is unavailable for non-admin users if the current date is
beyond the product's expiration date. Products are always available for admin users.
"""
is_staff = getattr(self.user, 'is_staff', False)
is_available = product.expires is None or (product.expires >= timezone.now())
if is_staff or is_available:
return super(CourseSeatAvailabilityPolicyMixin, self).availability_policy(product, stockrecord)
else:
return availability.Unavailable()
def availability_policy(self, product, stockrecord):
if product == new_product:
return availability.Unavailable()
return super().availability_policy(product, stockrecord)
def setUp(self):
self.availability = availability.Unavailable()
