feat(mail): support IMAP SOCKS5 proxy dialing

Use IMAP_PROXY_URL to route IMAP TLS connections through SOCKS5 when configured, while preserving direct dialing when unset. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-04-27 21:37:40 +07:00
parent 2333ec3f00
commit 14f2bd7eab
2 changed files with 136 additions and 3 deletions
@@ -0,0 +1,77 @@
+package mail
+
+import (
+	"context"
+	"errors"
+	"strings"
+	"testing"
+
+	"github.com/emersion/go-imap/v2/imapclient"
+	"thuanle.me/claw-email-bridge/internal/config"
+)
+
+func TestConnectAndWatch_UsesDirectDial_WhenProxyUnset(t *testing.T) {
+	watcher := &IMAPWatcher{
+		cfg: &config.Config{
+			IMAPHost: "imap.example.com",
+			IMAPPort: "993",
+			IMAPUser: "u",
+			IMAPPass: "p",
+		},
+	}
+
+	var directCalled bool
+	watcher.dialIMAP = func(addr string, _ *imapclient.Options) (*imapclient.Client, error) {
+		directCalled = true
+		return nil, errors.New("stop")
+	}
+	watcher.dialIMAPViaProxy = func(addr, proxyURL string, _ *imapclient.Options) (*imapclient.Client, error) {
+		t.Fatalf("did not expect proxy dial, got addr=%s proxy=%s", addr, proxyURL)
+		return nil, nil
+	}
+
+	err := watcher.connectAndWatch(context.Background())
+	if err == nil || !strings.Contains(err.Error(), "stop") {
+		t.Fatalf("expected stop error, got %v", err)
+	}
+	if !directCalled {
+		t.Fatal("expected direct dial path")
+	}
+}
+
+func TestConnectAndWatch_UsesProxyDial_WhenProxySet(t *testing.T) {
+	watcher := &IMAPWatcher{
+		cfg: &config.Config{
+			IMAPHost:     "imap.example.com",
+			IMAPPort:     "993",
+			IMAPUser:     "u",
+			IMAPPass:     "p",
+			IMAPProxyURL: "socks5://127.0.0.1:1080",
+		},
+	}
+
+	var proxyCalled bool
+	watcher.dialIMAP = func(addr string, _ *imapclient.Options) (*imapclient.Client, error) {
+		t.Fatalf("did not expect direct dial, got addr=%s", addr)
+		return nil, nil
+	}
+	watcher.dialIMAPViaProxy = func(addr, proxyURL string, _ *imapclient.Options) (*imapclient.Client, error) {
+		proxyCalled = true
+		return nil, errors.New("stop")
+	}
+
+	err := watcher.connectAndWatch(context.Background())
+	if err == nil || !strings.Contains(err.Error(), "stop") {
+		t.Fatalf("expected stop error, got %v", err)
+	}
+	if !proxyCalled {
+		t.Fatal("expected proxy dial path")
+	}
+}
+
+func TestDialTLSViaSOCKS5_RejectsNonSocks5Scheme(t *testing.T) {
+	_, err := dialTLSViaSOCKS5("imap.example.com:993", "http://proxy:8080", nil)
+	if err == nil || !strings.Contains(err.Error(), "unsupported proxy scheme") {
+		t.Fatalf("expected unsupported scheme error, got %v", err)
+	}
+}