Function TestCompleteJob

coderd/provisionerdserver/provisionerdserver_test.go:1508–3643

(t *testing.T)

}

func TestCompleteJob(t *testing.T) {
	t.Parallel()
	ctx := context.Background()
	t.Run("NotFound", func(t *testing.T) {
		t.Parallel()
		srv, _, _, _ := setup(t, false, nil)
		_, err := srv.CompleteJob(ctx, &proto.CompletedJob{
			JobId: "hello",
		})
		require.ErrorContains(t, err, "invalid UUID")

		_, err = srv.CompleteJob(ctx, &proto.CompletedJob{
			JobId: uuid.NewString(),
		})
		require.ErrorContains(t, err, "no rows in result set")
	})
	// This test prevents runners from updating jobs they don't own!
	t.Run("NotOwner", func(t *testing.T) {
		t.Parallel()
		srv, db, _, pd := setup(t, false, nil)
		user := dbgen.User(t, db, database.User{})
		version := dbgen.TemplateVersion(t, db, database.TemplateVersion{
			CreatedBy:      user.ID,
			OrganizationID: pd.OrganizationID,
			JobID:          uuid.New(),
		})
		job, err := db.InsertProvisionerJob(ctx, database.InsertProvisionerJobParams{
			ID:             version.JobID,
			Provisioner:    database.ProvisionerTypeEcho,
			StorageMethod:  database.ProvisionerStorageMethodFile,
			Type:           database.ProvisionerJobTypeTemplateVersionImport,
			OrganizationID: pd.OrganizationID,
			Input: must(json.Marshal(provisionerdserver.TemplateVersionDryRunJob{
				TemplateVersionID: version.ID,
			})),
			Tags: pd.Tags,
		})
		require.NoError(t, err)
		_, err = db.AcquireProvisionerJob(ctx, database.AcquireProvisionerJobParams{
			OrganizationID: pd.OrganizationID,
			WorkerID: uuid.NullUUID{
				UUID:  uuid.New(),
				Valid: true,
			},
			Types: []database.ProvisionerType{database.ProvisionerTypeEcho},
			StartedAt: sql.NullTime{
				Time:  dbtime.Now(),
				Valid: true,
			},
			ProvisionerTags: must(json.Marshal(job.Tags)),
		})
		require.NoError(t, err)
		_, err = srv.CompleteJob(ctx, &proto.CompletedJob{
			JobId: job.ID.String(),
		})
		require.ErrorContains(t, err, "you don't own this job")
	})