artifactapi/internal/api/v2/remotes_ttl_test.go

package v2

import (
	"context"
	"testing"

	"git.unkin.net/unkin/artifactapi/internal/database"
)

// An omitted mutable_ttl defaults to 3600 on create and keeps the stored value
// on update; an explicit 0 is stored as 0 in both.
func TestMutableTTLOmittedVsZero(t *testing.T) {
	if testDSN == "" {
		t.Skip("Docker unavailable")
	}
	db, err := database.New(testDSN)
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	h := NewRemotesHandler(db, nil, nil).Routes()
	ttl := func(name string) int {
		t.Helper()
		r, err := db.GetRemote(context.Background(), name)
		if err != nil {
			t.Fatalf("get %s: %v", name, err)
		}
		return r.MutableTTL
	}
	const body = `"package_type":"rpm","base_url":"https://example.com/repo"`

	steps := []struct {
		method, path, json string
		want               int
	}{
		{"POST", "/", `{"name":"ttl-omit",` + body + `}`, 3600},
		{"PUT", "/ttl-omit", `{` + body + `,"mutable_ttl":120}`, 120},
		{"PUT", "/ttl-omit", `{` + body + `}`, 120},
		{"PUT", "/ttl-omit", `{` + body + `,"mutable_ttl":0}`, 0},
		{"PUT", "/ttl-omit", `{` + body + `}`, 0},
	}
	for i, s := range steps {
		if c := do(t, h, s.method, s.path, s.json); c/100 != 2 {
			t.Fatalf("step %d %s %s = %d", i, s.method, s.path, c)
		}
		if got := ttl("ttl-omit"); got != s.want {
			t.Errorf("step %d: mutable_ttl = %d, want %d", i, got, s.want)
		}
	}

	if c := do(t, h, "POST", "/", `{"name":"ttl-zero",`+body+`,"mutable_ttl":0}`); c != 201 {
		t.Fatalf("create explicit 0 = %d", c)
	}
	if got := ttl("ttl-zero"); got != 0 {
		t.Errorf("explicit 0 stored as %d", got)
	}
}

func TestQuarantineDaysDefault(t *testing.T) {
	if testDSN == "" {
		t.Skip("Docker unavailable")
	}
	db, err := database.New(testDSN)
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	h := NewRemotesHandler(db, nil, nil).Routes()
	days := func() int {
		t.Helper()
		r, err := db.GetRemote(context.Background(), "q-days")
		if err != nil {
			t.Fatal(err)
		}
		return r.QuarantineDays
	}
	const body = `"package_type":"npm","base_url":"https://example.com","quarantine_enabled":true`
	for i, s := range []struct {
		method, path, json string
		want               int
	}{
		{"POST", "/", `{"name":"q-days",` + body + `}`, 3},
		{"PUT", "/q-days", `{` + body + `,"quarantine_days":14}`, 14},
		{"PUT", "/q-days", `{` + body + `}`, 14},
	} {
		if c := do(t, h, s.method, s.path, s.json); c/100 != 2 {
			t.Fatalf("step %d = %d", i, c)
		}
		if got := days(); got != s.want {
			t.Errorf("step %d: quarantine_days = %d, want %d", i, got, s.want)
		}
	}
}

func TestQuarantineDaysValidation(t *testing.T) {
	if testDSN == "" {
		t.Skip("Docker unavailable")
	}
	db, err := database.New(testDSN)
	if err != nil {
		t.Fatal(err)
	}
	defer db.Close()
	h := NewRemotesHandler(db, nil, nil).Routes()
	const body = `"package_type":"npm","base_url":"https://example.com"`
	for i, s := range []struct {
		method, path, json string
		want               int
	}{
		{"POST", "/", `{"name":"q-val",` + body + `,"quarantine_enabled":true,"quarantine_days":0}`, 400},
		{"POST", "/", `{"name":"q-val",` + body + `,"quarantine_enabled":true,"quarantine_days":-1}`, 400},
		{"POST", "/", `{"name":"q-val",` + body + `,"quarantine_days":0}`, 201},
		{"PUT", "/q-val", `{` + body + `,"quarantine_enabled":true}`, 400},
		{"PUT", "/q-val", `{` + body + `,"quarantine_enabled":true,"quarantine_days":1}`, 200},
		{"PUT", "/q-val", `{` + body + `,"quarantine_enabled":true,"quarantine_days":0}`, 400},
	} {
		if c := do(t, h, s.method, s.path, s.json); c != s.want {
			t.Errorf("step %d %s %s = %d, want %d", i, s.method, s.path, c, s.want)
		}
	}
}