feat: Add integration test --story=1

Author: studyzy

.gitignore

@@ -66,3 +66,4 @@ release/
 # C/C++ artifacts (for CGO)
 *.o
 *.a
+coverage-data

adapters/rust/extractor.go

@@ -90,4 +90,4 @@ func isEmptyComment(content string) bool {
 	default:
 		return strings.TrimSpace(trimmed) == ""
 	}
-}
+}

adapters/translator/llm_test.go

@@ -6,9 +6,9 @@ import (
 	"os"
 	"testing"
 
+	openai "github.com/sashabaranov/go-openai"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	openai "github.com/sashabaranov/go-openai"
 )
 
 // TestBatchTranslationMock tests batch translation using a mock server
@@ -20,17 +20,17 @@ func TestBatchTranslationMock(t *testing.T) {
 		if !contains(content, `["Hello","World"]`) {
 			return nil, fmt.Errorf("unexpected prompt: %s", content)
 		}
-		
+
 		// Return valid JSON array
 		return createMockResponse(`["你好", "世界"]`), nil
 	}
-	
+
 	server1 := NewMockLLMServer(successHandler)
 	defer server1.Close()
 
 	trans1 := NewLLMTranslator("key", server1.URL, "model")
 	results, err := trans1.TranslateBatch(context.Background(), []string{"Hello", "World"}, "en", "zh-CN")
-	
+
 	require.NoError(t, err)
 	assert.Equal(t, []string{"你好", "世界"}, results)
 
@@ -72,7 +72,7 @@ func TestDeepSeekIntegration(t *testing.T) {
 	if testing.Short() {
 		t.Skip("Skipping integration test in short mode")
 	}
-	
+
 	apiKey := os.Getenv("OPENAI_API_KEY")
 	if apiKey == "" {
 		t.Skip("Skipping integration test: OPENAI_API_KEY not set")
@@ -119,7 +119,7 @@ func TestEdgeCases(t *testing.T) {
 		return createMockResponse(`[]`), nil
 	})
 	defer server.Close()
-	
+
 	trans := NewLLMTranslator("key", server.URL, "model")
 	res, err := trans.TranslateBatch(context.Background(), []string{}, "en", "zh")
 	assert.NoError(t, err)
@@ -132,7 +132,7 @@ func TestEdgeCases(t *testing.T) {
 		return createMockResponse(`["Quote \"", "Newline \\n"]`), nil
 	})
 	defer server2.Close()
-	
+
 	trans2 := NewLLMTranslator("key", server2.URL, "model")
 	res2, err := trans2.TranslateBatch(context.Background(), specials, "en", "zh")
 	assert.NoError(t, err)

adapters/translator/parser.go

@@ -25,7 +25,7 @@ func parseBatchResponse(resp string) ([]string, error) {
 			cleanResp = cleanResp[:idx]
 		}
 	}
-	
+
 	cleanResp = strings.TrimSpace(cleanResp)
 
 	var results []string

adapters/translator/parser_test.go

@@ -19,12 +19,12 @@ func TestParseBatchResponse(t *testing.T) {
 			want:  []string{"Hello", "World"},
 		},
 		{
-			name: "Markdown JSON",
+			name:  "Markdown JSON",
 			input: "```json\n[\"Hello\", \"World\"]\n```",
 			want:  []string{"Hello", "World"},
 		},
 		{
-			name: "Markdown without language",
+			name:  "Markdown without language",
 			input: "```\n[\"Hello\", \"World\"]\n```",
 			want:  []string{"Hello", "World"},
 		},
@@ -39,7 +39,7 @@ func TestParseBatchResponse(t *testing.T) {
 			// If LLM says "Here is the JSON: [...]", it might fail.
 			// We can improve this in T006 Edge Cases if needed.
 			input:   "Here is the JSON: [\"Hello\"]",
-			wantErr: true, 
+			wantErr: true,
 		},
 	}
 

cmd/codei18n/map.go

