Merge pull request #3 from gaffo/add_job_with_state

Run::AddJobWithState

run.go

@@ -69,8 +69,7 @@ func (r *Run[OC, JC]) Init() {
 	}
 }
 
-// Add a job to the pool, this shouldn't be called once it's running
-func (r *Run[OC, JC]) AddJob(jc JC) {
+func (r *Run[OC, JC]) AddJobWithState(jc JC, state string) {
 	r.m.Lock()
 	defer r.m.Unlock()
 
@@ -79,7 +78,7 @@ func (r *Run[OC, JC]) AddJob(jc JC) {
 	j := Job[JC]{
 		Id:          id,
 		C:           jc,
-		State:       TRIGGER_STATE_NEW,
+		State:       state,
 		StateErrors: map[string][]string{},
 	}
 	j.UpdateLastEvent()
@@ -90,6 +89,11 @@ func (r *Run[OC, JC]) AddJob(jc JC) {
 	r.Jobs[id] = j
 }
 
+// Add a job to the pool, this shouldn't be called once it's running
+func (r *Run[OC, JC]) AddJob(jc JC) {
+	r.AddJobWithState(jc, TRIGGER_STATE_NEW)
+}
+
 func (r *Run[OC, JC]) NextJobForState(state string) (Job[JC], bool) {
 	r.m.Lock()
 	defer r.m.Unlock()

run_test.go

@@ -152,3 +152,18 @@ func TestRun_JobsInFlight2(t *testing.T) {
 	r.Return(j)
 	require.False(t, r.JobsInFlight())
 }
+
+func Test_AddJobWithState(t *testing.T) {
+	t.Parallel()
+	r := NewRun[MyOverallContext, MyJobContext]("job", MyOverallContext{})
+	r.AddJobWithState(MyJobContext{Count: 0}, "other_state")
+	assert.Equal(t, map[string]StatusCount{
+		"other_state": {
+			State:     "other_state",
+			Count:     1,
+			Executing: 0,
+		},
+	}, r.StatusCounts())
+	assert.Equal(t, 1, len(r.Jobs))
+	assert.Equal(t, "other_state", r.Jobs["0"].State)
+}