unkinben
428c6d0e97
First increment toward 90% core-package coverage. Adds Docker-free unit tests taking these packages to full or near-full coverage: - provider/npm, provider/alpine, provider/puppet: 100% - provider/pypi: index generation via a fake FileStore, upload validation, name parsing, classification, rewrite - config: defaults, overrides, DSN, invalid port - auth: basic header with/without credentials Infra-backed packages (database, storage, cache, proxy engine, api handlers, server) still need the testcontainers batch to reach 90%.
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package puppet
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.unkin.net/unkin/artifactapi/internal/provider"
|
|
"git.unkin.net/unkin/artifactapi/pkg/models"
|
|
)
|
|
|
|
func TestType(t *testing.T) {
|
|
if (&Provider{}).Type() != models.PackagePuppet {
|
|
t.Fatal("wrong type")
|
|
}
|
|
}
|
|
|
|
func TestClassify(t *testing.T) {
|
|
p := &Provider{}
|
|
if p.Classify("v3/modules/puppetlabs-stdlib") != provider.Mutable {
|
|
t.Error("modules should be mutable")
|
|
}
|
|
if p.Classify("v3/releases?module=x") != provider.Mutable {
|
|
t.Error("releases should be mutable")
|
|
}
|
|
if p.Classify("v3/files/puppetlabs-stdlib-1.0.0.tar.gz") != provider.Immutable {
|
|
t.Error("files should be immutable")
|
|
}
|
|
}
|
|
|
|
func TestContentType(t *testing.T) {
|
|
p := &Provider{}
|
|
if p.ContentType("x/mod-1.0.0.tar.gz") != "application/gzip" {
|
|
t.Error("tar.gz")
|
|
}
|
|
if p.ContentType("v3/modules/x") != "application/json" {
|
|
t.Error("v3 json")
|
|
}
|
|
if p.ContentType("other") != "application/octet-stream" {
|
|
t.Error("default")
|
|
}
|
|
}
|
|
|
|
func TestUpstreamURL(t *testing.T) {
|
|
got := (&Provider{}).UpstreamURL(models.Remote{BaseURL: "https://forgeapi.puppet.com/"}, "/v3/modules/x")
|
|
if got != "https://forgeapi.puppet.com/v3/modules/x" {
|
|
t.Errorf("got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestRewriteResponse(t *testing.T) {
|
|
p := &Provider{}
|
|
remote := models.Remote{Name: "forge", BaseURL: "https://forgeapi.puppet.com"}
|
|
|
|
if out, _ := p.RewriteResponse([]byte("x"), remote, ""); out != nil {
|
|
t.Error("empty proxyBaseURL is a no-op")
|
|
}
|
|
|
|
body := []byte(`{"file_uri":"/v3/files/mod.tar.gz","home":"https://forgeapi.puppet.com/x"}`)
|
|
out, err := p.RewriteResponse(body, remote, "http://proxy")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
s := string(out)
|
|
if !strings.Contains(s, "http://proxy/api/v1/remote/forge/v3/files/mod.tar.gz") {
|
|
t.Errorf("v3/files not rewritten: %s", s)
|
|
}
|
|
if !strings.Contains(s, "http://proxy/api/v1/remote/forge/x") {
|
|
t.Errorf("base URL not rewritten: %s", s)
|
|
}
|
|
}
|
|
|
|
func TestAuthHeaders(t *testing.T) {
|
|
h, _ := (&Provider{}).AuthHeaders(context.Background(), models.Remote{})
|
|
if h.Get("Authorization") != "" {
|
|
t.Error("no credentials, no header")
|
|
}
|
|
}
|