def testCleanExperiment(self):
experiment = Experiment(created_by=self.user)
experiment.save()
eh = EmbargoHandler(experiment.id)
self.assertTrue(eh.never_expires())
self.assertFalse(eh.has_any_expiry())
self.assertFalse(eh.can_be_defaulted())
self.assertTrue(eh.because_no_end_date())
self.assertEqual(None, eh.get_expiry_date())
self.assertRaises(Exception, eh.prevent_expiry)
eh.reset_to_default() # should not raise Exception
self.assertRaises(Exception, eh.set_expiry)
def testResetToDefault(self):
expiry = dt.datetime(2034,12,4)
expiry_as_string = '2034/12/04'
experiment = Experiment(created_by=self.user)
experiment.save()
eh = EmbargoHandler(experiment.id, True)
eh.set_expiry(expiry_as_string)
self.assertEqual(expiry, eh.get_expiry_date())
eh = EmbargoHandler(experiment.id, True)
eh.reset_to_default()
self.assertEqual(None, eh.get_expiry_date())
def testSetExpiry(self):
expiry = dt.datetime(2034, 12, 4)
expiry_as_string = '2034/12/04'
experiment = Experiment(created_by=self.user)
experiment.save()
eh = EmbargoHandler(experiment.id)
self.assertTrue(eh.never_expires())
self.assertEqual(None, eh.get_expiry_date())
eh = EmbargoHandler(experiment.id, True)
eh.set_expiry(expiry_as_string)
eh = EmbargoHandler(experiment.id)
self.assertEqual(expiry, eh.get_expiry_date())
def testCleanExperimentWithEndDate(self):
experiment = Experiment(created_by=self.user, end_time=dt.datetime(2011, 12, 31))
experiment.save()
expected_expiry = experiment.end_time + dt.timedelta(settings.EMBARGO_DAYS)
eh = EmbargoHandler(experiment.id)
self.assertFalse(eh.never_expires())
self.assertTrue(eh.has_any_expiry())
self.assertFalse(eh.can_be_defaulted())
self.assertFalse(eh.because_no_end_date())
self.assertEqual(expected_expiry, eh.get_expiry_date())
self.assertRaises(Exception, eh.prevent_expiry)
eh.reset_to_default() # should not raise Exception
self.assertRaises(Exception, eh.set_expiry)
def testNeverExpire(self):
experiment = Experiment(created_by=self.user, end_time=dt.datetime(1999,4,3))
experiment.save()
eh = EmbargoHandler(experiment.id, True)
self.assertFalse(eh.never_expires())
eh.prevent_expiry()
self.assertTrue(eh.never_expires())
self.assertFalse(eh.because_no_end_date())
def embargo_edit(experiment_id):
handler = EmbargoHandler(experiment_id)
inclusion_context = {"experiment_id": experiment_id}
if handler.never_expires():
inclusion_context["never_expires"] = True
inclusion_context["because_no_end_date"] = handler.because_no_end_date()
elif handler.has_any_expiry():
inclusion_context["has_any_expiry"] = True
inclusion_context["expiry_date"] = handler.get_expiry_date().strftime("%Y/%m/%d")
else:
raise Exception("unknown state - should never get here")
inclusion_context["can_be_defaulted"] = handler.can_be_defaulted()
return inclusion_context
