Function TestCancel

pkg/streamclient/stream_test.go:195–226

(t *testing.T)

}

func TestCancel(t *testing.T) {
	transport := newFakeTransport()

	reader := NewReader("1", 1024, transport)
	writer := NewWriter("1", 1024, transport)

	go func() {
		for dataPk := range transport.dataChan {
			reader.RecvData(dataPk)
		}
	}()

	go func() {
		for ackPk := range transport.ackChan {
			writer.RecvAck(ackPk)
		}
	}()

	reader.Close()

	select {
	case <-writer.GetCanceledChan():
		// Success
	case <-time.After(1 * time.Second):
		t.Fatal("Writer not notified of cancellation")
	}

	_, _, canceled := writer.GetAckState()
	if !canceled {
		t.Fatal("Writer should be in canceled state")
	}
}

func TestMultipleWrites(t *testing.T) {
	transport := newFakeTransport()