mirror of
https://github.com/Syngnat/GoNavi.git
synced 2026-05-11 18:19:51 +08:00
129 lines
3.3 KiB
Go
129 lines
3.3 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"GoNavi-Wails/internal/connection"
|
|
"GoNavi-Wails/internal/db"
|
|
"GoNavi-Wails/internal/secretstore"
|
|
)
|
|
|
|
type fakeBatchWriteDB struct {
|
|
batchCalls int
|
|
execCalls int
|
|
lastQuery string
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) Connect(config connection.ConnectionConfig) error {
|
|
return nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) Ping() error {
|
|
return nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) Query(query string) ([]map[string]interface{}, []string, error) {
|
|
return nil, nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) Exec(query string) (int64, error) {
|
|
f.execCalls++
|
|
return 1, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetDatabases() ([]string, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetTables(dbName string) ([]string, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetCreateStatement(dbName, tableName string) (string, error) {
|
|
return "", nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetColumns(dbName, tableName string) ([]connection.ColumnDefinition, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetAllColumns(dbName string) ([]connection.ColumnDefinitionWithTable, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetIndexes(dbName, tableName string) ([]connection.IndexDefinition, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetForeignKeys(dbName, tableName string) ([]connection.ForeignKeyDefinition, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) GetTriggers(dbName, tableName string) ([]connection.TriggerDefinition, error) {
|
|
return nil, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) ExecContext(ctx context.Context, query string) (int64, error) {
|
|
f.execCalls++
|
|
return 1, nil
|
|
}
|
|
|
|
func (f *fakeBatchWriteDB) ExecBatchContext(ctx context.Context, query string) (int64, error) {
|
|
f.batchCalls++
|
|
f.lastQuery = query
|
|
return 500, nil
|
|
}
|
|
|
|
var _ db.BatchWriteExecer = (*fakeBatchWriteDB)(nil)
|
|
|
|
func TestDBQueryMultiUsesBatchWriteExecerForAllWriteStatements(t *testing.T) {
|
|
originalNewDatabaseFunc := newDatabaseFunc
|
|
t.Cleanup(func() {
|
|
newDatabaseFunc = originalNewDatabaseFunc
|
|
})
|
|
|
|
fakeDB := &fakeBatchWriteDB{}
|
|
newDatabaseFunc = func(dbType string) (db.Database, error) {
|
|
return fakeDB, nil
|
|
}
|
|
|
|
app := NewAppWithSecretStore(secretstore.NewUnavailableStore("test"))
|
|
config := connection.ConnectionConfig{
|
|
Type: "mysql",
|
|
Host: "127.0.0.1",
|
|
Port: 1433,
|
|
User: "sa",
|
|
}
|
|
query := "INSERT INTO demo(id) VALUES (1);\nINSERT INTO demo(id) VALUES (2);"
|
|
|
|
result := app.DBQueryMulti(config, "testdb", query, "batch-write-test")
|
|
if !result.Success {
|
|
t.Fatalf("expected DBQueryMulti success, got failure: %s", result.Message)
|
|
}
|
|
if fakeDB.batchCalls != 1 {
|
|
t.Fatalf("expected batch path to run once, got %d", fakeDB.batchCalls)
|
|
}
|
|
if fakeDB.execCalls != 0 {
|
|
t.Fatalf("expected sequential exec path to be skipped, got execCalls=%d", fakeDB.execCalls)
|
|
}
|
|
if fakeDB.lastQuery != query {
|
|
t.Fatalf("expected batch query to stay intact, got %q", fakeDB.lastQuery)
|
|
}
|
|
|
|
resultSets, ok := result.Data.([]connection.ResultSetData)
|
|
if !ok {
|
|
t.Fatalf("expected []connection.ResultSetData, got %T", result.Data)
|
|
}
|
|
if len(resultSets) != 1 || len(resultSets[0].Rows) != 1 {
|
|
t.Fatalf("expected one affectedRows result set, got %#v", resultSets)
|
|
}
|
|
if got := resultSets[0].Rows[0]["affectedRows"]; got != int64(500) {
|
|
t.Fatalf("expected affectedRows=500, got %#v", got)
|
|
}
|
|
}
|