agentbbs/internal/forgejo/forgejo_test.go

package forgejo

import (
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"
)

func TestConfiguredRequiresURLAndToken(t *testing.T) {
	if (Config{BaseURL: "https://git.example.com"}).Configured() {
		t.Fatal("missing token should be unconfigured")
	}
	if (Config{Token: "t"}).Configured() {
		t.Fatal("missing URL should be unconfigured")
	}
	if !(Config{BaseURL: "https://git.example.com", Token: "t"}).Configured() {
		t.Fatal("URL+token should be configured")
	}
}

func TestEnsureUserNoOpWhenUnconfigured(t *testing.T) {
	if _, err := (Config{}).EnsureUser("alice", "a@x.com"); err == nil {
		t.Fatal("expected error when unconfigured")
	}
}

func TestEnsureUserCreatesWhenMissing(t *testing.T) {
	var got struct {
		lookup bool
		create bool
		body   map[string]any
	}
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		switch {
		case r.Method == http.MethodGet && r.URL.Path == "/api/v1/users/alice":
			got.lookup = true
			w.WriteHeader(http.StatusNotFound)
		case r.Method == http.MethodPost && r.URL.Path == "/api/v1/admin/users":
			got.create = true
			if r.Header.Get("Authorization") != "token secret" {
				t.Errorf("missing auth header: %q", r.Header.Get("Authorization"))
			}
			_ = json.NewDecoder(r.Body).Decode(&got.body)
			w.WriteHeader(http.StatusCreated)
			_, _ = w.Write([]byte(`{"id":1}`))
		default:
			t.Errorf("unexpected %s %s", r.Method, r.URL.Path)
			w.WriteHeader(http.StatusTeapot)
		}
	}))
	defer srv.Close()

	c := Config{BaseURL: srv.URL, Token: "secret"}
	created, err := c.EnsureUser("alice", "a@x.com")
	if err != nil {
		t.Fatalf("EnsureUser: %v", err)
	}
	if !created {
		t.Fatal("expected created=true")
	}
	if !got.lookup || !got.create {
		t.Fatalf("expected lookup+create, got %+v", got)
	}
	if got.body["must_change_password"] != true {
		t.Errorf("expected must_change_password=true, got %v", got.body["must_change_password"])
	}
	if got.body["username"] != "alice" {
		t.Errorf("expected username alice, got %v", got.body["username"])
	}
}

func TestEnsureUserNoOpWhenExists(t *testing.T) {
	created := false
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == http.MethodPost {
			created = true
		}
		w.WriteHeader(http.StatusOK)
		_, _ = w.Write([]byte(`{"id":1}`))
	}))
	defer srv.Close()

	c := Config{BaseURL: srv.URL, Token: "secret"}
	got, err := c.EnsureUser("alice", "a@x.com")
	if err != nil {
		t.Fatalf("EnsureUser: %v", err)
	}
	if got {
		t.Fatal("expected created=false for existing user")
	}
	if created {
		t.Fatal("must not POST when the user already exists")
	}
}

const aliceKey = "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAITESTKEY alice@bbs"

func TestEnsureKeyAddsWhenMissing(t *testing.T) {
	var posted map[string]any
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		switch {
		case r.Method == http.MethodGet && r.URL.Path == "/api/v1/users/alice/keys":
			_, _ = w.Write([]byte(`[]`))
		case r.Method == http.MethodPost && r.URL.Path == "/api/v1/admin/users/alice/keys":
			_ = json.NewDecoder(r.Body).Decode(&posted)
			w.WriteHeader(http.StatusCreated)
			_, _ = w.Write([]byte(`{"id":1}`))
		default:
			t.Errorf("unexpected %s %s", r.Method, r.URL.Path)
		}
	}))
	defer srv.Close()

	c := Config{BaseURL: srv.URL, Token: "secret"}
	added, err := c.EnsureKey("alice", "agentbbs", aliceKey)
	if err != nil {
		t.Fatalf("EnsureKey: %v", err)
	}
	if !added {
		t.Fatal("expected added=true")
	}
	if posted["key"] != aliceKey {
		t.Errorf("posted key = %v", posted["key"])
	}
}

func TestEnsureKeyIdempotentIgnoringComment(t *testing.T) {
	posted := false
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method == http.MethodPost {
			posted = true
		}
		// Same key material, different comment — must be treated as already present.
		_, _ = w.Write([]byte(`[{"key":"ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAITESTKEY different-comment"}]`))
	}))
	defer srv.Close()

	c := Config{BaseURL: srv.URL, Token: "secret"}
	added, err := c.EnsureKey("alice", "agentbbs", aliceKey)
	if err != nil {
		t.Fatalf("EnsureKey: %v", err)
	}
	if added {
		t.Fatal("expected added=false when key material already present")
	}
	if posted {
		t.Fatal("must not POST when the key already exists")
	}
}

func TestEnsureKeyBlankIsNoOp(t *testing.T) {
	c := Config{BaseURL: "https://git.example.com", Token: "t"}
	if added, err := c.EnsureKey("alice", "agentbbs", "  "); err != nil || added {
		t.Fatalf("blank key should be a silent no-op, got added=%v err=%v", added, err)
	}
}