package goproxy

import (
	"context"
	"testing"

	"git.unkin.net/unkin/artifactapi/pkg/models"
)

func TestGoProxyURLAuthRewrite(t *testing.T) {
	p := &Provider{}
	if got := p.UpstreamURL(models.Remote{BaseURL: "https://proxy.golang.org/"}, "/mod/@v/list"); got != "https://proxy.golang.org/mod/@v/list" {
		t.Errorf("upstream url %q", got)
	}
	if out, err := p.RewriteResponse([]byte("x"), models.Remote{}, "http://p"); out != nil || err != nil {
		t.Error("goproxy never rewrites")
	}
	if h, _ := p.AuthHeaders(context.Background(), models.Remote{Username: "u", Password: "p"}); h.Get("Authorization") == "" {
		t.Error("expected basic auth header")
	}
	if got := p.ContentType("mod/@v/v1.0.0.info"); got != "application/json" {
		t.Errorf("info content type %q", got)
	}
	if got := p.ContentType("mod/@v/v1.0.0.mod"); got != "text/plain" {
		t.Errorf("mod content type %q", got)
	}
}