pizzasql/pkg/kvmanager/kvmanager_test.go

package kvmanager

import (
	"testing"
)

func TestFindAvailablePort(t *testing.T) {
	port, err := findAvailablePort()
	if err != nil {
		t.Fatalf("Failed to find available port: %v", err)
	}

	if port < 1024 || port > 65535 {
		t.Errorf("Port %d is outside valid range 1024-65535", port)
	}
}

func TestFindAvailablePortInRange(t *testing.T) {
	// Test PizzaKV range (1024-9999)
	port, err := findAvailablePortInRange(1024, 9999)
	if err != nil {
		t.Fatalf("Failed to find available port: %v", err)
	}

	if port < 1024 || port > 9999 {
		t.Errorf("Port %d is outside requested range 1024-9999", port)
	}

	// Test custom range
	port, err = findAvailablePortInRange(5000, 5100)
	if err != nil {
		t.Fatalf("Failed to find available port in custom range: %v", err)
	}

	if port < 5000 || port > 5100 {
		t.Errorf("Port %d is outside requested range 5000-5100", port)
	}
}

func TestParseFlags(t *testing.T) {
	tests := []struct {
		name     string
		input    string
		expected []string
	}{
		{
			name:     "empty string",
			input:    "",
			expected: nil,
		},
		{
			name:     "single flag",
			input:    "-iwal",
			expected: []string{"-iwal"},
		},
		{
			name:     "multiple flags",
			input:    "-iwal -port=9090",
			expected: []string{"-iwal", "-port=9090"},
		},
		{
			name:     "flags with quotes",
			input:    "-path=\"/tmp/my path\" -verbose",
			expected: []string{"-path=/tmp/my path", "-verbose"},
		},
		{
			name:     "flags with extra spaces",
			input:    "  -iwal   -port=9090  ",
			expected: []string{"-iwal", "-port=9090"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := parseFlags(tt.input)

			if len(result) != len(tt.expected) {
				t.Errorf("Expected %d args, got %d", len(tt.expected), len(result))
				return
			}

			for i := range result {
				if result[i] != tt.expected[i] {
					t.Errorf("Arg %d: expected %q, got %q", i, tt.expected[i], result[i])
				}
			}
		})
	}
}

func TestParsePort(t *testing.T) {
	tests := []struct {
		name     string
		input    string
		expected int
		wantErr  bool
	}{
		{
			name:     "valid address",
			input:    "localhost:8085",
			expected: 8085,
			wantErr:  false,
		},
		{
			name:     "valid IP address",
			input:    "127.0.0.1:9090",
			expected: 9090,
			wantErr:  false,
		},
		{
			name:     "invalid format",
			input:    "localhost",
			expected: 0,
			wantErr:  true,
		},
		{
			name:     "invalid port",
			input:    "localhost:abc",
			expected: 0,
			wantErr:  true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result, err := ParsePort(tt.input)

			if tt.wantErr {
				if err == nil {
					t.Error("Expected error but got none")
				}
				return
			}

			if err != nil {
				t.Errorf("Unexpected error: %v", err)
				return
			}

			if result != tt.expected {
				t.Errorf("Expected port %d, got %d", tt.expected, result)
			}
		})
	}
}

func TestKVInfo(t *testing.T) {
	info := &KVInfo{
		PID:  12345,
		Port: 8085,
		Addr: "localhost:8085",
	}

	if info.PID != 12345 {
		t.Errorf("Expected PID 12345, got %d", info.PID)
	}

	if info.Port != 8085 {
		t.Errorf("Expected Port 8085, got %d", info.Port)
	}

	if info.Addr != "localhost:8085" {
		t.Errorf("Expected Addr 'localhost:8085', got %s", info.Addr)
	}
}