def test_get_current_sequence_number(self):
cs = sequence_model.get_current_sequence()
csn = sequence_model.get_current_sequence_number()
self.assertIsNone(cs)
self.assertIsNone(csn)
# test sequence in the past
now = datetime.datetime.utcnow().date()
start_date = now - datetime.timedelta(weeks=16)
signup_close_date = now - datetime.timedelta(weeks=17)
sequence_model.create_sequence(
start_date, signup_close_date
)
cs = sequence_model.get_current_sequence()
csn = sequence_model.get_current_sequence_number()
self.assertIsNone(cs)
self.assertIsNone(csn)
start_date = now + datetime.timedelta(weeks=16)
signup_close_date = now + datetime.timedelta(weeks=17)
sequence_model.create_sequence(
start_date, signup_close_date
)
cs = sequence_model.get_current_sequence()
csn = sequence_model.get_current_sequence_number()
self.assertEqual(cs['id'], csn)
def test_multiple_sequences(self):
now = datetime.datetime.utcnow().date()
start_date = now - datetime.timedelta(days=1)
signup_close_date = now - datetime.timedelta(days=1)
sequence_model.create_sequence(
start_date, signup_close_date
)
start_date = now
signup_close_date = now
sequence = sequence_model.create_sequence(
start_date, signup_close_date
)
start_date = now + datetime.timedelta(days=1)
signup_close_date = now + datetime.timedelta(days=1)
sequence_model.create_sequence(
start_date, signup_close_date
)
current_sequence = sequence_model.get_current_sequence()
self.assertEqual(sequence, current_sequence)
def test_get_all_sequences(self):
now = datetime.datetime.utcnow().date()
start_date = now - datetime.timedelta(weeks=16)
signup_close_date = now - datetime.timedelta(weeks=17)
sequence_model.create_sequence(
start_date, signup_close_date
)
start_date = now + datetime.timedelta(weeks=8)
signup_close_date = now + datetime.timedelta(weeks=7)
sequence = sequence_model.create_sequence(
start_date, signup_close_date
)
start_date = now + datetime.timedelta(weeks=16)
signup_close_date = now + datetime.timedelta(weeks=15)
sequence_model.create_sequence(
start_date, signup_close_date
)
all_sequences = sequence_model.get_all_sequences()
self.assertEqual(len(all_sequences), 3)
self.assertEqual(sequence, all_sequences[1])
def test_get_current_sequence(self):
current_sequence = sequence_model.get_current_sequence()
self.assertEquals(current_sequence, None)
now = datetime.datetime.utcnow().date()
start_date = now + datetime.timedelta(weeks=8)
signup_close_date = now + datetime.timedelta(weeks=7)
sequence = sequence_model.create_sequence(
start_date, signup_close_date
)
current_sequence = sequence_model.get_current_sequence()
self.assertEqual(sequence, current_sequence)
def test_create_sequence(self):
now = datetime.datetime.utcnow().date()
start_date = now + datetime.timedelta(weeks=8)
signup_close_date = now + datetime.timedelta(weeks=7)
sequence = sequence_model.create_sequence(
start_date, signup_close_date
)
self.assertIn('global_list', sequence)
self.assertIn('campaign_id', sequence)
self.assertEquals(sequence['start_date'], start_date)
self.assertEquals(sequence['signup_close_date'], signup_close_date)
self.assertTrue(self.create_list_patch.called)
self.assertTrue(self.create_campaign_patch.called)
