diff --git a/jobUpdate.go b/jobUpdate.go index 248a65d..d880c29 100644 --- a/jobUpdate.go +++ b/jobUpdate.go @@ -77,7 +77,7 @@ func (j *JobUpdate) BatchSize(size int32) *JobUpdate { // Minimum number of seconds a shard must remain in RUNNING state before considered a success. func (j *JobUpdate) WatchTime(timeout time.Duration) *JobUpdate { - j.request.Settings.MinWaitInInstanceRunningMs = int32(timeout.Seconds() * 1000) + j.request.Settings.MinWaitInInstanceRunningMs = int32(timeout.Milliseconds()) return j } diff --git a/realis_e2e_test.go b/realis_e2e_test.go index 676ed84..7977f62 100644 --- a/realis_e2e_test.go +++ b/realis_e2e_test.go @@ -822,7 +822,7 @@ func TestRealisClient_BatchAwareAutoPause(t *testing.T) { strategy := realis.JobUpdateFromAuroraTask(job.AuroraTask()). VariableBatchStrategy(true, updateGroups...). InstanceCount(6). - WatchTime(1000) + WatchTime(time.Second*1) result, err := r.StartJobUpdate(strategy, "") require.NoError(t, err)