MyGoNavi/internal/db/custom_impl_test.go

package db

import (
	"database/sql"
	"database/sql/driver"
	"strings"
	"testing"

	"GoNavi-Wails/internal/connection"
)

const customMySQLDSNRecordingDriverName = "custom-mysql-dsn-recording"

var customMySQLDSNRecordingLastDSN string

type customMySQLDSNRecordingDriver struct{}

func (d customMySQLDSNRecordingDriver) Open(name string) (driver.Conn, error) {
	customMySQLDSNRecordingLastDSN = name
	return customMySQLDSNRecordingConn{}, nil
}

type customMySQLDSNRecordingConn struct{}

func (c customMySQLDSNRecordingConn) Prepare(query string) (driver.Stmt, error) {
	return nil, driver.ErrSkip
}

func (c customMySQLDSNRecordingConn) Close() error {
	return nil
}

func (c customMySQLDSNRecordingConn) Begin() (driver.Tx, error) {
	return nil, driver.ErrSkip
}

func init() {
	sql.Register(customMySQLDSNRecordingDriverName, customMySQLDSNRecordingDriver{})
}

func TestCustomDBConnectReportsUnsupportedODBCDriverName(t *testing.T) {
	db := &CustomDB{}

	err := db.Connect(connection.ConnectionConfig{
		Driver: "InterSystems IRIS ODBC35",
		DSN:    "Driver={InterSystems IRIS ODBC35};Server=127.0.0.1;Port=1972;Database=USER;",
	})
	if err == nil {
		t.Fatal("expected unsupported ODBC driver error, got nil")
	}

	message := err.Error()
	for _, want := range []string{
		"ODBC/JDBC",
		"Go database/sql",
		"暂不支持",
		"InterSystems IRIS",
	} {
		if !strings.Contains(message, want) {
			t.Fatalf("expected error to contain %q, got %q", want, message)
		}
	}
}

func TestCustomDBConnectReportsUnregisteredGoDriver(t *testing.T) {
	db := &CustomDB{}

	err := db.Connect(connection.ConnectionConfig{
		Driver: "not-a-registered-go-driver",
		DSN:    "demo",
	})
	if err == nil {
		t.Fatal("expected unregistered Go driver error, got nil")
	}

	message := err.Error()
	for _, want := range []string{
		"未在 GoNavi 中注册",
		"Go database/sql",
	} {
		if !strings.Contains(message, want) {
			t.Fatalf("expected error to contain %q, got %q", want, message)
		}
	}
}

func TestNormalizeMySQLRawDSNCompatibilityParamsMapsAllowMultiQueries(t *testing.T) {
	got := normalizeMySQLRawDSNCompatibilityParams(
		"root:pass@tcp(127.0.0.1:3306)/app?charset=utf8mb4&allowMultiQueries=true#debug",
	)
	if strings.Contains(got, "allowMultiQueries") {
		t.Fatalf("allowMultiQueries should not remain in DSN: %s", got)
	}
	if !strings.Contains(got, "multiStatements=true") {
		t.Fatalf("allowMultiQueries=true should map to multiStatements=true: %s", got)
	}
	if !strings.HasSuffix(got, "#debug") {
		t.Fatalf("fragment should be preserved: %s", got)
	}
}

func TestNormalizeMySQLRawDSNCompatibilityParamsPreservesExplicitMultiStatements(t *testing.T) {
	got := normalizeMySQLRawDSNCompatibilityParams(
		"root:pass@tcp(127.0.0.1:3306)/app?allowMultiQueries=true&multiStatements=false",
	)
	if strings.Contains(got, "allowMultiQueries") {
		t.Fatalf("allowMultiQueries should not remain in DSN: %s", got)
	}
	if !strings.Contains(got, "multiStatements=false") {
		t.Fatalf("explicit multiStatements should win: %s", got)
	}
}

func TestCustomDBOnlyNormalizesBuiltInMySQLDriverDSN(t *testing.T) {
	customMySQLDSNRecordingLastDSN = ""
	rawDSN := "root:pass@tcp(127.0.0.1:3306)/app?allowMultiQueries=true"

	db := &CustomDB{}
	err := db.Connect(connection.ConnectionConfig{
		Driver: customMySQLDSNRecordingDriverName,
		DSN:    rawDSN,
	})
	if err != nil {
		t.Fatalf("Connect failed: %v", err)
	}
	t.Cleanup(func() {
		_ = db.Close()
	})
	if customMySQLDSNRecordingLastDSN != rawDSN {
		t.Fatalf("non-mysql custom driver DSN should stay untouched, got %q", customMySQLDSNRecordingLastDSN)
	}
}