part of increasing test coverage above 50% (#66)

Author: digitalghost-dev

cmd/pokemon_test.go

@@ -1,7 +1,9 @@
 package cmd
 
 import (
+	"bytes"
 	"github.com/stretchr/testify/assert"
+	"os"
 	"testing"
 )
 
@@ -49,3 +51,38 @@ func TestValidatePokemonArgs_TooManyArgs(t *testing.T) {
 		assert.NotEqual(t, expectedError, err.Error())
 	}
 }
+
+func TestPokemonCommand(t *testing.T) {
+	// Capture standard output
+	var output bytes.Buffer
+	stdout := os.Stdout
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	// Set up test arguments (focus only on Pokémon name and command)
+	os.Args = []string{"poke-cli", "pokemon", "bulbasaur"}
+
+	// Call the function
+	PokemonCommand()
+
+	// Close and restore stdout
+	if closeErr := w.Close(); closeErr != nil {
+		t.Fatalf("Failed to close pipe writer: %v", closeErr)
+	}
+	os.Stdout = stdout
+
+	_, readErr := output.ReadFrom(r)
+	if readErr != nil {
+		t.Fatalf("Failed to read from pipe: %v", readErr)
+	}
+
+	// Define expected output based on actual API response
+	expectedName := "Bulbasaur"
+	expectedID := "1"
+
+	// Assert output contains expected Pokémon details
+	assert.Contains(t, output.String(), "Your selected Pokémon:", "Output should contain Pokémon details header")
+	assert.Contains(t, output.String(), expectedName, "Output should contain the Pokémon name")
+	assert.Contains(t, output.String(), "National Pokédex #:", "Output should contain Pokémon ID header")
+	assert.Contains(t, output.String(), expectedID, "Output should contain the Pokémon ID")
+}

cmd/styles_test.go

@@ -0,0 +1,22 @@
+package cmd
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestGetTypeColor(t *testing.T) {
+	// Test known types
+	for typeName, expectedColor := range colorMap {
+		t.Run(typeName, func(t *testing.T) {
+			color := getTypeColor(typeName)
+			assert.Equal(t, expectedColor, color, "Expected color for type %s to be %s", typeName, expectedColor)
+		})
+	}
+
+	// Test unknown type
+	t.Run("unknown type", func(t *testing.T) {
+		color := getTypeColor("unknown")
+		assert.Equal(t, "", color, "Expected color for unknown type to be an empty string")
+	})
+}

cmd/types_test.go

@@ -29,3 +29,27 @@ func TestValidateTypesArgs_TooManyArgs(t *testing.T) {
 		assert.NotEqual(t, expectedError, err.Error())
 	}
 }
+
+func TestModelInit(t *testing.T) {
+	m := model{}
+	result := m.Init()
+
+	assert.Nil(t, result, "Expected Init() to return nil")
+}
+
+func TestModelView_SelectedOption(t *testing.T) {
+	m := model{selectedOption: "someOption"}
+
+	output := m.View()
+
+	assert.Equal(t, "", output, "Expected output to be an empty string when selectedOption is set")
+}
+
+func TestModelView_DisplayTable(t *testing.T) {
+	m := model{selectedOption: ""}
+	expectedOutput := "Select a type! Hit 'Q' or 'CTRL-C' to quit.\n" + typesTableBorder.Render(m.table.View()) + "\n"
+
+	output := m.View()
+
+	assert.Equal(t, expectedOutput, output, "Expected View output to include table view")
+}

flags/pokemonflagset_test.go

@@ -1,7 +1,9 @@
 package flags
 
 import (
+	"bytes"
 	"github.com/stretchr/testify/assert"
+	"os"
 	"testing"
 )
 
@@ -30,3 +32,76 @@ func TestSetupPokemonFlagSet(t *testing.T) {
 	assert.NotNil(t, shortAbilitiesFlag, "Short abilities flag should not be nil")
 	assert.Equal(t, false, *shortAbilitiesFlag, "Short abilities flag name should be 'a'")
 }
+
+func TestAbilitiesFlag(t *testing.T) {
+	// Capture standard output
+	var output bytes.Buffer
+	stdout := os.Stdout
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	// Call the function with a known Pokémon (e.g., bulbasaur)
+	err := AbilitiesFlag("pokemon", "bulbasaur")
+
+	// Close and restore stdout
+	if closeErr := w.Close(); closeErr != nil {
+		t.Fatalf("Failed to close pipe writer: %v", closeErr)
+	}
+	os.Stdout = stdout
+
+	_, readErr := output.ReadFrom(r)
+	if readErr != nil {
+		t.Fatalf("Failed to read from pipe: %v", readErr)
+	}
+
+	// Assert no errors occurred during execution
+	assert.NoError(t, err)
+
+	// Define the expected output based on the API response
+	expectedOutput := `─────────
+Abilities
+Ability 1: overgrow
+Hidden Ability: chlorophyll
+`
+
+	// Assert the actual output matches the expected output
+	assert.Contains(t, output.String(), "Abilities", "Output should contain 'Abilities'")
+	assert.Contains(t, output.String(), "Ability 1: overgrow", "Output should contain 'Ability 1: overgrow'")
+	assert.Contains(t, output.String(), "Hidden Ability: chlorophyll", "Output should contain 'Ability 2: chlorophyll'")
+	assert.Equal(t, expectedOutput, output.String(), "Output does not match the expected formatting")
+}
+
+func TestTypesFlag(t *testing.T) {
+	// Capture standard output
+	var output bytes.Buffer
+	stdout := os.Stdout
+	r, w, _ := os.Pipe()
+	os.Stdout = w
+
+	// Call the TypesFlag function with a valid Pokémon
+	err := TypesFlag("pokemon", "bulbasaur")
+
+	// Close and restore stdout
+	if closeErr := w.Close(); closeErr != nil {
+		t.Fatalf("Failed to close pipe writer: %v", closeErr)
+	}
+	os.Stdout = stdout
+
+	_, readErr := output.ReadFrom(r)
+	if readErr != nil {
+		t.Fatalf("Failed to read from pipe: %v", readErr)
+	}
+
+	// Assert no errors occurred
+	assert.NoError(t, err, "TypesFlag should not return an error for a valid Pokémon")
+
+	// Define expected output components
+	expectedHeader := "Typing"
+	expectedType1 := "Type 1: grass"
+	expectedType2 := "Type 2: poison"
+
+	// Assert output contains the expected header and typing information
+	assert.Contains(t, output.String(), expectedHeader, "Output should contain the 'Typing' header")
+	assert.Contains(t, output.String(), expectedType1, "Output should contain the Pokémon's first type")
+	assert.Contains(t, output.String(), expectedType2, "Output should contain the Pokémon's second type")
+}