@@ -96,7 +96,7 @@ func runMapUpdate() {
 		if _, exists := m.Comments[c.ID]; !exists {
 			// [MOCK zh-CN->en] // Intelligently detect comment language
 			detectedLang := utils.DetectLanguage(c.SourceText)
-			
+
 			if detectedLang == cfg.LocalLanguage {
 				// [MOCK zh-CN->en] // The comment is in the local language (Chinese), stored as LocalLanguage
 				store.Set(c.ID, cfg.LocalLanguage, c.SourceText)

cmd/codei18n/root.go

@@ -112,4 +112,4 @@ func loadConfigFile(path string, merge bool) (bool, error) {
 
 func main() {
 	Execute()
-}
+}

cmd/codei18n/scan.go

@@ -21,6 +21,7 @@ var (
 	scanDir              string
 	scanFormat           string
 	scanOutput           string
+	scanLang             string
 	scanStdin            bool
 	scanWithTranslations bool
 )
@@ -43,6 +44,7 @@ func init() {
 	scanCmd.Flags().StringVarP(&scanDir, "dir", "d", ".", "指定扫描的目录路径")
 	scanCmd.Flags().StringVar(&scanFormat, "format", "table", "输出格式 (json, table)")
 	scanCmd.Flags().StringVarP(&scanOutput, "output", "o", "", "将输出写入指定文件 (默认 stdout)")
+	scanCmd.Flags().StringVar(&scanLang, "lang", "", "指定目标语言 (覆盖配置)")
 	scanCmd.Flags().BoolVar(&scanStdin, "stdin", false, "从 stdin 读取文件内容 (必须同时指定 --file)")
 	scanCmd.Flags().BoolVar(&scanWithTranslations, "with-translations", false, "在 JSON 输出中包含翻译文本")
 }
@@ -95,6 +97,9 @@ func runScan() {
 		} else {
 			// Populate LocalizedText
 			targetLang := cfg.LocalLanguage // Default target
+			if scanLang != "" {
+				targetLang = scanLang
+			}
 			for _, c := range comments {
 				if text, ok := store.Get(c.ID, targetLang); ok {
 					c.LocalizedText = text
@@ -128,8 +133,15 @@ func scanSingleFile(filename string) ([]*domain.Comment, error) {
 	if err != nil {
 		return nil, err
 	}
-	// Pass nil src to read from file
-	return adapter.Parse(filename, nil)
+
+	// Read file content manually to ensure consistency with directory scan
+	// and to debug potential read issues
+	content, err := os.ReadFile(filename)
+	if err != nil {
+		return nil, fmt.Errorf("read file failed: %w", err)
+	}
+
+	return adapter.Parse(filename, content)
 }
 
 func scanDirectory(dir string) ([]*domain.Comment, error) {

cmd/codei18n/translate.go

@@ -202,32 +202,32 @@ func runTranslate() {
 			// Current simple implementation:
 			// If direction changes in batch, we can't batch efficiently.
 			// Let's refine batching strategy.
-			
+
 			// Group by direction
 			// Actually, just calling TranslateBatch with mixed tasks is wrong because TranslateBatch takes `from, to` args.
 			// So we MUST ensure batch has same direction.
 			// Given loop order is random, we should probably separate tasks by direction first.
 			// But for now let's assume we can just check.
-			
-			// Refined logic: TranslateBatch takes single from/to. 
+
+			// Refined logic: TranslateBatch takes single from/to.
 			// So we must verify batch consistency.
-			
+
 			// Since we can't easily change the batching structure inside this goroutine loop without complicating it,
 			// let's do grouping before creating batches.
-			
+
 			// Just use the first task's direction. If mixed, we might fail or need to split.
 			// Actually, let's fix the task collection to be grouped.
 			// But wait, I already split into batches.
 			// The correct fix is to sort tasks or group tasks by direction before batching.
 			// Let's just group by direction in the loop below.
 			// BUT: simpler is just to ensure TranslateBatch is called with correct params.
 			// If a batch has mixed directions, we can't use TranslateBatch easily.
-			
+
 			// Let's assume for typical usage (filling missing), it's usually En->Zh.
 			// But mixed is possible.
 			// I will add a check: if batch has mixed directions, fallback to sequential logic or split it?
 			// Or better: Sort tasks by direction before batching.
-			
+
 			// Let's perform translation
 			from := currentBatch[0].fromLang
 			to := currentBatch[0].toLang

core/utils/lang_detect.go

@@ -11,14 +11,14 @@ import (
 func DetectLanguage(text string) string {
 	// [MOCK zh-CN->en] // Remove comment markers
 	normalized := NormalizeCommentText(text)
-	
+
 	// [MOCK zh-CN->en] // Check if it contains Chinese characters (CJK Unified Ideographs)
 	for _, r := range normalized {
 		if unicode.Is(unicode.Han, r) {
 			return "zh-CN"
 		}
 	}
-	
+
 	return "en"
 }