dingtalk-workspace-cli/internal/helpers/connect_stream_test.go

// Copyright 2026 Alibaba Group
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.

package helpers

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"
)

// TestBrandReply covers the qoderwork identity rewrite using the exact replies
// captured from a real qodercli (QoderWork.app) headless run.
func TestBrandReply(t *testing.T) {
	const persona = "我是 QoderWork 助手，钉钉群里的智能助手。"
	cases := []struct {
		name    string
		channel string
		in      string
		want    string
	}{
		{
			name:    "real reply 1 — whole sentence is the self-intro",
			channel: "qoderwork",
			in:      "我是 Qoder，一个帮助你完成软件工程任务的交互式命令行工具。",
			want:    persona,
		},
		{
			name:    "real reply 2 — self-intro then capability text kept",
			channel: "qoderwork",
			in:      "我是 Qoder，一个交互式命令行工具，帮助用户完成软件工程任务。我可以协助你编写代码、调试问题。",
			want:    persona + "我可以协助你编写代码、调试问题。",
		},
		{
			name:    "real reply 3 — self-intro then newline block kept",
			channel: "qoderwork",
			in:      "我是 Qoder，一个交互式命令行工具，主要帮助用户完成软件工程相关的任务。\n\n我的核心能力包括：",
			want:    persona + "\n\n我的核心能力包括：",
		},
		{
			name:    "english self-intro",
			channel: "qoderwork",
			in:      "I am Qoder, an interactive CLI tool that helps with software engineering.",
			want:    persona,
		},
		{
			name:    "mid-text Qoder mention is NOT rewritten",
			channel: "qoderwork",
			in:      "Qoder 是一家做 AI 编程工具的公司，它的产品不错。",
			want:    "Qoder 是一家做 AI 编程工具的公司，它的产品不错。",
		},
		{
			name:    "normal answer untouched",
			channel: "qoderwork",
			in:      "1+1 等于 2。",
			want:    "1+1 等于 2。",
		},
		{
			name:    "other channels pass through even if they say Qoder",
			channel: "codebuddy",
			in:      "我是 Qoder，一个命令行工具。",
			want:    "我是 Qoder，一个命令行工具。",
		},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			if got := brandReply(tc.channel, tc.in); got != tc.want {
				t.Fatalf("brandReply(%q, %q)\n  got:  %q\n  want: %q", tc.channel, tc.in, got, tc.want)
			}
		})
	}
}

// TestClaudeUserSettingsEnv covers the third-party-provider auth passthrough
// for the claudecode channel (issue PeterGuy326#10): the env block of the
// user-level Claude settings must surface as process env entries, without
// clobbering variables the operator already exported.
func TestClaudeUserSettingsEnv(t *testing.T) {
	dir := t.TempDir()
	t.Setenv("CLAUDE_CONFIG_DIR", dir)

	// No settings file → no injection.
	if got := claudeUserSettingsEnv(); len(got) != 0 {
		t.Fatalf("expected no env without settings.json, got %v", got)
	}

	writeSettings := func(body string) {
		t.Helper()
		if err := os.WriteFile(filepath.Join(dir, "settings.json"), []byte(body), 0o600); err != nil {
			t.Fatal(err)
		}
	}

	// Malformed JSON → fail open with no injection.
	writeSettings(`{not json`)
	if got := claudeUserSettingsEnv(); len(got) != 0 {
		t.Fatalf("expected no env for malformed settings, got %v", got)
	}

	// Provider credentials in the env block are exposed; a key already present
	// in the process environment is left alone.
	t.Setenv("ANTHROPIC_MODEL", "from-shell")
	writeSettings(`{"env":{"ANTHROPIC_BASE_URL":"https://relay.example","ANTHROPIC_AUTH_TOKEN":"tok","ANTHROPIC_MODEL":"from-settings"}}`)
	got := claudeUserSettingsEnv()
	want := map[string]bool{
		"ANTHROPIC_BASE_URL=https://relay.example": false,
		"ANTHROPIC_AUTH_TOKEN=tok":                 false,
	}
	for _, kv := range got {
		if strings.HasPrefix(kv, "ANTHROPIC_MODEL=") {
			t.Fatalf("process env must win over settings env, got %q", kv)
		}
		if _, ok := want[kv]; ok {
			want[kv] = true
		}
	}
	for kv, seen := range want {
		if !seen {
			t.Fatalf("missing %q in %v", kv, got)
		}
	}
}

// TestCheckFDLimit verifies that checkFDLimit runs without panic and respects
// the envDurationMS pattern for the keepAlive default.
func TestCheckFDLimit(t *testing.T) {
	// Should not panic regardless of the actual ulimit.
	checkFDLimit()
}

func TestEnvDurationMS(t *testing.T) {
	def := 30000 * time.Millisecond // 30s
	if got := envDurationMS("DWS_CONNECT_KEEPALIVE_MS_TEST_ABSENT", def); got != def {
		t.Fatalf("default keepAlive = %v, want %v", got, def)
	}
	t.Setenv("DWS_CONNECT_KEEPALIVE_MS_TEST", "10000")
	if got := envDurationMS("DWS_CONNECT_KEEPALIVE_MS_TEST", def); got != 10*time.Second {
		t.Fatalf("env override = %v, want 10s", got)
	}
	t.Setenv("DWS_CONNECT_KEEPALIVE_MS_TEST", "bogus")
	if got := envDurationMS("DWS_CONNECT_KEEPALIVE_MS_TEST", def); got != def {
		t.Fatalf("invalid env falls back to default = %v, want %v", got, def)
	}
}