mirror of
https://github.com/flipped-aurora/gin-vue-admin.git
synced 2026-10-04 09:10:57 +00:00
15e6e7685c
* feat: 调整AI工作模式，更符合harness基准 * feat: 添加BusinessDB字段到PluginInitializeGorm结构体，并增加相关测试用例 * [middleware/jwt.go]: fix #2192 issues bug * docs: add auto plugin design spec * chore: ignore worktrees directory * feat: 调整代码辅助能力至插件 * feat: 添加警告条组件，提示授权用户访问限制 * fix: 修正商业用途版权声明链接 * feat: 调整agent.md 更加节省token * fix: 修改casbin版本为v3 * feat: 更新JSONMap和JSONSlice类型，优化GORM数据类型处理 * feat: 添加数据库就绪通知机制，优化插件注册流程 * feat: 重构API路径和描述，优化代码生成器和模板配置分组 * feat: 优化 person 页面的 css 作用域限制 * feat: 更新 vite 至 vite8 * 新增：MCP工具为指定URL的角色ID授权 * fix: 增加文件名合法性检查，拒绝包含非法字符的文件写入 * chore: 更新CI配置，升级Node.js和Go版本，调整checkout和setup动作版本 * feat: 为数据库连接增加最大复用时间配置 * feat: 为各数据库连接配置增加最大连接生命周期设置 * feat: 更新插件注册逻辑并优化API和菜单组件的标签显示 * feat: 更新版本号至v2.9.2并添加新插件路径信息 --------- Co-authored-by: taincheng <zhangtc@gmail.com> Co-authored-by: Azir-11 <2075125282@qq.com> Co-authored-by: lanxi <1220lanxi@gmail.com>
80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
package ast
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestPluginInitializeGormInjectionUsesBusinessDB(t *testing.T) {
|
|
const source = `package initialize
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/flipped-aurora/gin-vue-admin/server/global"
|
|
"github.com/pkg/errors"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func Gorm(ctx context.Context) {
|
|
err := global.GVA_DB.WithContext(ctx).AutoMigrate()
|
|
if err != nil {
|
|
err = errors.Wrap(err, "注册表失败!")
|
|
zap.L().Error(fmt.Sprintf("%+v", err))
|
|
}
|
|
}
|
|
`
|
|
|
|
dir := t.TempDir()
|
|
path := filepath.Join(dir, "gorm.go")
|
|
if err := os.WriteFile(path, []byte(source), 0o666); err != nil {
|
|
t.Fatalf("WriteFile() error = %v", err)
|
|
}
|
|
|
|
injection := &PluginInitializeGorm{
|
|
Type: TypePluginInitializeGorm,
|
|
Path: path,
|
|
ImportPath: `"github.com/flipped-aurora/gin-vue-admin/server/plugin/demo/model"`,
|
|
StructName: "Demo",
|
|
PackageName: "model",
|
|
IsNew: true,
|
|
}
|
|
|
|
businessField := reflect.ValueOf(injection).Elem().FieldByName("Business")
|
|
if !businessField.IsValid() {
|
|
t.Fatal("expected PluginInitializeGorm.Business field")
|
|
}
|
|
businessField.SetString("bizdb")
|
|
|
|
file, err := injection.Parse(path, nil)
|
|
if err != nil {
|
|
t.Fatalf("Parse() error = %v", err)
|
|
}
|
|
if err := injection.Injection(file); err != nil {
|
|
t.Fatalf("Injection() error = %v", err)
|
|
}
|
|
if err := injection.Format(path, nil, file); err != nil {
|
|
t.Fatalf("Format() error = %v", err)
|
|
}
|
|
|
|
content, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("ReadFile() error = %v", err)
|
|
}
|
|
|
|
got := string(content)
|
|
if !strings.Contains(got, "global.GVA_DB.WithContext(ctx).AutoMigrate()") {
|
|
t.Fatalf("expected default gorm block to remain, got:\n%s", got)
|
|
}
|
|
if !strings.Contains(got, `global.MustGetGlobalDBByDBName("bizdb").WithContext(ctx).AutoMigrate(`) {
|
|
t.Fatalf("expected gorm injection to use business db, got:\n%s", got)
|
|
}
|
|
if !strings.Contains(got, "model.Demo{}") {
|
|
t.Fatalf("expected model injection, got:\n%s", got)
|
|
}
|
|
}
|