Skip to content

Commit

Permalink
feat: add prefix "_gent_test_" for test global var (#1060)
Browse files Browse the repository at this point in the history
* feat: add prefix "_gent_test_" for test global var

* tests: update test data
  • Loading branch information
tr1v3r authored Jan 16, 2024
1 parent 7a33439 commit 9c187af
Show file tree
Hide file tree
Showing 24 changed files with 138 additions and 138 deletions.
6 changes: 3 additions & 3 deletions internal/template/method.go
Original file line number Diff line number Diff line change
Expand Up @@ -265,14 +265,14 @@ func ({{.S}} *{{.QueryStructName}}Do) withDO(do gen.Dao) (*{{.QueryStructName}}D
const CRUDMethodTest = `
func init() {
InitializeDB()
err := db.AutoMigrate(&{{.StructInfo.Package}}.{{.ModelStructName}}{})
err := _gen_test_db.AutoMigrate(&{{.StructInfo.Package}}.{{.ModelStructName}}{})
if err != nil{
fmt.Printf("Error: AutoMigrate(&{{.StructInfo.Package}}.{{.ModelStructName}}{}) fail: %s", err)
}
}
func Test_{{.QueryStructName}}Query(t *testing.T) {
{{.QueryStructName}} := new{{.ModelStructName}}(db)
{{.QueryStructName}} := new{{.ModelStructName}}(_gen_test_db)
{{.QueryStructName}} = *{{.QueryStructName}}.As({{.QueryStructName}}.TableName())
_do := {{.QueryStructName}}.WithContext(context.Background()).Debug()
Expand Down Expand Up @@ -416,7 +416,7 @@ const DIYMethodTest = `
var {{.OriginStruct.Type}}{{.MethodName}}TestCase = []TestCase{}
func Test_{{.TargetStruct}}_{{.MethodName}}(t *testing.T) {
{{.TargetStruct}} := new{{.OriginStruct.Type}}(db)
{{.TargetStruct}} := new{{.OriginStruct.Type}}(_gen_test_db)
do := {{.TargetStruct}}.WithContext(context.Background()).Debug()
for i, tt := range {{.OriginStruct.Type}}{{.MethodName}}TestCase {
Expand Down
24 changes: 12 additions & 12 deletions internal/template/query.go
Original file line number Diff line number Diff line change
Expand Up @@ -114,22 +114,22 @@ func (q *QueryTx) RollbackTo(name string) error {
// QueryMethodTest query method test template
const QueryMethodTest = `
const dbName = "gen_test.db"
const _gen_test_db_name = "gen_test.db"
var db *gorm.DB
var once sync.Once
var _gen_test_db *gorm.DB
var _gen_test_once sync.Once
func init() {
InitializeDB()
db.AutoMigrate(&_another{})
_gen_test_db.AutoMigrate(&_another{})
}
func InitializeDB() {
once.Do(func() {
_gen_test_once.Do(func() {
var err error
db, err = gorm.Open(sqlite.Open(dbName), &gorm.Config{})
_gen_test_db, err = gorm.Open(sqlite.Open(_gen_test_db_name), &gorm.Config{})
if err != nil {
panic(fmt.Errorf("open sqlite %q fail: %w", dbName, err))
panic(fmt.Errorf("open sqlite %q fail: %w", _gen_test_db_name, err))
}
})
}
Expand All @@ -147,15 +147,15 @@ type _another struct {
func (*_another) TableName() string { return "another_for_unit_test" }
func Test_Available(t *testing.T) {
if !Use(db).Available() {
if !Use(_gen_test_db).Available() {
t.Errorf("query.Available() == false")
}
}
func Test_WithContext(t *testing.T) {
query := Use(db)
query := Use(_gen_test_db)
if !query.Available() {
t.Errorf("query Use(db) fail: query.Available() == false")
t.Errorf("query Use(_gen_test_db) fail: query.Available() == false")
}
type Content string
Expand All @@ -174,9 +174,9 @@ func Test_WithContext(t *testing.T) {
}
func Test_Transaction(t *testing.T) {
query := Use(db)
query := Use(_gen_test_db)
if !query.Available() {
t.Errorf("query Use(db) fail: query.Available() == false")
t.Errorf("query Use(_gen_test_db) fail: query.Available() == false")
}
err := query.Transaction(func(tx *Query) error { return nil })
Expand Down
4 changes: 2 additions & 2 deletions tests/.expect/dal_2/query/banks.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_2/query/credit_cards.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_2/query/customers.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

24 changes: 12 additions & 12 deletions tests/.expect/dal_2/query/gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_2/query/people.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_2/query/users.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_3/query/banks.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_3/query/credit_cards.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_3/query/customers.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

24 changes: 12 additions & 12 deletions tests/.expect/dal_3/query/gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_3/query/people.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_3/query/users.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions tests/.expect/dal_4/query/banks.gen_test.go

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading

0 comments on commit 9c187af

Please sign in to comment.