Bridge-and-Join-s/internal/lkgateway/fancyfb_test.go

package lkgateway

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

	"git.zetit.ru/zuevav/Bridge-and-Join-s/internal/lkgateway/fsql"
)

func TestFBSQLMechanical(t *testing.T) {
	cases := []struct{ in, want string }{
		{"SELECT TOP 1 ID FROM OD_VALUES WHERE ISIN=@i ORDER BY ID", "SELECT FIRST 1 ID FROM OD_VALUES WHERE ISIN=@i ORDER BY ID"},
		{"SELECT DISTINCT TOP (5000) PORT_NUM FROM OD_CLIENTS_PORTS", "SELECT FIRST (5000) DISTINCT PORT_NUM FROM OD_CLIENTS_PORTS"},
		{"SELECT ISNULL(STATE,'') FROM dbo.FTV_M2M_EVENTS WHERE GUID=@g", "SELECT COALESCE(STATE,'') FROM FTV_M2M_EVENTS WHERE GUID=@g"},
		{"INSERT INTO [dbo].FTV_M2M_EVENTS (EVENT_TYPE, EVENT_DATE) VALUES ('HANDBOOK', GETDATE())", "INSERT INTO FTV_M2M_EVENTS (EVENT_TYPE, EVENT_DATE) VALUES ('HANDBOOK', CURRENT_TIMESTAMP)"},
		// N'…' — без N; внутри строк и комментариев ничего не меняется
		{"SELECT 1 FROM OD_FACES WHERE MNEM = N'НДЦ' AND NAME <> ' N' /* ISNULL( dbo.x */", "SELECT 1 FROM OD_FACES WHERE MNEM = 'НДЦ' AND NAME <> ' N' /* ISNULL( dbo.x */"},
		{"SELECT 'TOP 1 ISNULL(' FROM RDB$DATABASE", "SELECT 'TOP 1 ISNULL(' FROM RDB$DATABASE"},
		{"exec dbo.FTV_M2M_JOB", "EXECUTE PROCEDURE FTV_M2M_JOB"},
		{"SELECT OBJECT_ID('dbo.FTV_M2M')", "SELECT (SELECT FIRST 1 1 FROM RDB$RELATIONS WHERE RDB$RELATION_NAME = 'FTV_M2M') FROM RDB$DATABASE"},
		{fsql.Hlth03, fsql.FB[fsql.Hlth03]},
	}
	for _, c := range cases {
		if got := fbSQL(c.in); got != c.want {
			t.Errorf("fbSQL(%q)\n got %q\nwant %q", c.in, got, c.want)
		}
	}
}

func TestFBParamsAndArgs(t *testing.T) {
	q, names := fbParams("SELECT @g, '@x', @r /* @y */, @g -- @z\nFROM T WHERE A=@p1")
	if q != "SELECT ?, '@x', ? /* @y */, ? -- @z\nFROM T WHERE A=?" {
		t.Fatalf("q = %q", q)
	}
	if !reflect.DeepEqual(names, []string{"g", "r", "g", "p1"}) {
		t.Fatalf("names = %v", names)
	}
	args := []driver.NamedValue{
		{Name: "r", Ordinal: 1, Value: "ref"},
		{Name: "g", Ordinal: 2, Value: "bae1fa60-eb0a-4fee-b12c-018d1eafcbd2"},
		{Ordinal: 3, Value: int64(7)},
		{Name: "unused", Ordinal: 4, Value: "x"},
	}
	// @p1 — первый параметр без имени (Ordinal 1 в порядке вызова): здесь его нет — ошибка
	if _, err := fbArgs(names, args); err == nil {
		t.Fatal("ожидалась ошибка: @p1 без позиционного параметра")
	}
	got, err := fbArgs([]string{"g", "r", "g"}, args)
	if err != nil {
		t.Fatal(err)
	}
	want := []any{"BAE1FA60-EB0A-4FEE-B12C-018D1EAFCBD2", "ref", "BAE1FA60-EB0A-4FEE-B12C-018D1EAFCBD2"}
	for i, a := range got {
		if a.Value != want[i] || a.Ordinal != i+1 {
			t.Fatalf("arg %d = %+v, want %v", i, a, want[i])
		}
	}
	pos, err := fbArgs([]string{"p1", "p2"}, []driver.NamedValue{{Ordinal: 1, Value: "dbo.FTV_M2M"}, {Ordinal: 2, Value: "2026-10-27T15:00:00.000"}})
	if err != nil || pos[0].Value != "dbo.FTV_M2M" || pos[1].Value != "2026-10-27 15:00:00.000" {
		t.Fatalf("позиционные: %+v %v", pos, err)
	}
}

