Function newMockUploadStream

coderd/provisionerdserver/upload_file_test.go:179–193

(up *sdkproto.DataUpload, chunks ...*sdkproto.ChunkPiece)

}

func newMockUploadStream(up *sdkproto.DataUpload, chunks ...*sdkproto.ChunkPiece) *mockUploadStream {
	stream := &mockUploadStream{
		done:     make(chan struct{}),
		messages: make(chan *sdkproto.FileUpload, 1+len(chunks)),
	}
	if up != nil {
		stream.messages <- &sdkproto.FileUpload{Type: &sdkproto.FileUpload_DataUpload{DataUpload: up}}
	}

	for _, chunk := range chunks {
		stream.messages <- &sdkproto.FileUpload{Type: &sdkproto.FileUpload_ChunkPiece{ChunkPiece: chunk}}
	}
	close(stream.messages)
	return stream
}