Function compareBinaryStream

pkg/util.go:1309–1375

(expectedResp models.HTTPResp, stream io.Reader, logger *zap.Logger)

}

func compareBinaryStream(expectedResp models.HTTPResp, stream io.Reader, logger *zap.Logger) (bool, string, *StreamMismatchInfo, error) {
	expectedBytes := expectedBinaryBytes(expectedResp)
	expectedSize := len(expectedBytes)
	actualSize := 0
	contentMatch := true
	mismatchOffset := -1
	buffer := make([]byte, 32*1024)

	for {
		n, err := stream.Read(buffer)
		if n > 0 {
			if contentMatch && actualSize < expectedSize {
				end := actualSize + n
				if end > expectedSize {
					end = expectedSize
				}
				if !bytes.Equal(buffer[:end-actualSize], expectedBytes[actualSize:end]) {
					contentMatch = false
					mismatchOffset = actualSize
				}
			}
			actualSize += n
			if actualSize >= expectedSize {
				if actualSize > expectedSize {
					logger.Debug("received additional binary stream data after expected size was matched; closing stream capture",
						zap.Int("expected_size", expectedSize),
						zap.Int("actual_size", actualSize))
				}
				break
			}
		}
		if err == io.EOF {
			break
		}
		if err != nil {
			return false, strconv.Itoa(actualSize), nil, err
		}
	}

	if actualSize != expectedSize {
		logger.Debug("binary stream size mismatch",
			zap.Int("expected_size", expectedSize),
			zap.Int("actual_size", actualSize))
		mismatchInfo := &StreamMismatchInfo{
			FrameIndex:    0,
			ExpectedFrame: fmt.Sprintf("%d bytes", expectedSize),
			ActualFrame:   fmt.Sprintf("%d bytes", actualSize),
			Reason:        fmt.Sprintf("size mismatch: expected %d bytes, got %d bytes", expectedSize, actualSize),
		}
		return false, strconv.Itoa(actualSize), mismatchInfo, nil
	}

	if !contentMatch {
		logger.Debug("binary stream content mismatch",
			zap.Int("size", actualSize),
			zap.Int("first_mismatch_offset", mismatchOffset))
		mismatchInfo := &StreamMismatchInfo{
			FrameIndex:    0,