package volume_server_http_test

import (
	"net/http"
	"strings"
	"testing"

	"github.com/seaweedfs/seaweedfs/test/volume_server/framework"
	"github.com/seaweedfs/seaweedfs/test/volume_server/matrix"
)

func TestMultiRangeReadReturnsMultipartPayload(t *testing.T) {
	if testing.Short() {
		t.Skip("skipping integration test in short mode")
	}

	clusterHarness := framework.StartSingleVolumeCluster(t, matrix.P1())
	conn, grpcClient := framework.DialVolumeServer(t, clusterHarness.VolumeGRPCAddress())
	defer conn.Close()

	const volumeID = uint32(97)
	framework.AllocateVolume(t, grpcClient, volumeID, "")

	client := framework.NewHTTPClient()
	fid := framework.NewFileID(volumeID, 771999, 0x0A1B2C3D)
	payload := []byte("0123456789abcdef")
	uploadResp := framework.UploadBytes(t, client, clusterHarness.VolumeAdminURL(), fid, payload)
	_ = framework.ReadAllAndClose(t, uploadResp)
	if uploadResp.StatusCode != http.StatusCreated {
		t.Fatalf("upload expected 201, got %d", uploadResp.StatusCode)
	}

	multiRangeReq := mustNewRequest(t, http.MethodGet, clusterHarness.VolumeAdminURL()+"/"+fid)
	multiRangeReq.Header.Set("Range", "bytes=0-1,4-5")
	multiRangeResp := framework.DoRequest(t, client, multiRangeReq)
	multiRangeBody := framework.ReadAllAndClose(t, multiRangeResp)
	if multiRangeResp.StatusCode != http.StatusPartialContent {
		t.Fatalf("multi-range expected 206, got %d", multiRangeResp.StatusCode)
	}
	if !strings.Contains(multiRangeResp.Header.Get("Content-Type"), "multipart/byteranges") {
		t.Fatalf("multi-range content-type mismatch: %q", multiRangeResp.Header.Get("Content-Type"))
	}

	bodyText := string(multiRangeBody)
	if !strings.Contains(bodyText, "01") || !strings.Contains(bodyText, "45") {
		t.Fatalf("multi-range body missing expected segments: %q", bodyText)
	}
}

func TestOversizedCombinedRangesAreIgnored(t *testing.T) {
	if testing.Short() {
		t.Skip("skipping integration test in short mode")
	}

	clusterHarness := framework.StartSingleVolumeCluster(t, matrix.P1())
	conn, grpcClient := framework.DialVolumeServer(t, clusterHarness.VolumeGRPCAddress())
	defer conn.Close()

	const volumeID = uint32(100)
	framework.AllocateVolume(t, grpcClient, volumeID, "")

	client := framework.NewHTTPClient()
	fid := framework.NewFileID(volumeID, 772003, 0x3C4D5E6F)
	payload := []byte("0123456789abcdef")
	uploadResp := framework.UploadBytes(t, client, clusterHarness.VolumeAdminURL(), fid, payload)
	_ = framework.ReadAllAndClose(t, uploadResp)
	if uploadResp.StatusCode != http.StatusCreated {
		t.Fatalf("upload expected 201, got %d", uploadResp.StatusCode)
	}

	// Range bytes sum is 22 (> payload size 16), which exercises the oversized-range guard path.
	oversizedRangeReq := mustNewRequest(t, http.MethodGet, clusterHarness.VolumeAdminURL()+"/"+fid)
	oversizedRangeReq.Header.Set("Range", "bytes=0-10,5-15")
	oversizedRangeResp := framework.DoRequest(t, client, oversizedRangeReq)
	oversizedRangeBody := framework.ReadAllAndClose(t, oversizedRangeResp)
	if oversizedRangeResp.StatusCode != http.StatusOK {
		t.Fatalf("oversized combined range expected 200, got %d", oversizedRangeResp.StatusCode)
	}
	if len(oversizedRangeBody) != 0 {
		t.Fatalf("oversized combined range expected empty body, got %d bytes", len(oversizedRangeBody))
	}
}