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 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
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)