updating tests

Author: digitalghost-dev

cli_test.go

@@ -7,6 +7,63 @@ import (
 	"testing"
 )
 
+func TestCurrentVersion(t *testing.T) {
+	tests := []struct {
+		name           string
+		version        string
+		expectedOutput string
+	}{
+		{
+			name:           "Version set by ldflags",
+			version:        "v1.0.0",
+			expectedOutput: "Version: v1.0.0\n",
+		},
+		{
+			name:           "Version set to (devel)",
+			version:        "(devel)",
+			expectedOutput: "Version: (devel)\n", // Simplified assumption
+		},
+	}
+
+	// Save the original version and restore it after tests
+	originalVersion := version
+	defer func() { version = originalVersion }()
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			// Set the version for this test case
+			version = tt.version
+
+			// Redirect stdout to capture the output
+			r, w, _ := os.Pipe()
+			oldStdout := os.Stdout
+			os.Stdout = w
+
+			// Call the function
+			currentVersion()
+
+			// Close the writer and restore stdout
+			err := w.Close()
+			if err != nil {
+				t.Fatalf("Failed to close pipe: %v", err)
+			}
+			os.Stdout = oldStdout
+
+			// Read the output from the pipe
+			var buf bytes.Buffer
+			if _, err := buf.ReadFrom(r); err != nil {
+				t.Fatalf("Failed to read from pipe: %v", err)
+			}
+
+			// Compare the output with the expected result
+			got := buf.String()
+			if got != tt.expectedOutput {
+				t.Errorf("Expected %q, got %q", tt.expectedOutput, got)
+			}
+		})
+	}
+}
+
 var ansiRegex = regexp.MustCompile(`\x1b\[[0-9;]*m`)
 
 func captureOutput(f func()) string {
@@ -110,7 +167,7 @@ func TestRunCLI(t *testing.T) {
 		{
 			name:           "Latest Flag",
 			args:           []string{"-l"},
-			expectedOutput: "Latest Docker image version: v0.9.0\nLatest release tag: v0.9.0\n",
+			expectedOutput: "Latest Docker image version: v0.9.1\nLatest release tag: v0.9.1\n",
 			expectedCode:   0,
 		},
 	}
@@ -133,3 +190,44 @@ func TestRunCLI(t *testing.T) {
 		})
 	}
 }
+
+func TestMainFunction(t *testing.T) {
+	originalExit := exit
+	defer func() { exit = originalExit }() // Restore original exit after test
+
+	tests := []struct {
+		name           string
+		args           []string
+		expectedOutput string
+		expectedCode   int
+	}{
+		{
+			name:           "Run main command",
+			args:           []string{"poke-cli"},
+			expectedOutput: "Welcome! This tool displays data related to Pokémon!",
+			expectedCode:   0,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			exitCode := 0
+			exit = func(code int) { exitCode = code }
+
+			output := captureOutput(func() {
+				os.Args = tt.args
+				main()
+			})
+
+			output = stripANSI(output)
+
+			if exitCode != tt.expectedCode {
+				t.Errorf("Expected exit code %d, got %d", tt.expectedCode, exitCode)
+			}
+
+			if !bytes.Contains([]byte(output), []byte(tt.expectedOutput)) {
+				t.Errorf("Expected output to contain %q, got %q", tt.expectedOutput, output)
+			}
+		})
+	}
+}

flags/pokemonflagset_test.go

@@ -74,8 +74,8 @@ func TestAbilitiesFlag(t *testing.T) {
 	// Define the expected output based on the API response
 	expectedOutput := `─────────
 Abilities
-Ability 1: overgrow
-Hidden Ability: chlorophyll
+Ability 1: Overgrow
+Hidden Ability: Chlorophyll
 `
 
 	// Assert the actual output matches the expected output
@@ -153,8 +153,8 @@ func TestTypesFlag(t *testing.T) {
 	// Define expected output components
 	expectedOutput := `──────
 Typing
-Type 1: grass
-Type 2: poison
+Type 1: Grass
+Type 2: Poison
 `
 
 	// Assert output contains the expected header and typing information