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