CyberStrikeAI/internal/handler/openapi_i18n_test.go

package handler

import "testing"

func TestEnrichSpecWithI18nKeysForAssetImport(t *testing.T) {
	responses := map[string]interface{}{
		"200": map[string]interface{}{"description": "导入完成"},
		"400": map[string]interface{}{"description": "数量或资产字段校验失败"},
		"403": map[string]interface{}{"description": "缺少 asset:write 权限或无权访问指定项目"},
		"500": map[string]interface{}{"description": "导入事务失败"},
	}
	operation := map[string]interface{}{
		"tags":      []string{"资产管理"},
		"summary":   "批量导入资产",
		"responses": responses,
	}
	spec := map[string]interface{}{
		"paths": map[string]interface{}{
			"/api/assets/import": map[string]interface{}{
				"post": operation,
			},
		},
	}

	enrichSpecWithI18nKeys(spec)

	tagKeys, ok := operation["x-i18n-tags"].([]string)
	if !ok || len(tagKeys) != 1 || tagKeys[0] != "assetManagement" {
		t.Fatalf("unexpected asset tag i18n keys: %#v", operation["x-i18n-tags"])
	}
	if got := operation["x-i18n-summary"]; got != "importAssets" {
		t.Fatalf("unexpected asset summary i18n key: %#v", got)
	}
	expectedResponseKeys := map[string]string{
		"200": "assetImportCompleted",
		"400": "assetImportValidationFailed",
		"403": "assetImportForbidden",
		"500": "assetImportTransactionFailed",
	}
	for status, want := range expectedResponseKeys {
		response := responses[status].(map[string]interface{})
		if got := response["x-i18n-description"]; got != want {
			t.Errorf("unexpected asset response i18n key for %s: got %#v, want %q", status, got, want)
		}
	}
}