add TestWorkflowConcurrency_WithCancellation

Author: Zettat123

tests/integration/actions_concurrency_test.go

@@ -14,6 +14,7 @@ import (
 	"code.gitea.io/gitea/models/unittest"
 	user_model "code.gitea.io/gitea/models/user"
 	api "code.gitea.io/gitea/modules/structs"
+
 	"github.com/stretchr/testify/assert"
 )
 
@@ -23,7 +24,7 @@ func TestWorkflowConcurrency_NoCancellation(t *testing.T) {
 		session := loginUser(t, user2.Name)
 		token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteRepository, auth_model.AccessTokenScopeWriteUser)
 
-		apiRepo := createActionsTestRepo(t, token, "actions-download-task-logs", false)
+		apiRepo := createActionsTestRepo(t, token, "actions-concurrency", false)
 		repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: apiRepo.ID})
 		runner := newMockRunner()
 		runner.registerAsRepoRunner(t, user2.Name, repo.Name, "mock-runner", []string{"ubuntu-latest"})
@@ -126,3 +127,87 @@ jobs:
 		doAPIDeleteRepository(httpContext)(t)
 	})
 }
+
+func TestWorkflowConcurrency_WithCancellation(t *testing.T) {
+	onGiteaRun(t, func(t *testing.T, u *url.URL) {
+		user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
+		session := loginUser(t, user2.Name)
+		token := getTokenForLoggedInUser(t, session, auth_model.AccessTokenScopeWriteRepository, auth_model.AccessTokenScopeWriteUser)
+
+		apiRepo := createActionsTestRepo(t, token, "actions-concurrency", false)
+		repo := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: apiRepo.ID})
+		runner := newMockRunner()
+		runner.registerAsRepoRunner(t, user2.Name, repo.Name, "mock-runner", []string{"ubuntu-latest"})
+
+		req := NewRequestWithJSON(t, "POST",
+			fmt.Sprintf("/api/v1/repos/%s/%s/actions/variables/qwe", user2.Name, repo.Name), &api.CreateVariableOption{
+				Value: "abc123",
+			}).
+			AddTokenAuth(token)
+		MakeRequest(t, req, http.StatusNoContent)
+
+		wf1TreePath := ".gitea/workflows/concurrent-workflow-1.yml"
+		wf1FileContent := `name: concurrent-workflow-1
+on: 
+  push:
+    paths:
+      - '.gitea/workflows/concurrent-workflow-1.yml'
+concurrency:
+  group: workflow-main-abc123
+jobs:
+  wf1-job:
+    runs-on: ubuntu-latest
+    steps:
+      - run: echo 'job from workflow1'
+`
+		wf2TreePath := ".gitea/workflows/concurrent-workflow-2.yml"
+		wf2FileContent := `name: concurrent-workflow-2
+on: 
+  push:
+    paths:
+      - '.gitea/workflows/concurrent-workflow-2.yml'
+concurrency:
+  group: workflow-${{ github.ref_name }}-${{ vars.qwe }}
+  cancel-in-progress: ${{ github.ref_name == 'main' }}
+jobs:
+  wf2-job:
+    runs-on: ubuntu-latest
+    steps:
+      - run: echo 'job from workflow2'
+`
+
+		// push workflow1
+		opts1 := getWorkflowCreateFileOptions(user2, repo.DefaultBranch, fmt.Sprintf("create %s", wf1TreePath), wf1FileContent)
+		createWorkflowFile(t, token, user2.Name, repo.Name, wf1TreePath, opts1)
+		// fetch the task of workflow1
+		task1 := runner.fetchTask(t)
+		actionTask1 := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionTask{ID: task1.Id})
+		actionRunJob1 := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRunJob{ID: actionTask1.JobID})
+		actionRun1 := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRun{ID: actionRunJob1.RunID})
+		assert.Equal(t, "workflow-main-abc123", actionRun1.ConcurrencyGroup)
+		assert.Equal(t, "concurrent-workflow-1.yml", actionRun1.WorkflowID)
+		assert.False(t, actionRun1.ConcurrencyCancel)
+		assert.True(t, actionRun1.Status.IsRunning())
+
+		// push workflow2
+		opts2 := getWorkflowCreateFileOptions(user2, repo.DefaultBranch, fmt.Sprintf("create %s", wf2TreePath), wf2FileContent)
+		createWorkflowFile(t, token, user2.Name, repo.Name, wf2TreePath, opts2)
+		// fetch the task of workflow2
+		task2 := runner.fetchTask(t)
+		actionTask2 := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionTask{ID: task2.Id})
+		actionRunJob2 := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRunJob{ID: actionTask2.JobID})
+		actionRun2 := unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRun{ID: actionRunJob2.RunID})
+		assert.Equal(t, "workflow-main-abc123", actionRun2.ConcurrencyGroup)
+		assert.Equal(t, "concurrent-workflow-2.yml", actionRun2.WorkflowID)
+		assert.True(t, actionRun2.ConcurrencyCancel)
+		assert.True(t, actionRun2.Status.IsRunning())
+
+		// after pushing workflow2, the status of the run of workflow1 should be "cancelled"
+		// fetch the last workflow (workflow2 or workflow3)
+		actionRun1 = unittest.AssertExistsAndLoadBean(t, &actions_model.ActionRun{ID: actionRunJob1.RunID})
+		assert.True(t, actionRun1.Status.IsCancelled())
+
+		httpContext := NewAPITestContext(t, user2.Name, repo.Name, auth_model.AccessTokenScopeWriteRepository)
+		doAPIDeleteRepository(httpContext)(t)
+	})
+}