mirror of
https://github.com/go-i2p/go-sam-go.git
synced 2025-06-07 09:03:18 -04:00
add session tests to common
This commit is contained in:
eyedeekay
@ -378,7 +378,14 @@ func (f *I2PConfig) Print() []string {
|
||||
settings = append(settings, f.collectLeaseSetSettings()...)
|
||||
// Collect access control settings
|
||||
settings = append(settings, f.collectAccessSettings()...)
|
||||
return settings
|
||||
// Filter out empty strings to prevent duplicates
|
||||
var filtered []string
|
||||
for _, config := range settings {
|
||||
if strings.TrimSpace(config) != "" {
|
||||
filtered = append(filtered, config)
|
||||
}
|
||||
}
|
||||
return filtered
|
||||
}
|
||||
|
||||
// Accesslisttype returns the I2CP access list configuration string based on the AccessListType setting
|
||||
|
||||
@ -168,8 +168,7 @@ func TestTunnelConfigNoDuplicates(t *testing.T) {
|
||||
OutQuantity: 2,
|
||||
}
|
||||
|
||||
config := cfg.TunnelConfig()
|
||||
params := strings.Split(config, " ")
|
||||
params := cfg.Print()
|
||||
|
||||
// Verify no duplicate parameters
|
||||
seen := make(map[string]bool)
|
||||
@ -189,8 +188,15 @@ func TestTunnelConfigNoDuplicates(t *testing.T) {
|
||||
}
|
||||
|
||||
for _, expected := range expectedParams {
|
||||
if !strings.Contains(config, expected) {
|
||||
t.Errorf("Missing expected parameter: %s", expected)
|
||||
found := false
|
||||
for _, param := range params {
|
||||
if strings.Contains(param, expected) {
|
||||
found = true
|
||||
}
|
||||
}
|
||||
if !found {
|
||||
t.Errorf("Expected parameter not found: %s", expected)
|
||||
}
|
||||
}
|
||||
t.Logf("Tunnel configuration parameters: %v", params)
|
||||
}
|
||||
|
||||
358
common/session_test.go
Normal file
358
common/session_test.go
Normal file
@ -0,0 +1,358 @@
|
||||
package common
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/go-i2p/i2pkeys"
|
||||
)
|
||||
|
||||
func TestNewGenericSession(t *testing.T) {
|
||||
// Create SAM connection
|
||||
sam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer sam.Close()
|
||||
|
||||
// Generate keys for testing
|
||||
_, err = sam.NewKeys()
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to generate keys: %v", err)
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
style string
|
||||
id string
|
||||
extras []string
|
||||
wantErr bool
|
||||
}{
|
||||
{
|
||||
name: "stream session",
|
||||
style: SESSION_STYLE_STREAM,
|
||||
id: "test-stream-session",
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "datagram session",
|
||||
style: SESSION_STYLE_DATAGRAM,
|
||||
id: "test-datagram-session",
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "raw session",
|
||||
style: SESSION_STYLE_RAW,
|
||||
id: "test-raw-session",
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "invalid style",
|
||||
style: "INVALID",
|
||||
id: "test-invalid-session",
|
||||
extras: []string{},
|
||||
wantErr: true,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
// Create new SAM connection for each test
|
||||
testSam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer testSam.Close()
|
||||
|
||||
// Generate unique keys for each test
|
||||
testKeys, err := testSam.NewKeys()
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to generate keys: %v", err)
|
||||
}
|
||||
|
||||
session, err := testSam.NewGenericSession(tt.style, tt.id, testKeys, tt.extras)
|
||||
if (err != nil) != tt.wantErr {
|
||||
t.Errorf("NewGenericSession() error = %v, wantErr %v", err, tt.wantErr)
|
||||
return
|
||||
}
|
||||
|
||||
if !tt.wantErr {
|
||||
if session == nil {
|
||||
t.Error("NewGenericSession() returned nil session")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify session properties
|
||||
if session.ID() != tt.id {
|
||||
t.Errorf("Session ID = %v, want %v", session.ID(), tt.id)
|
||||
}
|
||||
|
||||
if session.Keys().String() != testKeys.String() {
|
||||
t.Error("Session keys don't match expected keys")
|
||||
}
|
||||
|
||||
// Clean up session
|
||||
session.Close()
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewGenericSessionWithSignature(t *testing.T) {
|
||||
// Create SAM connection
|
||||
sam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer sam.Close()
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
style string
|
||||
id string
|
||||
sigType string
|
||||
extras []string
|
||||
wantErr bool
|
||||
}{
|
||||
{
|
||||
name: "ed25519 signature",
|
||||
style: SESSION_STYLE_STREAM,
|
||||
id: "test-ed25519-session",
|
||||
sigType: SIG_EdDSA_SHA512_Ed25519,
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "dsa signature",
|
||||
style: SESSION_STYLE_STREAM,
|
||||
id: "test-dsa-session",
|
||||
sigType: SIG_DSA_SHA1,
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "ecdsa p256 signature",
|
||||
style: SESSION_STYLE_STREAM,
|
||||
id: "test-ecdsa-p256-session",
|
||||
sigType: SIG_ECDSA_SHA256_P256,
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
// Create new SAM connection for each test
|
||||
testSam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer testSam.Close()
|
||||
|
||||
// Generate keys with specific signature type
|
||||
var testKeys i2pkeys.I2PKeys
|
||||
if strings.Contains(tt.sigType, "EdDSA") {
|
||||
testKeys, err = testSam.NewKeys("EdDSA_SHA512_Ed25519")
|
||||
} else if strings.Contains(tt.sigType, "DSA") {
|
||||
testKeys, err = testSam.NewKeys("DSA_SHA1")
|
||||
} else if strings.Contains(tt.sigType, "ECDSA_SHA256_P256") {
|
||||
testKeys, err = testSam.NewKeys("ECDSA_SHA256_P256")
|
||||
} else {
|
||||
testKeys, err = testSam.NewKeys()
|
||||
}
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to generate keys: %v", err)
|
||||
}
|
||||
|
||||
session, err := testSam.NewGenericSessionWithSignature(tt.style, tt.id, testKeys, tt.sigType, tt.extras)
|
||||
if (err != nil) != tt.wantErr {
|
||||
t.Errorf("NewGenericSessionWithSignature() error = %v, wantErr %v", err, tt.wantErr)
|
||||
return
|
||||
}
|
||||
|
||||
if !tt.wantErr {
|
||||
if session == nil {
|
||||
t.Error("NewGenericSessionWithSignature() returned nil session")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify session properties
|
||||
if session.ID() != tt.id {
|
||||
t.Errorf("Session ID = %v, want %v", session.ID(), tt.id)
|
||||
}
|
||||
|
||||
// Clean up session
|
||||
session.Close()
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewGenericSessionWithSignatureAndPorts(t *testing.T) {
|
||||
// Create SAM connection
|
||||
sam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer sam.Close()
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
style string
|
||||
id string
|
||||
from string
|
||||
to string
|
||||
sigType string
|
||||
extras []string
|
||||
wantErr bool
|
||||
}{
|
||||
{
|
||||
name: "with ports",
|
||||
style: SESSION_STYLE_STREAM,
|
||||
id: "test-ports-session",
|
||||
from: "8080",
|
||||
to: "9090",
|
||||
sigType: SIG_EdDSA_SHA512_Ed25519,
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "default ports",
|
||||
style: SESSION_STYLE_STREAM,
|
||||
id: "test-default-ports-session",
|
||||
from: "0",
|
||||
to: "0",
|
||||
sigType: SIG_EdDSA_SHA512_Ed25519,
|
||||
extras: []string{},
|
||||
wantErr: false,
|
||||
},
|
||||
{
|
||||
name: "with extras",
|
||||
style: SESSION_STYLE_STREAM,
|
||||
id: "test-extras-session",
|
||||
from: "0",
|
||||
to: "0",
|
||||
sigType: SIG_EdDSA_SHA512_Ed25519,
|
||||
extras: []string{"inbound.length=2", "outbound.length=2"},
|
||||
wantErr: false,
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
// Create new SAM connection for each test
|
||||
testSam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer testSam.Close()
|
||||
|
||||
// Generate unique keys for each test
|
||||
testKeys, err := testSam.NewKeys()
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to generate keys: %v", err)
|
||||
}
|
||||
|
||||
session, err := testSam.NewGenericSessionWithSignatureAndPorts(tt.style, tt.id, tt.from, tt.to, testKeys, tt.sigType, tt.extras)
|
||||
if (err != nil) != tt.wantErr {
|
||||
t.Errorf("NewGenericSessionWithSignatureAndPorts() error = %v, wantErr %v", err, tt.wantErr)
|
||||
return
|
||||
}
|
||||
|
||||
if !tt.wantErr {
|
||||
if session == nil {
|
||||
t.Error("NewGenericSessionWithSignatureAndPorts() returned nil session")
|
||||
return
|
||||
}
|
||||
|
||||
// Verify session properties
|
||||
if session.ID() != tt.id {
|
||||
t.Errorf("Session ID = %v, want %v", session.ID(), tt.id)
|
||||
}
|
||||
|
||||
if session.Keys().String() != testKeys.String() {
|
||||
t.Error("Session keys don't match expected keys")
|
||||
}
|
||||
|
||||
// Verify port settings if available
|
||||
baseSession, ok := session.(*BaseSession)
|
||||
if ok {
|
||||
if tt.from != "0" && baseSession.From() != tt.from {
|
||||
t.Errorf("Session FromPort = %v, want %v", baseSession.From(), tt.from)
|
||||
}
|
||||
if tt.to != "0" && baseSession.To() != tt.to {
|
||||
t.Errorf("Session ToPort = %v, want %v", baseSession.To(), tt.to)
|
||||
}
|
||||
}
|
||||
|
||||
// Clean up session
|
||||
session.Close()
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestSessionCreationErrors(t *testing.T) {
|
||||
// Create SAM connection
|
||||
sam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer sam.Close()
|
||||
|
||||
// Generate keys for testing
|
||||
keys, err := sam.NewKeys()
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to generate keys: %v", err)
|
||||
}
|
||||
|
||||
t.Run("duplicate session ID", func(t *testing.T) {
|
||||
// Create first session
|
||||
testSam1, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer testSam1.Close()
|
||||
|
||||
session1, err := testSam1.NewGenericSession(SESSION_STYLE_STREAM, "duplicate-id", keys, []string{})
|
||||
if err != nil {
|
||||
t.Fatalf("Failed to create first session: %v", err)
|
||||
}
|
||||
defer session1.Close()
|
||||
|
||||
// Try to create second session with same ID
|
||||
testSam2, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer testSam2.Close()
|
||||
|
||||
_, err = testSam2.NewGenericSession(SESSION_STYLE_STREAM, "duplicate-id", keys, []string{})
|
||||
if err == nil {
|
||||
t.Error("Expected error for duplicate session ID")
|
||||
}
|
||||
|
||||
if !strings.Contains(err.Error(), "Duplicate") {
|
||||
t.Errorf("Expected duplicate error, got: %v", err)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("invalid keys", func(t *testing.T) {
|
||||
testSam, err := NewSAM("127.0.0.1:7656")
|
||||
if err != nil {
|
||||
t.Skipf("Failed to connect to SAM bridge: %v", err)
|
||||
}
|
||||
defer testSam.Close()
|
||||
|
||||
// Create invalid keys
|
||||
invalidKeys := i2pkeys.NewKeys(i2pkeys.I2PAddr("invalid"), "invalid")
|
||||
|
||||
_, err = testSam.NewGenericSession(SESSION_STYLE_STREAM, "invalid-keys-session", invalidKeys, []string{})
|
||||
if err == nil {
|
||||
t.Error("Expected error for invalid keys")
|
||||
}
|
||||
})
|
||||
}
|
||||
Reference in New Issue
Block a user