mirror of
https://github.com/go-i2p/go-i2ptunnel.git
synced 2025-12-20 15:15:52 -05:00
Add unit tests for LoadConfig in TCP client implementation
This commit is contained in:
eyedeekay
257
lib/tcp/client/loadconfig_test.go
Normal file
257
lib/tcp/client/loadconfig_test.go
Normal file
@@ -0,0 +1,257 @@
|
|||||||
|
package tcpclient
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
i2pconv "github.com/go-i2p/go-i2ptunnel-config/lib"
|
||||||
|
i2ptunnel "github.com/go-i2p/go-i2ptunnel/lib/core"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TestLoadConfig_Success verifies LoadConfig correctly loads and applies configuration
|
||||||
|
func TestLoadConfig_Success(t *testing.T) {
|
||||||
|
// Create temporary config file
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
configPath := filepath.Join(tmpDir, "tunnel.yaml")
|
||||||
|
|
||||||
|
// YAML format requires tunnels: map with tunnel name as key
|
||||||
|
configContent := `tunnels:
|
||||||
|
test-tcp-client:
|
||||||
|
name: test-tcp-client
|
||||||
|
type: tcpclient
|
||||||
|
interface: 127.0.0.1
|
||||||
|
port: 9999
|
||||||
|
target: ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(configPath, []byte(configContent), 0644); err != nil {
|
||||||
|
t.Fatalf("Failed to create config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Create a stopped tunnel
|
||||||
|
initialConfig := i2pconv.TunnelConfig{
|
||||||
|
Name: "initial-tunnel",
|
||||||
|
Type: "tcpclient",
|
||||||
|
Interface: "0.0.0.0",
|
||||||
|
Port: 8888,
|
||||||
|
Target: "ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p",
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel, err := NewTCPClient(initialConfig, "localhost:7656")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create tunnel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Ensure tunnel is stopped
|
||||||
|
tunnel.I2PTunnelStatus = i2ptunnel.I2PTunnelStatusStopped
|
||||||
|
|
||||||
|
// Load new configuration
|
||||||
|
if err := tunnel.LoadConfig(configPath); err != nil {
|
||||||
|
t.Fatalf("LoadConfig failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify configuration was updated
|
||||||
|
if tunnel.TunnelConfig.Name != "test-tcp-client" {
|
||||||
|
t.Errorf("Expected name 'test-tcp-client', got '%s'", tunnel.TunnelConfig.Name)
|
||||||
|
}
|
||||||
|
if tunnel.TunnelConfig.Port != 9999 {
|
||||||
|
t.Errorf("Expected port 9999, got %d", tunnel.TunnelConfig.Port)
|
||||||
|
}
|
||||||
|
if tunnel.TunnelConfig.Interface != "127.0.0.1" {
|
||||||
|
t.Errorf("Expected interface '127.0.0.1', got '%s'", tunnel.TunnelConfig.Interface)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestLoadConfig_RunningTunnel verifies LoadConfig fails when tunnel is running
|
||||||
|
func TestLoadConfig_RunningTunnel(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
configPath := filepath.Join(tmpDir, "tunnel.yaml")
|
||||||
|
|
||||||
|
configContent := `tunnels:
|
||||||
|
test-tunnel:
|
||||||
|
name: test-tunnel
|
||||||
|
type: tcpclient
|
||||||
|
interface: 127.0.0.1
|
||||||
|
port: 9999
|
||||||
|
target: ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(configPath, []byte(configContent), 0644); err != nil {
|
||||||
|
t.Fatalf("Failed to create config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
initialConfig := i2pconv.TunnelConfig{
|
||||||
|
Name: "running-tunnel",
|
||||||
|
Type: "tcpclient",
|
||||||
|
Interface: "0.0.0.0",
|
||||||
|
Port: 8888,
|
||||||
|
Target: "ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p",
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel, err := NewTCPClient(initialConfig, "localhost:7656")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create tunnel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set tunnel to running state
|
||||||
|
tunnel.I2PTunnelStatus = i2ptunnel.I2PTunnelStatusRunning
|
||||||
|
|
||||||
|
// Attempt to load config should fail
|
||||||
|
err = tunnel.LoadConfig(configPath)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("Expected LoadConfig to fail for running tunnel, but it succeeded")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify error message contains expected text
|
||||||
|
if err.Error() != "cannot load config while tunnel is running - stop tunnel first" {
|
||||||
|
t.Errorf("Unexpected error message: %s", err.Error())
|
||||||
|
}
|
||||||
|
} // TestLoadConfig_InvalidFile verifies LoadConfig handles missing files gracefully
|
||||||
|
func TestLoadConfig_InvalidFile(t *testing.T) {
|
||||||
|
initialConfig := i2pconv.TunnelConfig{
|
||||||
|
Name: "test-tunnel",
|
||||||
|
Type: "tcpclient",
|
||||||
|
Interface: "0.0.0.0",
|
||||||
|
Port: 8888,
|
||||||
|
Target: "ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p",
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel, err := NewTCPClient(initialConfig, "localhost:7656")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create tunnel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel.I2PTunnelStatus = i2ptunnel.I2PTunnelStatusStopped
|
||||||
|
|
||||||
|
// Try to load non-existent file
|
||||||
|
err = tunnel.LoadConfig("/nonexistent/path/tunnel.yaml")
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("Expected LoadConfig to fail for non-existent file, but it succeeded")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestLoadConfig_WrongType verifies LoadConfig rejects config with wrong tunnel type
|
||||||
|
func TestLoadConfig_WrongType(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
configPath := filepath.Join(tmpDir, "tunnel.yaml")
|
||||||
|
|
||||||
|
// Create config with tcpserver type instead of tcpclient
|
||||||
|
configContent := `tunnels:
|
||||||
|
test-tunnel:
|
||||||
|
name: test-tunnel
|
||||||
|
type: tcpserver
|
||||||
|
interface: 127.0.0.1
|
||||||
|
port: 9999
|
||||||
|
target: localhost:80
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(configPath, []byte(configContent), 0644); err != nil {
|
||||||
|
t.Fatalf("Failed to create config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
initialConfig := i2pconv.TunnelConfig{
|
||||||
|
Name: "test-tunnel",
|
||||||
|
Type: "tcpclient",
|
||||||
|
Interface: "0.0.0.0",
|
||||||
|
Port: 8888,
|
||||||
|
Target: "ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p",
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel, err := NewTCPClient(initialConfig, "localhost:7656")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create tunnel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel.I2PTunnelStatus = i2ptunnel.I2PTunnelStatusStopped
|
||||||
|
|
||||||
|
// Try to load config with wrong type
|
||||||
|
err = tunnel.LoadConfig(configPath)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("Expected LoadConfig to fail for wrong tunnel type, but it succeeded")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify error message mentions type mismatch
|
||||||
|
expectedMsg := "config file contains tcpserver tunnel, expected tcpclient"
|
||||||
|
if err.Error() != expectedMsg {
|
||||||
|
t.Errorf("Expected error message '%s', got '%s'", expectedMsg, err.Error())
|
||||||
|
}
|
||||||
|
} // TestLoadConfig_InvalidTarget verifies LoadConfig validates target addresses
|
||||||
|
func TestLoadConfig_InvalidTarget(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
configPath := filepath.Join(tmpDir, "tunnel.yaml")
|
||||||
|
|
||||||
|
// Create config with invalid I2P address
|
||||||
|
configContent := `tunnels:
|
||||||
|
test-tunnel:
|
||||||
|
name: test-tunnel
|
||||||
|
type: tcpclient
|
||||||
|
interface: 127.0.0.1
|
||||||
|
port: 9999
|
||||||
|
target: invalid-address-not-base32
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(configPath, []byte(configContent), 0644); err != nil {
|
||||||
|
t.Fatalf("Failed to create config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
initialConfig := i2pconv.TunnelConfig{
|
||||||
|
Name: "test-tunnel",
|
||||||
|
Type: "tcpclient",
|
||||||
|
Interface: "0.0.0.0",
|
||||||
|
Port: 8888,
|
||||||
|
Target: "ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p",
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel, err := NewTCPClient(initialConfig, "localhost:7656")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create tunnel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel.I2PTunnelStatus = i2ptunnel.I2PTunnelStatusStopped
|
||||||
|
|
||||||
|
// Try to load config with invalid target
|
||||||
|
err = tunnel.LoadConfig(configPath)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("Expected LoadConfig to fail for invalid target address, but it succeeded")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestLoadConfig_PropertiesFormat verifies LoadConfig works with .properties format
|
||||||
|
func TestLoadConfig_PropertiesFormat(t *testing.T) {
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
configPath := filepath.Join(tmpDir, "tunnel.properties")
|
||||||
|
|
||||||
|
// Create .properties format config
|
||||||
|
// Properties format uses different field names: listenPort instead of port
|
||||||
|
configContent := `type=tcpclient
|
||||||
|
name=test-properties-tunnel
|
||||||
|
interface=127.0.0.1
|
||||||
|
listenPort=7777
|
||||||
|
targetDestination=ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p
|
||||||
|
`
|
||||||
|
if err := os.WriteFile(configPath, []byte(configContent), 0644); err != nil {
|
||||||
|
t.Fatalf("Failed to create config file: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
initialConfig := i2pconv.TunnelConfig{
|
||||||
|
Name: "initial-tunnel",
|
||||||
|
Type: "tcpclient",
|
||||||
|
Interface: "0.0.0.0",
|
||||||
|
Port: 8888,
|
||||||
|
Target: "ukeu3k5oycgaauneqgtnvselmt4yemvoilkln7jpvamvfx7dnkdq.b32.i2p",
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel, err := NewTCPClient(initialConfig, "localhost:7656")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to create tunnel: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
tunnel.I2PTunnelStatus = i2ptunnel.I2PTunnelStatusStopped
|
||||||
|
|
||||||
|
// Load properties config
|
||||||
|
if err := tunnel.LoadConfig(configPath); err != nil {
|
||||||
|
t.Fatalf("LoadConfig failed for .properties format: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify config was loaded
|
||||||
|
if tunnel.TunnelConfig.Port != 7777 {
|
||||||
|
t.Errorf("Expected port 7777, got %d", tunnel.TunnelConfig.Port)
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user