forked from mirrors/pkg-proxy
599fe9e254* Fix all golangci-lint issues across the codebase Resolve 77 lint issues reported by golangci-lint with gocritic, gocognit, gocyclo, maintidx, dupl, mnd, unparam, ireturn, goconst, and errcheck enabled. Net reduction of ~175 lines through shared helpers and deduplication. * Suppress staticcheck SA1019 for intentional deprecated field usage The Storage.Path field is deprecated but still read for backwards compatibility with existing configs that haven't migrated to the URL field.
90 lines
2.5 KiB
Go
90 lines
2.5 KiB
Go
package storage
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestArtifactPath(t *testing.T) {
|
|
tests := []struct {
|
|
ecosystem string
|
|
namespace string
|
|
name string
|
|
version string
|
|
filename string
|
|
want string
|
|
}{
|
|
{"npm", "", "lodash", "4.17.21", "lodash-4.17.21.tgz", "npm/lodash/4.17.21/lodash-4.17.21.tgz"},
|
|
{"npm", "babel", "core", "7.0.0", "core-7.0.0.tgz", "npm/babel/core/7.0.0/core-7.0.0.tgz"},
|
|
{"cargo", "", "serde", "1.0.0", "serde-1.0.0.crate", "cargo/serde/1.0.0/serde-1.0.0.crate"},
|
|
{"pypi", "", "requests", "2.28.0", "requests-2.28.0.tar.gz", "pypi/requests/2.28.0/requests-2.28.0.tar.gz"},
|
|
{"maven", "org.apache", "commons-lang3", "3.12.0", "commons-lang3-3.12.0.jar", "maven/org.apache/commons-lang3/3.12.0/commons-lang3-3.12.0.jar"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
got := ArtifactPath(tt.ecosystem, tt.namespace, tt.name, tt.version, tt.filename)
|
|
if got != tt.want {
|
|
t.Errorf("ArtifactPath(%q, %q, %q, %q, %q) = %q, want %q",
|
|
tt.ecosystem, tt.namespace, tt.name, tt.version, tt.filename, got, tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHashingReader(t *testing.T) {
|
|
content := "hello world"
|
|
r := NewHashingReader(strings.NewReader(content))
|
|
|
|
data, err := io.ReadAll(r)
|
|
if err != nil {
|
|
t.Fatalf("ReadAll failed: %v", err)
|
|
}
|
|
|
|
if string(data) != content {
|
|
t.Errorf("got content %q, want %q", string(data), content)
|
|
}
|
|
|
|
if r.Size() != int64(len(content)) {
|
|
t.Errorf("got size %d, want %d", r.Size(), len(content))
|
|
}
|
|
|
|
h := sha256.Sum256([]byte(content))
|
|
wantHash := hex.EncodeToString(h[:])
|
|
if r.Sum() != wantHash {
|
|
t.Errorf("got hash %s, want %s", r.Sum(), wantHash)
|
|
}
|
|
}
|
|
|
|
// assertLargeFileRoundTrip stores a 1MB file in the given storage, verifies size and
|
|
// hash, then reads it back and confirms the content matches.
|
|
func assertLargeFileRoundTrip(t *testing.T, s Storage) {
|
|
t.Helper()
|
|
ctx := context.Background()
|
|
|
|
data := bytes.Repeat([]byte("x"), 1024*1024)
|
|
|
|
size, hash, err := s.Store(ctx, "large/file.bin", bytes.NewReader(data))
|
|
if err != nil {
|
|
t.Fatalf("Store large file failed: %v", err)
|
|
}
|
|
if size != int64(len(data)) {
|
|
t.Errorf("size = %d, want %d", size, len(data))
|
|
}
|
|
|
|
h := sha256.Sum256(data)
|
|
wantHash := hex.EncodeToString(h[:])
|
|
if hash != wantHash {
|
|
t.Errorf("hash mismatch for large file")
|
|
}
|
|
|
|
r, _ := s.Open(ctx, "large/file.bin")
|
|
defer func() { _ = r.Close() }()
|
|
readBack, _ := io.ReadAll(r)
|
|
if !bytes.Equal(readBack, data) {
|
|
t.Error("large file content mismatch")
|
|
}
|
|
}
|