✨ : save job once it is started

gaia-app · Aug 5, 2019 · bd6f2c4 · bd6f2c4
1 parent 62566fc
commit bd6f2c4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/main/java/io/codeka/gaia/runner/StackRunner.java b/src/main/java/io/codeka/gaia/runner/StackRunner.java
@@ -110,6 +110,7 @@ public void apply(Job job, TerraformModule module, Stack stack) {
         this.jobs.put(job.getId(), job);
         job.setCliVersion(module.getCliVersion());
         job.start(JobType.RUN);
+        jobRepository.save(job);
 
         var applyScript = stackCommandBuilder.buildApplyScript(stack, module);
 
@@ -143,6 +144,7 @@ public void plan(Job job, TerraformModule module, Stack stack) {
         this.jobs.put(job.getId(), job);
         job.setCliVersion(module.getCliVersion());
         job.start(JobType.PREVIEW);
+        jobRepository.save(job);
 
         var planScript = stackCommandBuilder.buildPlanScript(stack, module);
 
@@ -191,6 +193,7 @@ public void stop(Job job, TerraformModule module, Stack stack) {
         this.jobs.put(job.getId(), job);
         job.setCliVersion(module.getCliVersion());
         job.start(JobType.STOP);
+        jobRepository.save(job);
 
         var destroyScript = stackCommandBuilder.buildDestroyScript(stack, module);
 

diff --git a/src/test/java/io/codeka/gaia/runner/StackRunnerTest.java b/src/test/java/io/codeka/gaia/runner/StackRunnerTest.java
@@ -93,7 +93,7 @@ void job_shouldBeSavedToDatabaseAfterRun() throws Exception {
         stackRunner.apply(job, module, stack);
 
         // then
-        verify(jobRepository).save(job);
+        verify(jobRepository, times(2)).save(job);
     }
 
     @Test
@@ -124,7 +124,7 @@ void plan_shouldUpdateTheStackState_whenThereIsADiffForRunningStacks() throws Ex
         stackRunner.plan(job, module, stack);
 
         // then
-        verify(jobRepository).save(job);
+        verify(jobRepository, times(2)).save(job);
 
         assertEquals(StackState.TO_UPDATE, stack.getState());
         verify(stackRepository).save(stack);
@@ -143,7 +143,7 @@ void plan_shouldNotUpdateTheStackState_whenThereIsADiffForNewStacks() throws Exc
         stackRunner.plan(job, module, stack);
 
         // then
-        verify(jobRepository).save(job);
+        verify(jobRepository, times(2)).save(job);
 
         assertEquals(StackState.NEW, stack.getState());
         verifyZeroInteractions(stackRepository);
@@ -162,7 +162,7 @@ void stop_shouldUpdateTheStackState_whenSuccessful() throws Exception {
         stackRunner.stop(job, module, stack);
 
         // then
-        verify(jobRepository).save(job);
+        verify(jobRepository, times(2)).save(job);
 
         assertEquals(StackState.STOPPED, stack.getState());
         verify(stackRepository).save(stack);
@@ -183,7 +183,7 @@ void jobShouldFail_whenFailingToStartContainer() throws Exception {
 
         // then
         assertEquals(JobStatus.FAILED, job.getStatus());
-        verify(jobRepository).save(job);
+        verify(jobRepository, times(2)).save(job);
     }
 
     @Test