style: adjustment code

This commit is contained in:
zhuyasen
2024-11-25 14:41:53 +08:00
parent e1ee704fb7
commit 091ad8b529
17 changed files with 44 additions and 54 deletions

View File

@@ -85,24 +85,6 @@ func TestGenerate(t *testing.T) {
}},
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{
@@ -112,16 +94,34 @@ func TestGenerate(t *testing.T) {
}},
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,
},
//{
// 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 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) {