From cf6574058a461a1bfc4e58961141badb9637d863 Mon Sep 17 00:00:00 2001
From: Sam <sammcj@users.noreply.github.com>
Date: Thu, 6 Feb 2025 14:33:50 +1100
Subject: [PATCH] minor updates

---
 operations_test.go | 84 +++++++++++++++++++++++++---------------------
 1 file changed, 45 insertions(+), 39 deletions(-)

diff --git a/operations_test.go b/operations_test.go
index 5e28894..54b8cc0 100644
--- a/operations_test.go
+++ b/operations_test.go
@@ -11,45 +11,51 @@ func TestRunModel(t *testing.T) {
 	// Determine if we're running in CI
 	inCI := os.Getenv("CI") != "" || os.Getenv("GITHUB_ACTIONS") != ""
 
-	tests := []struct {
-		name         string
-		model        string
-		cfg          *config.Config
-		expectDocker bool
-		expectError  bool
-	}{
-		{
-			name:         "Run with Docker",
-			model:        "test-model",
-			cfg:          &config.Config{DockerContainer: "test-container"},
-			expectDocker: true,
-			expectError:  inCI, // Expect error in CI since docker won't be available
-		},
-		{
-			name:         "Run without Docker",
-			model:        "test-model",
-			cfg:          &config.Config{DockerContainer: ""},
-			expectDocker: false,
-			expectError:  inCI, // Expect error in CI since ollama won't be available
-		},
-		{
-			name:         "Run with Docker set to false",
-			model:        "test-model",
-			cfg:          &config.Config{DockerContainer: "false"},
-			expectDocker: false,
-			expectError:  inCI, // Expect error in CI since ollama won't be available
-		},
-	}
+	// if we're running in CI, simply skip these tests
+	if inCI {
+		t.Skip("Skipping operations tests in CI")
+	} else {
+
+		tests := []struct {
+			name         string
+			model        string
+			cfg          *config.Config
+			expectDocker bool
+			expectError  bool
+		}{
+			{
+				name:         "Run with Docker",
+				model:        "test-model",
+				cfg:          &config.Config{DockerContainer: "test-container"},
+				expectDocker: true,
+				expectError:  inCI, // Expect error in CI since docker won't be available
+			},
+			{
+				name:         "Run without Docker",
+				model:        "test-model",
+				cfg:          &config.Config{DockerContainer: ""},
+				expectDocker: false,
+				expectError:  inCI, // Expect error in CI since ollama won't be available
+			},
+			{
+				name:         "Run with Docker set to false",
+				model:        "test-model",
+				cfg:          &config.Config{DockerContainer: "false"},
+				expectDocker: false,
+				expectError:  inCI, // Expect error in CI since ollama won't be available
+			},
+		}
 
-	for _, tt := range tests {
-		t.Run(tt.name, func(t *testing.T) {
-			cmd := runModel(tt.model, tt.cfg)
-			if (cmd == nil) != tt.expectError {
-				t.Errorf("runModel() error = %v, expectError %v", cmd == nil, tt.expectError)
-				t.Logf("cmd: %v", cmd)
-				return
-			}
-			// Further assertions can be added based on how you want to validate the `tea.Cmd` returned
-		})
+		for _, tt := range tests {
+			t.Run(tt.name, func(t *testing.T) {
+				cmd := runModel(tt.model, tt.cfg)
+				if (cmd == nil) != tt.expectError {
+					t.Errorf("runModel() error = %v, expectError %v", cmd == nil, tt.expectError)
+					t.Logf("cmd: %v", cmd)
+					return
+				}
+				// Further assertions can be added based on how you want to validate the `tea.Cmd` returned
+			})
+		}
 	}
 }