feat: get table info

This commit is contained in:
zhuyasen
2024-11-24 15:05:26 +08:00
parent 21272aee58
commit e3b6e11e6b
7 changed files with 287 additions and 56 deletions

View File

@@ -85,14 +85,43 @@ func TestGenerate(t *testing.T) {
}},
wantErr: false,
},
//{
// name: "sql from db",
// args: args{args: &Args{
// DBDsn: "root:123456@(127.0.0.1:3306)/test",
// DBTable: "user",
// }},
// wantErr: false,
//},
{
name: "sql from mysql",
args: args{args: &Args{
DBDsn: "root:123456@(192.168.3.37:3306)/account",
DBTable: "user",
DBDriver: "mysql",
}},
wantErr: false,
},
{
name: "sql from postgresql",
args: args{args: &Args{
DBDsn: "root:123456@(192.168.3.37:5432)/account",
DBTable: "user",
DBDriver: "postgresql",
}},
wantErr: false,
},
{
name: "sql from sqlite",
args: args{args: &Args{
DBDsn: "C:\\Users\\zhuyasen\\Desktop\\genTest\\sql\\sqlite\\sponge.db",
DBTable: "user",
DBDriver: "sqlite",
}},
wantErr: false,
},
{
name: "sql from mongodb",
args: args{args: &Args{
DBDsn: "root:123456@(192.168.3.37:27017)/account",
DBTable: "people",
DBDriver: "mongodb",
IsCustomTemplate: true,
}},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {