def test_no_delivery_after_abort(self):
"""Test that notifs queued for an aborted transaction are not sent."""
source_session = uuid.uuid4()
udf_a = 'udf-a'
udf_b = 'udf-b'
udf_c = 'udf-c'
try:
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_a, source_session)
self.notifier.queue_udf_delete(1, udf_b)
self.assertEqual([], self.notifications)
raise ValueError('Foo')
except ValueError:
# should not send any notifications
self.assertEqual([], self.notifications)
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_c)
self.assertEqual([], self.notifications)
# make sure we don't re-send from previous transactions
event_c = self.notifications[0]
self.assertEqual(event_c.udf_id, udf_c)
self.assertEqual(event_c.source_session, None)
def test_no_delivery_after_abort(self):
"""Test that notifs queued for an aborted transaction are not sent."""
source_session = uuid.uuid4()
udf_a = 'udf-a'
udf_b = 'udf-b'
udf_c = 'udf-c'
try:
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_a, source_session)
self.notifier.queue_udf_delete(1, udf_b)
self.assertEqual([], self.notifications)
raise ValueError('Foo')
except ValueError:
# should not send any notifications
self.assertEqual([], self.notifications)
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_c)
self.assertEqual([], self.notifications)
# make sure we don't re-send from previous transactions
event_c = self.notifications[0]
self.assertEqual(event_c.udf_id, udf_c)
self.assertEqual(event_c.source_session, None)
def test_deliver_after_commit(self):
"""Test that notifications are delivered after committing."""
source_session = uuid.uuid4()
udf_a = uuid.uuid4()
udf_b = uuid.uuid4()
udf_c = uuid.uuid4()
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_a, source_session)
self.notifier.queue_udf_delete(1, udf_b)
self.assertEqual([], self.notifications)
# should have sent notifications after commit
event_a = self.notifications[0]
event_b = self.notifications[1]
self.assertEqual(event_a.udf_id, udf_a)
self.assertEqual(event_a.source_session, source_session)
self.assertEqual(event_b.udf_id, udf_b)
self.assertEqual(event_b.source_session, None)
self.notifications[:] = []
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_c)
self.assertEqual([], self.notifications)
# make sure we don't re-send from previous transactions
event_c = self.notifications[0]
self.assertEqual(event_c.udf_id, udf_c)
self.assertEqual(event_c.source_session, None)
def test_deliver_after_commit(self):
"""Test that notifications are delivered after committing."""
source_session = uuid.uuid4()
udf_a = uuid.uuid4()
udf_b = uuid.uuid4()
udf_c = uuid.uuid4()
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_a, source_session)
self.notifier.queue_udf_delete(1, udf_b)
self.assertEqual([], self.notifications)
# should have sent notifications after commit
event_a = self.notifications[0]
event_b = self.notifications[1]
self.assertEqual(event_a.udf_id, udf_a)
self.assertEqual(event_a.source_session, source_session)
self.assertEqual(event_b.udf_id, udf_b)
self.assertEqual(event_b.source_session, None)
self.notifications[:] = []
with fsync_commit():
self.notifier.queue_udf_delete(1, udf_c)
self.assertEqual([], self.notifications)
# make sure we don't re-send from previous transactions
event_c = self.notifications[0]
self.assertEqual(event_c.udf_id, udf_c)
self.assertEqual(event_c.source_session, None)
def test_broadcast_udf_delete(self):
"""Test broadcast of an udf creation."""
udf_id = uuid.uuid4()
with fsync_commit():
self.notifier.queue_udf_delete(0, udf_id)
self.assertEqual(self.notifications,
[UDFDelete(0, udf_id, None)])
def test_broadcast_share_declined(self):
"""Test broadcast of an share declined."""
share = FakeShare()
with fsync_commit():
self.notifier.queue_share_declined(share)
self.assertEqual(self.notifications,
[ShareDeclined(*share.event_args)])
def test_broadcast_share_declined(self):
"""Test broadcast of an share declined."""
share = FakeShare()
with fsync_commit():
self.notifier.queue_share_declined(share)
self.assertEqual(self.notifications,
[ShareDeclined(*share.event_args)])
def test_broadcast_udf_create(self):
"""Test broadcast of an udf creation."""
udf_id = uuid.uuid4()
root_id = uuid.uuid4()
suggested_path = "foo"
with fsync_commit():
self.notifier.queue_udf_create(0, udf_id, root_id, suggested_path)
self.assertEqual(self.notifications,
[UDFCreate(0, udf_id, root_id, suggested_path, None)])
def test_broadcast_udf_create(self):
"""Test broadcast of an udf creation."""
udf_id = uuid.uuid4()
root_id = uuid.uuid4()
suggested_path = "foo"
with fsync_commit():
self.notifier.queue_udf_create(0, udf_id, root_id, suggested_path)
self.assertEqual(self.notifications,
[UDFCreate(0, udf_id, root_id, suggested_path, None)])
def test_broadcast_vol_new_gen(self):
"""Test the broadcast of a new generation for the volume."""
user_id = 1
volume_id = uuid.uuid4()
new_gen = 77
with fsync_commit():
self.notifier.queue_volume_new_generation(
user_id, volume_id, new_gen)
self.assertEqual(self.notifications,
[VolumeNewGeneration(user_id, volume_id,
new_gen, None)])
def test_broadcast_vol_new_gen(self):
"""Test the broadcast of a new generation for the volume."""
user_id = 1
volume_id = uuid.uuid4()
new_gen = 77
with fsync_commit():
self.notifier.queue_volume_new_generation(user_id, volume_id,
new_gen)
self.assertEqual(
self.notifications,
[VolumeNewGeneration(user_id, volume_id, new_gen, None)])
def test_broadcast_udf_delete(self):
"""Test broadcast of an udf creation."""
udf_id = uuid.uuid4()
with fsync_commit():
self.notifier.queue_udf_delete(0, udf_id)
self.assertEqual(self.notifications, [UDFDelete(0, udf_id, None)])
