class AdminTest(TestCase):
def setUp(self):
now = datetime.now()
self.ten_days_ago = now - timedelta(days=10)
self.ten_days_later = now + timedelta(days=10)
self.feature_flag = FeatureFlag.objects.create(name='expire_flag', enabled=True)
self.feature_flag_admin = FeatureFlagAdmin(FeatureFlag, AdminSite())
def test_when_flag_expires(self):
# when the feature flag expires, make sure that the admin reflects that.
self.feature_flag.expiration_warning = self.ten_days_ago
# the _expires function will return the object with red html mock up, so these values won't be equal.
self.assertNotEqual(self.feature_flag_admin._expiration_warning(self.feature_flag), self.feature_flag)
def test_when_flag_does_not_expires(self):
# when the feature flag doesn't expire, make sure that the admin reflects that.
self.feature_flag.expiration_warning = self.ten_days_later
# the _expires function will just return the object.expires attr, so these values will be equal.
self.assertNotEqual(self.feature_flag_admin._expiration_warning(self.feature_flag), self.feature_flag)
def setUp(self):
now = datetime.now()
self.ten_days_ago = now - timedelta(days=10)
self.ten_days_later = now + timedelta(days=10)
self.feature_flag = FeatureFlag.objects.create(name='expire_flag', enabled=True)
self.feature_flag_admin = FeatureFlagAdmin(FeatureFlag, AdminSite())