

def _prepareSinkStack(self):
    self.sink_stack = MockSinkStack()

    def save_message(sink_stack, context, stream, msg):
        self.return_message = msg
        self.return_stream = stream
    terminator_sink = MockSink({
        SinkProperties.Endpoint: None,
    })
    terminator_sink.ProcessResponse = save_message
    self.sink_stack.Push(terminator_sink)

dataset/ETHPy150Open steveniemitz/scales/test/scales/util/base.py SinkTestCase._prepareSinkStack/original