mirror of
https://gitea.com/gitea/gitea-mcp.git
synced 2026-08-26 10:07:45 +00:00
Lunny Xiao
bfe0d4c9b0
Adds a Docker healthcheck and the small HTTP/CLI plumbing it needs. - Expose `GET /healthz` on the HTTP-mode server. - Add `-healthcheck` to dial `http://127.0.0.1:<port>/healthz` and exit with status. - Add a Dockerfile `HEALTHCHECK` instruction using `-healthcheck`. - Document HTTP-mode usage and stdio override guidance. Closes https://gitea.com/gitea/gitea-mcp/issues/146 Assisted by Codet. Reviewed-on: https://gitea.com/gitea/gitea-mcp/pulls/254 Reviewed-by: silverwind <2021+silverwind@noreply.gitea.com>
107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"flag"
|
|
"maps"
|
|
"slices"
|
|
"testing"
|
|
|
|
flagPkg "gitea.com/gitea/gitea-mcp/pkg/flag"
|
|
)
|
|
|
|
func TestInitFlagSetBind(t *testing.T) {
|
|
for _, test := range []struct {
|
|
name string
|
|
args []string
|
|
want string
|
|
}{
|
|
{name: "default is empty, meaning all interfaces", args: []string{}},
|
|
{name: "-b sets the address", args: []string{"-b", "127.0.0.1"}, want: "127.0.0.1"},
|
|
{name: "-bind sets an IPv6 literal", args: []string{"-bind", "::1"}, want: "::1"},
|
|
} {
|
|
t.Run(test.name, func(t *testing.T) {
|
|
t.Cleanup(func() { flagPkg.Bind = "" })
|
|
fs := flag.NewFlagSet("test", flag.ContinueOnError)
|
|
initFlagSet(fs, test.args, func(string) string { return "" }, func(string) ([]byte, error) { return nil, nil }, &bytes.Buffer{})
|
|
if flagPkg.Bind != test.want {
|
|
t.Errorf("Bind = %q, want %q", flagPkg.Bind, test.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestInitFlagSetScopes(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
args []string
|
|
env map[string]string
|
|
want []string
|
|
}{
|
|
{
|
|
name: "no scope flag or env leaves AllowedScopes unset",
|
|
args: []string{},
|
|
want: nil,
|
|
},
|
|
{
|
|
name: "-S sets a single scope",
|
|
args: []string{"-S", "repository"},
|
|
want: []string{"repository"},
|
|
},
|
|
{
|
|
name: "-scope sets a comma-separated list",
|
|
args: []string{"-scope", "repository,file"},
|
|
want: []string{"file", "repository"},
|
|
},
|
|
{
|
|
name: "GITEA_SCOPES env sets the default",
|
|
args: []string{},
|
|
env: map[string]string{"GITEA_SCOPES": "issue,pull_request"},
|
|
want: []string{"issue", "pull_request"},
|
|
},
|
|
{
|
|
name: "-S flag takes precedence over GITEA_SCOPES env",
|
|
args: []string{"-S", "file"},
|
|
env: map[string]string{"GITEA_SCOPES": "issue"},
|
|
want: []string{"file"},
|
|
},
|
|
{
|
|
name: "normalizes case, whitespace, and hyphens/spaces to underscores",
|
|
args: []string{"-S", " Pull Request , pull-request , PULL_REQUEST "},
|
|
want: []string{"pull_request"},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
origScopes := flagPkg.AllowedScopes
|
|
t.Cleanup(func() {
|
|
flagPkg.AllowedScopes = origScopes
|
|
})
|
|
flagPkg.AllowedScopes = nil
|
|
|
|
getenv := func(key string) string { return tt.env[key] }
|
|
readFile := func(string) ([]byte, error) { return nil, nil }
|
|
fs := flag.NewFlagSet("test", flag.ContinueOnError)
|
|
var stderr bytes.Buffer
|
|
|
|
initFlagSet(fs, tt.args, getenv, readFile, &stderr)
|
|
|
|
got := slices.Sorted(maps.Keys(flagPkg.AllowedScopes))
|
|
want := slices.Clone(tt.want)
|
|
slices.Sort(want)
|
|
if !slices.Equal(got, want) {
|
|
t.Errorf("AllowedScopes = %v, want %v", got, want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestInitFlagSetHealthcheck(t *testing.T) {
|
|
t.Cleanup(func() { healthcheck = false })
|
|
fs := flag.NewFlagSet("test", flag.ContinueOnError)
|
|
initFlagSet(fs, []string{"-healthcheck"}, func(string) string { return "" }, func(string) ([]byte, error) { return nil, nil }, &bytes.Buffer{})
|
|
if !healthcheck {
|
|
t.Error("healthcheck = false, want true")
|
|
}
|
|
}
|