gitea-tea/internal/app/checkout_test.go

// Copyright 2026 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package app

import (
	"context"
	"fmt"
	"net/http"
	"strings"
	"testing"

	"gitea.dev/tea/internal/git"
	"gitea.dev/tea/internal/settings"
	"github.com/stretchr/testify/require"
)

func TestCheckoutRemoteUsesOnlyMatchingRepository(t *testing.T) {
	deps, _, _ := testDeps(t)
	deps.Git = &fakeGit{results: map[string]string{
		"remote": "fork\norigin\nupstream\ntracking", "symbolic-ref --short HEAD": "main", "config --get branch.main.remote": "tracking",
		"remote get-url fork":     "https://example.org/bob/project.git",
		"remote get-url origin":   "https://example.org/alice/project.git",
		"remote get-url upstream": "https://other.example.org/alice/project.git",
		"remote get-url tracking": "https://example.org/alice/project.git",
	}}
	a := &application{deps: deps}
	snapshot := settings.Snapshot{Preferences: settings.Defaults()}
	target := target{Host: "https://example.org", Repo: "alice/project"}
	remote, err := a.checkoutRemote(context.Background(), target, snapshot)
	require.NoError(t, err)
	require.Equal(t, "tracking", remote)
	a.options.remote = "upstream"
	_, err = a.checkoutRemote(context.Background(), target, snapshot)
	require.ErrorContains(t, err, "does not match")
	a.options.remote = "origin"
	remote, err = a.checkoutRemote(context.Background(), target, snapshot)
	require.NoError(t, err)
	require.Equal(t, "origin", remote)
}

func TestCheckoutSelectedInstanceDisambiguatesSharedSSHRemote(t *testing.T) {
	deps, _, _ := testDeps(t)
	deps.Git = &fakeGit{results: map[string]string{
		"remote": "local\norigin", "symbolic-ref --short HEAD": "main", "config --get branch.main.remote": "origin",
		"remote get-url local":  "/srv/git/project.git",
		"remote get-url origin": "ssh://git@git.example.org:2222/alice/project.git",
	}}
	a := &application{deps: deps}
	snapshot := settings.Snapshot{Preferences: settings.Defaults(), Hosts: settings.Hosts{Hosts: map[string]settings.Host{
		"https://example.org/first":  {SSHHost: "git.example.org:2222"},
		"https://example.org/second": {SSHHost: "git.example.org:2222"},
	}}}
	remote, err := a.checkoutRemote(context.Background(), target{Host: "https://example.org/second", Repo: "alice/project", host: snapshot.Hosts.Hosts["https://example.org/second"]}, snapshot)
	require.NoError(t, err)
	require.Equal(t, "origin", remote)
}

func TestCheckoutMatchesRemoteThroughEnvironmentSSHHost(t *testing.T) {
	server := newAuthenticatedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		require.Equal(t, "/api/v1/repos/alice/project/pulls/7", r.URL.Path)
		_, _ = fmt.Fprint(w, `{"number":7,"head":{"sha":"abc123","ref":"feature","repo":{"full_name":"alice/project","ssh_url":"git@ssh.example.org:alice/project.git"}}}`)
	}))
	defer server.Close()
	deps, out, _ := testDeps(t)
	deps.Getenv = serverEnv(server.URL, map[string]string{"GITEA_INSTANCE_SSH_HOST": "ssh.example.org"})
	deps.Git = &fakeGit{results: map[string]string{
		"remote": "origin", "remote get-url origin": "git@ssh.example.org:alice/project.git",
		"rev-parse --is-inside-work-tree": "true", "check-ref-format --branch pr/7": "",
		"check-ref-format --branch feature":              "",
		"status --porcelain=v1 --untracked-files=normal": "", "for-each-ref --format=%(refname) refs/heads/": "",
		"fetch --no-tags -- origin refs/heads/feature": "", "rev-parse --verify FETCH_HEAD^{commit}": "abc123", "switch --create pr/7 abc123": "",
		"config branch.pr/7.remote origin": "", "config branch.pr/7.merge refs/heads/feature": "",
		"config branch.pr/7.tea-pr 7": "", "config branch.pr/7.tea-base-remote origin": "",
	}}
	require.NoError(t, Execute(context.Background(), deps, []string{"pr", "checkout", "7", "--repo", "alice/project", "--remote", "origin", "--no-input", "--json"}))
	require.JSONEq(t, `{"branch":"pr/7","commit":"abc123","detached":false,"remote":"origin","tracking":"origin/feature"}`, out.String())
}

func TestSSHCheckoutUsesAccountKeyOnlyForFetch(t *testing.T) {
	server := newAuthenticatedServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		require.Equal(t, "/api/v1/repos/alice/project/pulls/7", r.URL.Path)
		_, _ = fmt.Fprint(w, `{"number":7,"head":{"sha":"abc123","ref":"feature","repo":{"full_name":"alice/project","clone_url":"https://example.org/alice/project.git","ssh_url":"ssh://git@git.example.org:2222/alice/project.git","permissions":{"push":true}}}}`)
	}))
	defer server.Close()
	deps, out, _ := serverDeps(t, server.URL)
	configureHost(t, deps, server.URL, settings.Host{User: "alice", SSHHost: "git.example.org:2222", Users: map[string]settings.Account{
		"alice": {AuthMethod: "token", CredentialStore: "file", SSHKey: "/keys/alice key"},
		"bob":   {AuthMethod: "token", CredentialStore: "file", SSHKey: "/keys/bob"},
	}})
	sshArgs, err := git.SSHArguments("/keys/alice key", false)
	require.NoError(t, err)
	fetch := strings.Join(append(sshArgs, "fetch", "--no-tags", "--", "origin", "refs/heads/feature"), " ")
	runner := &fakeGit{results: map[string]string{
		"remote": "origin", "remote get-url origin": "ssh://git@git.example.org:2222/alice/project.git",
		"rev-parse --is-inside-work-tree": "true", "check-ref-format --branch pr/7": "",
		"check-ref-format --branch feature":              "",
		"status --porcelain=v1 --untracked-files=normal": "", "for-each-ref --format=%(refname) refs/heads/": "",
		fetch: "", "rev-parse --verify FETCH_HEAD^{commit}": "abc123", "switch --create pr/7 abc123": "",
		"config branch.pr/7.remote origin": "", "config branch.pr/7.merge refs/heads/feature": "",
		"config branch.pr/7.tea-pr 7": "", "config branch.pr/7.tea-base-remote origin": "",
	}}
	deps.Git = runner
	require.NoError(t, Execute(context.Background(), deps, []string{"pr", "checkout", "7", "--repo", "alice/project", "--remote", "origin", "--no-input", "--json"}))
	require.Contains(t, runner.calls, fetch)
	require.Contains(t, runner.calls, "config branch.pr/7.remote origin")
	require.Contains(t, runner.calls, "config branch.pr/7.merge refs/heads/feature")
	configured := 0
	for _, call := range runner.calls {
		if strings.HasPrefix(call, "-c ") {
			configured++
			require.Equal(t, fetch, call)
		}
	}
	require.Equal(t, 1, configured)
	require.False(t, runner.prompt)
	require.JSONEq(t, `{"branch":"pr/7","commit":"abc123","detached":false,"remote":"origin","tracking":"origin/feature"}`, out.String())
}