✅ Add update test on multiple checks

Delni · Jan 4, 2023 · abcdca5 · abcdca5
1 parent 6f2f22d
commit abcdca5
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 3 deletions.
diff --git a/src/preflight/update.go b/src/preflight/update.go
@@ -2,7 +2,6 @@ package preflight
 
 import (
 	"fmt"
-	"os"
 	"os/exec"
 	"preflight/src/io"
 	"runtime"
@@ -18,7 +17,7 @@ func (p PreflightModel) runCheckpoint() tea.Cmd {
 	interpreter, err := io.GetInterpreterCommand(runtime.GOOS)
 	if err != nil {
 		fmt.Println(err)
-		os.Exit(1)
+		return tea.Quit
 	}
 	interpreterArg := interpreter.InterpreterArgs
 	if checkpoint.UseInteractive {

diff --git a/src/preflight/update_test.go b/src/preflight/update_test.go
@@ -1,6 +1,9 @@
 package preflight
 
-import "testing"
+import (
+	"preflight/src/systemcheck"
+	"testing"
+)
 
 func TestUpdateInternalStateFinished(t *testing.T) {
 	p := makeTestModel()
@@ -11,3 +14,25 @@ func TestUpdateInternalStateFinished(t *testing.T) {
 		t.Errorf("got %+v", ans)
 	}
 }
+
+func TestUpdateInternalStateContinue(t *testing.T) {
+	p := makeTestModel()
+	p.checks = []systemcheck.SystemCheck{
+		testSystemCheck,
+		testSystemCheck,
+	}
+
+	ans, _ := p.UpdateInternalState(systemCheckMsg{check: true})
+
+	if ans.done != false {
+		t.Errorf("got %+v", ans.done)
+	}
+
+	if ans.activeIndex != 1 {
+		t.Errorf("got %+v", ans.activeIndex)
+	}
+
+	if ans.progress.Percent() != .5 {
+		t.Errorf("got %+v", ans.progress.Percent())
+	}
+}