def testInsertMessage(self):
t = FakeStub()
s = client.OutgoingConnection(None, 'test', t)
s.InsertMessage(common_pb2.Message())
self.assertTrue(t.insert_done)
self.assertFalse(t.insert_errors)
self.assertEqual(t.message.source.service_name, 'test')
def testKeepAlive(self):
t = FakeStub()
s = client.OutgoingConnection(None, 'test', t)
self.assertTrue(t.event.wait(10))
s.Shutdown()
def testDeletePendingMessages(self): t = FakeStub() s = client.OutgoingConnection(None, 'test', t) s.DeletePendingMessages(admin_pb2.DeletePendingMessagesRequest()) self.assertTrue(t.delete_done) self.assertFalse(t.delete_errors)