def test__exit__(self):
master = queue.ExclusiveResourceProcessor(FAKE_ID)
not_master = queue.ExclusiveResourceProcessor(FAKE_ID)
master.__enter__()
self.assertIn(FAKE_ID, queue.ExclusiveResourceProcessor._masters)
not_master.__enter__()
not_master.__exit__(None, None, None)
self.assertIn(FAKE_ID, queue.ExclusiveResourceProcessor._masters)
master.__exit__(None, None, None)
self.assertNotIn(FAKE_ID, queue.ExclusiveResourceProcessor._masters)
def test__exit__(self):
primary = queue.ExclusiveResourceProcessor(FAKE_ID)
not_primary = queue.ExclusiveResourceProcessor(FAKE_ID)
primary.__enter__()
self.assertIn(FAKE_ID, queue.ExclusiveResourceProcessor._primaries)
not_primary.__enter__()
not_primary.__exit__(None, None, None)
self.assertIn(FAKE_ID, queue.ExclusiveResourceProcessor._primaries)
primary.__exit__(None, None, None)
self.assertNotIn(FAKE_ID, queue.ExclusiveResourceProcessor._primaries)
def test_updates(self):
primary = queue.ExclusiveResourceProcessor(FAKE_ID)
not_primary = queue.ExclusiveResourceProcessor(FAKE_ID)
primary.queue_update(queue.ResourceUpdate(FAKE_ID, 0))
not_primary.queue_update(queue.ResourceUpdate(FAKE_ID, 0))
for update in not_primary.updates():
raise Exception("Only the primary should process a resource")
self.assertEqual(2, len([i for i in primary.updates()]))
def test_updates(self):
master = queue.ExclusiveResourceProcessor(FAKE_ID)
not_master = queue.ExclusiveResourceProcessor(FAKE_ID)
master.queue_update(queue.ResourceUpdate(FAKE_ID, 0))
not_master.queue_update(queue.ResourceUpdate(FAKE_ID, 0))
for update in not_master.updates():
raise Exception("Only the master should process a resource")
self.assertEqual(2, len([i for i in master.updates()]))
def test_primary(self):
primary = queue.ExclusiveResourceProcessor(FAKE_ID)
not_primary = queue.ExclusiveResourceProcessor(FAKE_ID)
primary_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
not_primary_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
self.assertEqual(primary, primary._primary)
self.assertEqual(primary, not_primary._primary)
self.assertEqual(primary_2, primary_2._primary)
self.assertEqual(primary_2, not_primary_2._primary)
primary.__exit__(None, None, None)
primary_2.__exit__(None, None, None)
def test_i_am_primary(self):
primary = queue.ExclusiveResourceProcessor(FAKE_ID)
not_primary = queue.ExclusiveResourceProcessor(FAKE_ID)
primary_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
not_primary_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
self.assertTrue(primary._i_am_primary())
self.assertFalse(not_primary._i_am_primary())
self.assertTrue(primary_2._i_am_primary())
self.assertFalse(not_primary_2._i_am_primary())
primary.__exit__(None, None, None)
primary_2.__exit__(None, None, None)
def test_master(self):
master = queue.ExclusiveResourceProcessor(FAKE_ID)
not_master = queue.ExclusiveResourceProcessor(FAKE_ID)
master_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
not_master_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
self.assertEqual(master, master._master)
self.assertEqual(master, not_master._master)
self.assertEqual(master_2, master_2._master)
self.assertEqual(master_2, not_master_2._master)
master.__exit__(None, None, None)
master_2.__exit__(None, None, None)
def test_i_am_master(self):
master = queue.ExclusiveResourceProcessor(FAKE_ID)
not_master = queue.ExclusiveResourceProcessor(FAKE_ID)
master_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
not_master_2 = queue.ExclusiveResourceProcessor(FAKE_ID_2)
self.assertTrue(master._i_am_master())
self.assertFalse(not_master._i_am_master())
self.assertTrue(master_2._i_am_master())
self.assertFalse(not_master_2._i_am_master())
master.__exit__(None, None, None)
master_2.__exit__(None, None, None)
def test_data_fetched_since(self):
primary = queue.ExclusiveResourceProcessor(FAKE_ID)
self.assertEqual(datetime.datetime.min,
primary._get_resource_data_timestamp())
ts1 = datetime.datetime.utcnow() - datetime.timedelta(seconds=10)
ts2 = datetime.datetime.utcnow()
primary.fetched_and_processed(ts2)
self.assertEqual(ts2, primary._get_resource_data_timestamp())
primary.fetched_and_processed(ts1)
self.assertEqual(ts2, primary._get_resource_data_timestamp())
primary.__exit__(None, None, None)