def METHOD_NAME(self):
    self.protocol.writeMessageBegin("name", "type", 1)
    self.mock_wrapped_protocol.writeMessageBegin.assert_called_with("name",
                                                                    "type",
                                                                    1)