Agent-System/internal/server/project_workflow_api_test.go

package server_test

import (
	"bytes"
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"chat-agent-system/internal/bizdev"
	"chat-agent-system/internal/config"
	"chat-agent-system/internal/projectworkflow"
	"chat-agent-system/internal/server"
)

func setupTestServerForProjectWorkflow(t *testing.T) (*server.Server, http.Handler) {
	cfg := &config.Config{
		DataDir: t.TempDir(),
	}
	svc := bizdev.NewService(cfg)
	srv := server.NewServer(cfg, svc)
	handler := srv.Routes()
	return srv, handler
}

func TestProjectWorkflowAPI_Lifecycle(t *testing.T) {
	_, handler := setupTestServerForProjectWorkflow(t)

	// 1. 未认证创建项目 -> 401
	bodyUnauth, _ := json.Marshal(projectworkflow.CreateProjectInput{
		Name: "未授权项目",
	})
	reqUnauth := httptest.NewRequest(http.MethodPost, "/api/v1/projects", bytes.NewReader(bodyUnauth))
	recUnauth := httptest.NewRecorder()
	handler.ServeHTTP(recUnauth, reqUnauth)
	assert.Equal(t, http.StatusUnauthorized, recUnauth.Code)

	// 2. 认证用户创建项目 (User 2: 张工) -> 201
	bodyCreate, _ := json.Marshal(projectworkflow.CreateProjectInput{
		Name:         "智能高频电刀 (ElectroCutter)",
		DeviceType:   "高频手术设备",
		IntendedUse:  "微创外科软组织切割与止血",
		TargetMarket: "NMPA Class III",
	})
	reqCreate := httptest.NewRequest(http.MethodPost, "/api/v1/projects", bytes.NewReader(bodyCreate))
	reqCreate.Header.Set("X-User-ID", "2")
	recCreate := httptest.NewRecorder()
	handler.ServeHTTP(recCreate, reqCreate)
	require.Equal(t, http.StatusCreated, recCreate.Code)

	var createdProj projectworkflow.Project
	err := json.Unmarshal(recCreate.Body.Bytes(), &createdProj)
	require.NoError(t, err)
	assert.Equal(t, int64(2), createdProj.CreatorUserID)
	assert.Equal(t, "main", createdProj.ActiveWorkflowKey)
	assert.Equal(t, 1, createdProj.WorkflowSelectionVersion)

	// 3. 列出项目列表 -> 包含刚创建的项目
	reqList := httptest.NewRequest(http.MethodGet, "/api/v1/projects", nil)
	reqList.Header.Set("X-User-ID", "2")
	recList := httptest.NewRecorder()
	handler.ServeHTTP(recList, reqList)
	assert.Equal(t, http.StatusOK, recList.Code)

	var projList []*projectworkflow.Project
	err = json.Unmarshal(recList.Body.Bytes(), &projList)
	require.NoError(t, err)
	assert.NotEmpty(t, projList)

	// 4. 查询项目挂载的工作流清单 -> 包含 3 个默认工作流
	reqWFs := httptest.NewRequest(http.MethodGet, "/api/v1/projects/1001/workflows", nil)
	recWFs := httptest.NewRecorder()
	handler.ServeHTTP(recWFs, reqWFs)
	assert.Equal(t, http.StatusOK, recWFs.Code)

	var wfList []*projectworkflow.ProjectWorkflow
	err = json.Unmarshal(recWFs.Body.Bytes(), &wfList)
	require.NoError(t, err)
	assert.Equal(t, 3, len(wfList))

	// 5. 非创建人尝试切换工作流 (User 3) -> 403 Forbidden
	bodySwitch, _ := json.Marshal(projectworkflow.ActivateWorkflowInput{
		WorkflowKey:     "product_manager",
		ExpectedVersion: 1,
		Reason:          "非法切换",
	})
	reqSwitchOther := httptest.NewRequest(http.MethodPut, "/api/v1/projects/1001/active-workflow", bytes.NewReader(bodySwitch))
	reqSwitchOther.Header.Set("X-User-ID", "3") // Other user
	recSwitchOther := httptest.NewRecorder()
	handler.ServeHTTP(recSwitchOther, reqSwitchOther)
	assert.Equal(t, http.StatusForbidden, recSwitchOther.Code)

	// 6. 创建人切换工作流 (User 2) -> 200 OK
	bodySwitchCreator, _ := json.Marshal(projectworkflow.ActivateWorkflowInput{
		WorkflowKey:     "product_manager",
		ExpectedVersion: 1,
		Reason:          "切换至智能产品经理编制需求",
	})
	reqSwitchCreator := httptest.NewRequest(http.MethodPut, "/api/v1/projects/1001/active-workflow", bytes.NewReader(bodySwitchCreator))
	reqSwitchCreator.Header.Set("X-User-ID", "2") // Creator
	recSwitchCreator := httptest.NewRecorder()
	handler.ServeHTTP(recSwitchCreator, reqSwitchCreator)
	require.Equal(t, http.StatusOK, recSwitchCreator.Code)

	var switchResult projectworkflow.ActivateWorkflowResult
	err = json.Unmarshal(recSwitchCreator.Body.Bytes(), &switchResult)
	require.NoError(t, err)
	assert.Equal(t, "product_manager", switchResult.ActiveWorkflowKey)
	assert.Equal(t, 2, switchResult.Version)

	// 7. 再次用旧版本 1 切换 -> 409 Conflict
	bodySwitchConflict, _ := json.Marshal(projectworkflow.ActivateWorkflowInput{
		WorkflowKey:     "test_manager",
		ExpectedVersion: 1,
		Reason:          "并发冲突测试",
	})
	reqSwitchConflict := httptest.NewRequest(http.MethodPut, "/api/v1/projects/1001/active-workflow", bytes.NewReader(bodySwitchConflict))
	reqSwitchConflict.Header.Set("X-User-ID", "2")
	recSwitchConflict := httptest.NewRecorder()
	handler.ServeHTTP(recSwitchConflict, reqSwitchConflict)
	assert.Equal(t, http.StatusConflict, recSwitchConflict.Code)

	// 8. 查询全局工作流规范目录 -> 200 OK
	reqDefs := httptest.NewRequest(http.MethodGet, "/api/v1/workflows", nil)
	recDefs := httptest.NewRecorder()
	handler.ServeHTTP(recDefs, reqDefs)
	assert.Equal(t, http.StatusOK, recDefs.Code)
	var defs []*projectworkflow.Definition
	err = json.Unmarshal(recDefs.Body.Bytes(), &defs)
	require.NoError(t, err)
	assert.Equal(t, 3, len(defs))
}