test: stewardship with erasure encoding

Author: acha-bill

pkg/api/stewardship_test.go

@@ -5,18 +5,27 @@
 package api_test
 
 import (
+	"bytes"
+	"context"
 	"encoding/hex"
+	"fmt"
 	"net/http"
+	"strconv"
 	"testing"
+	"time"
 
 	"github.com/ethersphere/bee/v2/pkg/api"
+	"github.com/ethersphere/bee/v2/pkg/file/redundancy"
 	"github.com/ethersphere/bee/v2/pkg/jsonhttp"
 	"github.com/ethersphere/bee/v2/pkg/jsonhttp/jsonhttptest"
 	"github.com/ethersphere/bee/v2/pkg/log"
 	mockpost "github.com/ethersphere/bee/v2/pkg/postage/mock"
+	"github.com/ethersphere/bee/v2/pkg/steward"
 	"github.com/ethersphere/bee/v2/pkg/steward/mock"
+	"github.com/ethersphere/bee/v2/pkg/storage"
 	mockstorer "github.com/ethersphere/bee/v2/pkg/storer/mock"
 	"github.com/ethersphere/bee/v2/pkg/swarm"
+	"gitlab.com/nolash/go-mockbytes"
 )
 
 // nolint:paralleltest
@@ -60,6 +69,58 @@ func TestStewardship(t *testing.T) {
 	})
 }
 
+type localRetriever struct {
+	getter storage.Getter
+}
+
+func (lr *localRetriever) RetrieveChunk(ctx context.Context, addr, sourceAddr swarm.Address) (chunk swarm.Chunk, err error) {
+	ch, err := lr.getter.Get(ctx, addr)
+	if err != nil {
+		return nil, fmt.Errorf("retrieve chunk %s: %w", addr, err)
+	}
+	return ch, nil
+}
+
+func TestStewardshipWithRedundancy(t *testing.T) {
+	t.Parallel()
+
+	var (
+		storerMock      = mockstorer.New()
+		localRetrieval  = &localRetriever{getter: storerMock.ChunkStore()}
+		s               = steward.New(storerMock, localRetrieval, storerMock.Cache())
+		client, _, _, _ = newTestServer(t, testServerOptions{
+			Storer:  storerMock,
+			Logger:  log.Noop,
+			Steward: s,
+			Post:    mockpost.New(mockpost.WithAcceptAll()),
+		})
+	)
+
+	g := mockbytes.New(0, mockbytes.MockTypeStandard).WithModulus(255)
+	content, err := g.SequentialBytes(512000) // 500KB
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	for _, l := range []redundancy.Level{redundancy.NONE, redundancy.MEDIUM, redundancy.STRONG, redundancy.INSANE, redundancy.PARANOID} {
+		t.Run(fmt.Sprintf("rLevel-%d", l), func(t *testing.T) {
+			res := new(api.BytesPostResponse)
+			jsonhttptest.Request(t, client, http.MethodPost, "/bytes", http.StatusCreated,
+				jsonhttptest.WithRequestHeader(api.SwarmDeferredUploadHeader, "true"),
+				jsonhttptest.WithRequestHeader(api.SwarmPostageBatchIdHeader, batchOkStr),
+				jsonhttptest.WithRequestHeader(api.SwarmRedundancyLevelHeader, strconv.Itoa(int(l))),
+				jsonhttptest.WithRequestBody(bytes.NewReader(content)),
+				jsonhttptest.WithUnmarshalJSONResponse(res),
+			)
+
+			time.Sleep(2 * time.Second)
+			jsonhttptest.Request(t, client, http.MethodGet, "/stewardship/"+res.Reference.String(), http.StatusOK,
+				jsonhttptest.WithExpectedJSONResponse(api.IsRetrievableResponse{IsRetrievable: true}),
+			)
+		})
+	}
+}
+
 func TestStewardshipInvalidInputs(t *testing.T) {
 	t.Parallel()