def test_with_notification(self):
context = trove_testtools.TroveTestContext(self)
self.assertTrue(context.notification.needs_end_notification)
with notification.NotificationCastWrapper(context, "foo"):
self.assertEqual('foo', context.notification.server_type)
self.assertEqual('api', context.notification.server_type)
self.assertFalse(context.notification.needs_end_notification)
def test_no_notification(self):
with notification.NotificationCastWrapper(TroveContext(), "foo"):
pass