dps-sdk/test/v1/v1_test.go

package v1

import (
	"git.sxidc.com/service-supports/dps-sdk/ports"
	"math/rand"
	"testing"
	"time"
)

var tableModelDescribe = ports.TableModelDescribe{
	Fields: []ports.TableModelField{
		{"ID", "gorm:\"primary_key;type:varchar(32);comment:id;\""},
		{"Name", "gorm:\"not null;type:varchar(128);comment:数据库名称;\""},
		{"Time", "gorm:\"not null;type:timestamp with time zone;comment:数据库时间;\""},
		{"TableNum", "gorm:\"not null;type:integer;comment:数据库表数量;\""},
	},
}

func TestAutoMigrate(t *testing.T) {
	initClient(t, "localhost:30170")
	defer destroyClient(t)

	newToolKit(t).autoMigrate(&ports.AutoMigrateRequest{
		DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
		TablePrefixWithSchema: "test." + simpleUUID()[0:8],
		Version:               "v1",
		TableModelDescribe:    tableModelDescribe,
	})
}

func TestInsert(t *testing.T) {
	initClient(t, "localhost:30170")
	defer destroyClient(t)

	tablePrefix := "test." + simpleUUID()[0:8]

	id := simpleUUID()
	name := simpleUUID()
	now := time.Now().Local()
	tableNum := rand.New(rand.NewSource(now.Unix())).Intn(10)

	newToolKit(t).autoMigrate(&ports.AutoMigrateRequest{
		DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
		TablePrefixWithSchema: tablePrefix,
		Version:               "v1",
		TableModelDescribe:    tableModelDescribe,
	}).insert(&ports.InsertRequest{
		DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
		TablePrefixWithSchema: tablePrefix,
		Version:               "v1",
		KeyColumns:            []string{"id"},
		TableRow: map[string]any{
			"id":        id,
			"name":      name,
			"time":      now,
			"table_num": tableNum,
		},
		UserID: "test",
	})
}