increasing test coverage

Author: digitalghost-dev

cmd/berry/berry_test.go

@@ -10,6 +10,8 @@ import (
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
 	"github.com/charmbracelet/x/exp/teatest"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestBerryCommand(t *testing.T) {
@@ -251,3 +253,15 @@ func TestTableInitialSelection(t *testing.T) {
 		t.Errorf("Expected initial selectedOption to be 'Aguav', got %q", final.selectedOption)
 	}
 }
+
+func TestBerryCommandValidationError(t *testing.T) {
+	originalArgs := os.Args
+	defer func() { os.Args = originalArgs }()
+
+	// Set os.Args with extra argument to trigger validation error
+	os.Args = []string{"poke-cli", "berry", "cheri", "extra-arg"}
+
+	output, err := BerryCommand()
+	require.Error(t, err, "TypesCommand should return error for invalid args")
+	assert.Contains(t, output, "Error", "Output should contain error message")
+}

cmd/search/search_test.go

@@ -124,3 +124,37 @@ func TestModelQuit(t *testing.T) {
 		t.Errorf("Expected tea.Quit command, but got nil")
 	}
 }
+
+func TestSearchCommandValidationError(t *testing.T) {
+	originalArgs := os.Args
+	defer func() { os.Args = originalArgs }()
+
+	// Set os.Args with extra argument to trigger validation error
+	os.Args = []string{"poke-cli", "search", "pokemon", "extra-arg"}
+
+	err := SearchCommand()
+	assert.Error(t, err, "SearchCommand should return error for invalid args")
+}
+
+func TestModelViewQuitting(t *testing.T) {
+	m := Model{Quitting: true}
+	view := m.View()
+	assert.Contains(t, view, "Quitting search", "View should show quitting message")
+}
+
+func TestModelViewShowResults(t *testing.T) {
+	m := Model{
+		ShowResults:   true,
+		SearchResults: "Test Results",
+	}
+	view := m.View()
+	// View calls RenderInput when ShowResults is true
+	assert.NotEmpty(t, view, "View should render results")
+}
+
+func TestModelViewNotChosen(t *testing.T) {
+	m := Model{Chosen: false}
+	view := m.View()
+	// View calls RenderSelection when not chosen
+	assert.Contains(t, view, "Search for a resource", "View should show selection prompt")
+}

cmd/types/types_test.go

@@ -12,6 +12,7 @@ import (
 	"github.com/digitalghost-dev/poke-cli/cmd/utils"
 	"github.com/digitalghost-dev/poke-cli/styling"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestTypesCommand(t *testing.T) {
@@ -177,3 +178,15 @@ func TestTypeSelection(t *testing.T) {
 		t.Errorf("Expected selectedOption to be 'Water', got '%s'", final.selectedOption)
 	}
 }
+
+func TestTypesCommandValidationError(t *testing.T) {
+	originalArgs := os.Args
+	defer func() { os.Args = originalArgs }()
+
+	// Set os.Args with extra argument to trigger validation error
+	os.Args = []string{"poke-cli", "types", "fire", "extra-arg"}
+
+	output, err := TypesCommand()
+	require.Error(t, err, "TypesCommand should return error for invalid args")
+	assert.Contains(t, output, "Error", "Output should contain error message")
+}