57 lines
2.1 KiB
Go
57 lines
2.1 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/domain"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/repository"
|
|
)
|
|
|
|
func TestTaskAPICancelsTrackedCertificateWorkflowTask(t *testing.T) {
|
|
server, tasks, _, closeDB := newTaskAPITestServer(t, nil)
|
|
defer closeDB()
|
|
item := repository.Task{
|
|
ID: "tracked-workflow-task", DomainID: "domain-1",
|
|
HistoryScope: repository.HistoryScope{ScopeType: repository.ScopeProject, ProjectID: domain.DefaultProjectID},
|
|
TaskType: "certificate_deploy", Status: "running", Stage: "certificate_deploy",
|
|
}
|
|
if err := tasks.CreateTask(t.Context(), item); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
cancelled := make(chan struct{}, 1)
|
|
server.registerTaskCancel(item.ID, func() { cancelled <- struct{}{} })
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, httptest.NewRequest(http.MethodDelete, "/api/v1/tasks/"+item.ID, nil))
|
|
stored, err := tasks.GetTask(t.Context(), item.ID)
|
|
if response.Code != http.StatusOK || err != nil || stored.Status != "cancelled" {
|
|
t.Fatalf("tracked workflow task was not cancelled: %d %+v %v", response.Code, stored, err)
|
|
}
|
|
select {
|
|
case <-cancelled:
|
|
case <-time.After(time.Second):
|
|
t.Fatal("tracked workflow cancellation callback was not called")
|
|
}
|
|
}
|
|
|
|
func TestTaskAPICannotFalselyCancelWorkflowTasks(t *testing.T) {
|
|
for _, kind := range []string{"renewal_run"} {
|
|
t.Run(kind, func(t *testing.T) {
|
|
server, tasks, _, closeDB := newTaskAPITestServer(t, nil)
|
|
defer closeDB()
|
|
item := repository.Task{ID: "workflow-task", DomainID: "domain-1", TaskType: kind, Status: "running", Stage: "certificate_issue"}
|
|
if err := tasks.CreateTask(t.Context(), item); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
response := httptest.NewRecorder()
|
|
serveAdminHTTP(server, response, httptest.NewRequest(http.MethodDelete, "/api/v1/tasks/"+item.ID, nil))
|
|
stored, err := tasks.GetTask(t.Context(), item.ID)
|
|
if response.Code != http.StatusConflict || err != nil || stored.Status != "running" {
|
|
t.Fatalf("工作流任务被伪造为已取消：%d %+v %v", response.Code, stored, err)
|
|
}
|
|
})
|
|
}
|
|
}
|