func TestFBNativePassThrough(t *testing.T) {
	ddl := "CREATE OR ALTER PROCEDURE X AS BEGIN CMT = ' N' || 'ISNULL(dbo.' ; END"
	q, a, err := fbRewrite(ddl, nil)
	if err != nil || q != ddl || a != nil {
		t.Fatalf("DDL изменён: %q %v", q, err)
	}
	if !fbNative("/* шапка */\nEXECUTE BLOCK AS BEGIN END") || fbNative("EXECUTE PROCEDURE FTV_M2M_JOB") {
		t.Fatal("fbNative")
	}
}

func TestFBStatements(t *testing.T) {
	script := `/* шапка; с точкой с запятой */
SET TERM ^ ;
EXECUTE BLOCK AS
BEGIN
  IF (1 = 1) THEN EXECUTE STATEMENT 'CREATE TABLE T (A VARCHAR(10) DEFAULT ''x;y'')';
END^
COMMIT^
CREATE OR ALTER TRIGGER T_BI FOR T BEFORE INSERT AS BEGIN NEW.A = 'a^b'; END^
SET TERM ; ^
CREATE OR ALTER EXCEPTION E 'Ошибка; текст';
-- комментарий без команды
COMMIT;
`
	got := fbStatements(script)
	if len(got) != 3 {
		t.Fatalf("команд %d: %q", len(got), got)
	}
	if !strings.HasPrefix(strings.TrimSpace(fbStripComments(got[0])), "EXECUTE BLOCK") || !strings.Contains(got[0], "''x;y''") {
		t.Errorf("блок: %q", got[0])
	}
	if !strings.Contains(got[1], "'a^b'") || !strings.HasPrefix(got[2], "CREATE OR ALTER EXCEPTION") {
		t.Errorf("команды: %q", got)
	}
}

// Встроенные скрипты Firebird — те же, что в deploy/depo/fancy-m2m-firebird (их ставят вручную по README).
func TestFBScriptsMatchDeploy(t *testing.T) {
	for _, n := range depoFBScriptFiles {
		b, err := readRepoFile("deploy/depo/fancy-m2m-firebird/" + n)
		if err != nil {
			t.Fatal(err)
		}
		if string(b) != depoFBScript(n) {
			t.Errorf("%s: встроенный текст отличается от deploy/depo/fancy-m2m-firebird — скопируйте изменения", n)
		}
		if len(fbStatements(depoFBScript(n))) == 0 {
			t.Errorf("%s: нет команд", n)
		}
	}
}

func TestFBSplitConn(t *testing.T) {
	cases := []struct {
		in, host string
		port     int
		path     string
		bad      bool
	}{
		{`D:\Fancy\Bases\DEPO.GDB`, "", 0, `D:\Fancy\Bases\DEPO.GDB`, false},
		{`/var/db/depo.fdb`, "", 0, `/var/db/depo.fdb`, false},
		{`DEPO`, "", 0, `DEPO`, false},
		{`10.0.0.5:D:\Fancy\Bases\DEPO.GDB`, "10.0.0.5", 0, `D:\Fancy\Bases\DEPO.GDB`, false},
		{`srv-depo/3051:D:\Show Depo\Bases\SPI_DEPO.GDB`, "srv-depo", 3051, `D:\Show Depo\Bases\SPI_DEPO.GDB`, false},
		{`fbsrv:/var/db/depo.fdb`, "fbsrv", 0, `/var/db/depo.fdb`, false},
		{`fbsrv:DEPO`, "fbsrv", 0, `DEPO`, false},
		{`\\SRV-DEPO\D:\Fancy\DEPO.GDB`, "SRV-DEPO", 0, `D:\Fancy\DEPO.GDB`, false},
		{`\\SRV-DEPO\d$\Fancy\DEPO.GDB`, "SRV-DEPO", 0, `D:\Fancy\DEPO.GDB`, false},
		{`inet://10.0.0.5:3050/D:\Fancy\DEPO.GDB`, "10.0.0.5", 3050, `D:\Fancy\DEPO.GDB`, false},
		{`inet://10.0.0.5/depo`, "10.0.0.5", 0, `depo`, false},
		{`\\SRV-DEPO\Bases\DEPO.GDB`, "", 0, "", true},
	}
	for _, c := range cases {
		h, p, path, err := fbSplitConn(c.in)
		if (err != nil) != c.bad || h != c.host || p != c.port || path != c.path {
			t.Errorf("fbSplitConn(%q) = %q, %d, %q, %v", c.in, h, p, path, err)
		}
	}
}