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