Function setupBrokerPair

pkg/streamclient/streambroker_test.go:27–65

()

}

func setupBrokerPair() (*Broker, *Broker) {
	rpc1 := &mockRpcInterface{
		dataChan: make(chan wshrpc.CommandStreamData, 10),
		ackChan:  make(chan wshrpc.CommandStreamAckData, 10),
	}
	rpc2 := &mockRpcInterface{
		dataChan: make(chan wshrpc.CommandStreamData, 10),
		ackChan:  make(chan wshrpc.CommandStreamAckData, 10),
	}

	broker1 := NewBroker(rpc1)
	broker2 := NewBroker(rpc2)

	go func() {
		for data := range rpc1.dataChan {
			broker2.RecvData(data)
		}
	}()

	go func() {
		for ack := range rpc1.ackChan {
			broker2.RecvAck(ack)
		}
	}()

	go func() {
		for data := range rpc2.dataChan {
			broker1.RecvData(data)
		}
	}()

	go func() {
		for ack := range rpc2.ackChan {
			broker1.RecvAck(ack)
		}
	}()

	return broker1, broker2
}

func TestBrokerBasicReadWrite(t *testing.T) {
	broker1, broker2 := setupBrokerPair()