mrleo1nid
7d62531c02
ci / mobile / mobile (Metro-бандл Android) (push) Successful in 1m30s
ci / ts / ts (packages + web) (push) Canceled after 4m13s
ci / module boundaries / boundaries (push) Successful in 8s
ci / api / api (push) Successful in 56s
ci / contract / contract (сгенерированный код совпадает с openapi.yaml) (push) Successful in 1m1s
ci / db / db (push) Successful in 1m7s
58 lines
2.0 KiB
Go
58 lines
2.0 KiB
Go
package uploads
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"pipu/api/internal/api/gen"
|
|
"pipu/api/internal/platform/sign"
|
|
)
|
|
|
|
func TestPrivateUploadStreamIsServedByAPIWithRange(t *testing.T) {
|
|
file := filepath.Join(t.TempDir(), "audio")
|
|
require.NoError(t, os.WriteFile(file, []byte("fLaC0123456789"), 0600))
|
|
req := httptest.NewRequest(http.MethodGet, "/audio", nil)
|
|
req.Header.Set("Range", "bytes=4-7")
|
|
rec := httptest.NewRecorder()
|
|
require.NoError(t, (uploadStream{req: req, file: file, format: "flac"}).VisitStreamUploadResponse(rec))
|
|
require.Equal(t, http.StatusPartialContent, rec.Code)
|
|
require.Equal(t, "audio/flac", rec.Header().Get("Content-Type"))
|
|
require.Equal(t, "bytes 4-7/14", rec.Header().Get("Content-Range"))
|
|
require.Equal(t, "0123", rec.Body.String())
|
|
require.Empty(t, rec.Header().Get("X-Accel-Redirect"))
|
|
|
|
rec = httptest.NewRecorder()
|
|
require.NoError(t, (uploadStream{req: httptest.NewRequest(http.MethodGet, "/audio", nil), file: file, format: "flac"}).VisitStreamUploadResponse(rec))
|
|
require.Equal(t, http.StatusOK, rec.Code)
|
|
require.Empty(t, rec.Header().Get("X-Accel-Redirect"))
|
|
require.Equal(t, "fLaC0123456789", rec.Body.String())
|
|
require.Equal(t, "audio/flac", rec.Header().Get("Content-Type"))
|
|
}
|
|
|
|
func TestUploadStreamRejectsInvalidAndExpiredLinksBeforeReadingStaging(t *testing.T) {
|
|
signer, err := sign.New("")
|
|
require.NoError(t, err)
|
|
h := NewHandler(nil, signer)
|
|
id := uuid.New()
|
|
for _, expired := range []bool{false, true} {
|
|
exp := time.Now().Add(time.Hour).Unix()
|
|
sig := "invalid"
|
|
if expired {
|
|
exp = time.Now().Add(-time.Hour).Unix()
|
|
sig = signer.Sign("upload", id.String(), strconv.FormatInt(exp, 10))
|
|
}
|
|
res, err := h.StreamUpload(context.Background(), gen.StreamUploadRequestObject{Id: id, Params: gen.StreamUploadParams{Exp: exp, Sig: sig}})
|
|
require.Nil(t, res)
|
|
require.ErrorContains(t, err, "link is invalid or expired")
|
|
}
|
|
}
|