updating tests

Author: digitalghost-dev

cli_test.go

@@ -173,7 +173,7 @@ func TestRunCLI(t *testing.T) {
 		{
 			name:           "Latest Flag",
 			args:           []string{"-l"},
-			expectedOutput: "Latest Docker image version: v0.12.0\nLatest release tag: v0.12.0\n",
+			expectedOutput: "Latest Docker image version: v0.12.1\nLatest release tag: v0.12.1\n",
 			expectedCode:   0,
 		},
 	}

cmd/ability_test.go

@@ -3,6 +3,7 @@ package cmd
 import (
 	"bytes"
 	"fmt"
+	"github.com/digitalghost-dev/poke-cli/styling"
 	"log"
 	"os"
 	"strings"
@@ -61,7 +62,7 @@ func TestAbilityCommand(t *testing.T) {
 		{
 			name:           "Valid Execution",
 			args:           []string{"ability", "stench"},
-			expectedOutput: stripANSI("Stench\nEffect: Has a 10% chance of making target Pokémon flinch with each hit."),
+			expectedOutput: styling.StripANSI("Stench\nEffect: Has a 10% chance of making target Pokémon flinch with each hit."),
 			expectError:    false,
 		},
 	}

cmd/types_test.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"github.com/digitalghost-dev/poke-cli/styling"
 	"github.com/stretchr/testify/assert"
 	"testing"
 )
@@ -50,9 +51,9 @@ func TestModelView_DisplayTable(t *testing.T) {
 
 	// Construct the expected output exactly as `View()` should render it
 	expectedOutput := "Select a type!\n" +
-		typesTableBorder.Render(m.table.View()) +
+		styling.TypesTableBorder.Render(m.table.View()) +
 		"\n" +
-		keyMenu.Render("↑ (move up) • ↓ (move down)\nctrl+c | esc (quit) • enter (select)")
+		styling.KeyMenu.Render("↑ (move up) • ↓ (move down)\nctrl+c | esc (quit) • enter (select)")
 
 	output := m.View()
 

cmd/validateargs_test.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"github.com/digitalghost-dev/poke-cli/styling"
 	"strings"
 	"testing"
 )
@@ -21,7 +22,7 @@ func TestValidatePokemonArgs(t *testing.T) {
 	}
 
 	// Strip ANSI codes for comparison
-	actualError := stripANSI(err.Error())
+	actualError := styling.StripANSI(err.Error())
 	expectedError = strings.TrimSpace(expectedError)
 
 	if actualError != expectedError {

connections/connection_test.go

@@ -2,6 +2,7 @@ package connections
 
 import (
 	"encoding/json"
+	"github.com/digitalghost-dev/poke-cli/structs"
 	"github.com/stretchr/testify/assert"
 	"net/http"
 	"net/http/httptest"
@@ -30,7 +31,7 @@ func TestApiCallSetup(t *testing.T) {
 }
 
 func TestAbilityApiCall(t *testing.T) {
-	expectedAbility := AbilityJSONStruct{
+	expectedAbility := structs.AbilityJSONStruct{
 		Name: "Unaware",
 	}
 
@@ -47,7 +48,7 @@ func TestAbilityApiCall(t *testing.T) {
 }
 
 func TestPokemonApiCall(t *testing.T) {
-	expectedPokemon := PokemonJSONStruct{
+	expectedPokemon := structs.PokemonJSONStruct{
 		Name:   "pikachu",
 		ID:     25,
 		Weight: 60,
@@ -84,7 +85,7 @@ func TestPokemonApiCall(t *testing.T) {
 
 // TestTypesApiCall - Test for the TypesApiCall function
 func TestTypesApiCall(t *testing.T) {
-	expectedTypes := TypesJSONStruct{
+	expectedTypes := structs.TypesJSONStruct{
 		Name: "electric",
 		ID:   13,
 		Pokemon: []struct {

flags/abilityflagset_test.go

@@ -2,6 +2,7 @@ package flags
 
 import (
 	"bytes"
+	"github.com/digitalghost-dev/poke-cli/styling"
 	"github.com/stretchr/testify/assert"
 	"os"
 	"reflect"
@@ -58,7 +59,7 @@ Pokemon with Stench
  7. Skuntank                    8. Trubbish                    9. Garbodor                   
 10. Garbodor-Gmax              
 `
-	actualOutput := stripANSI(output.String())
+	actualOutput := styling.StripANSI(output.String())
 
 	assert.True(t, strings.Contains(actualOutput, expectedOutput), "Output should contain Pokémon with the ability")
 }

flags/pokemonflagset_test.go

@@ -3,6 +3,7 @@ package flags
 import (
 	"bytes"
 	"fmt"
+	"github.com/digitalghost-dev/poke-cli/styling"
 	"github.com/stretchr/testify/assert"
 	"os"
 	"reflect"
@@ -73,7 +74,7 @@ Hidden Ability: Chlorophyll
 `
 
 	// Assert the actual output matches the expected output
-	actualOutput := stripANSI(output.String())
+	actualOutput := styling.StripANSI(output.String())
 
 	assert.Equal(t, expectedOutput, actualOutput, "Output should contain data for the abilities flag")
 }
@@ -103,7 +104,7 @@ func TestImageFlag(t *testing.T) {
 	assert.NoError(t, err)
 
 	// Validate that the output contains some expected patterns
-	actualOutput := stripANSI(output.String())
+	actualOutput := styling.StripANSI(output.String())
 
 	// Since the output is an ASCII image, we can't hardcode the expected output,
 	// but we can check that it contains some general expected structure
@@ -177,7 +178,7 @@ Total      318
 `
 
 	// Assert output contains the expected header and typing information
-	actualOutput := stripANSI(output.String())
+	actualOutput := styling.StripANSI(output.String())
 
 	assert.Equal(t, expectedOutput, actualOutput, "Output should contain data for the stats flag")
 
@@ -215,7 +216,7 @@ Type 2: Poison
 `
 
 	// Assert output contains the expected header and typing information
-	actualOutput := stripANSI(output.String())
+	actualOutput := styling.StripANSI(output.String())
 
 	assert.Equal(t, expectedOutput, actualOutput, "Output should contain data for the types flag")
 }