3x-ui/internal/web/service/key_gen_parse_test.go

package service

import "testing"

// The x25519 / mldsa65 / mlkem768 key generators parse xray's stdout by fixed
// slice index. A short or reformatted output (a future xray release, or a
// binary that failed silently) must yield an error, never an out-of-range
// panic that 500s the handler.
func TestParseXrayKeyPairOutput(t *testing.T) {
	a, b, err := parseXrayKeyPairOutput("Private key: abc123\nPublic key: def456\n")
	if err != nil {
		t.Fatalf("well-formed output errored: %v", err)
	}
	if a != "abc123" || b != "def456" {
		t.Fatalf("got (%q, %q), want (abc123, def456)", a, b)
	}

	malformed := []string{
		"",
		"only one line: value",
		"Private key: abc\n",
		"no colon here\nno colon two",
		"Private key\nPublic key",
	}
	for _, out := range malformed {
		func() {
			defer func() {
				if r := recover(); r != nil {
					t.Errorf("parseXrayKeyPairOutput panicked on %q: %v", out, r)
				}
			}()
			if _, _, err := parseXrayKeyPairOutput(out); err == nil {
				t.Errorf("expected error for malformed output %q, got nil", out)
			}
		}()
	}
}