mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
d3b89fb2be
* entc/gen: add enttest package for generated code. It's still WIP and another 2 things need to be added: - Add an option to skip schema migration and pass client options - Update docs and replace existing testing code to use enttest * entc/gen: finish enttest package tempalte Signed-off-by: Alex Snast <alexsn@fb.com> Co-authored-by: Alex Snast <alexsn@fb.com>
101 lines
2.2 KiB
Cheetah
101 lines
2.2 KiB
Cheetah
{{/*
|
|
Copyright 2019-present Facebook Inc. All rights reserved.
|
|
This source code is licensed under the Apache 2.0 license found
|
|
in the LICENSE file in the root directory of this source tree.
|
|
*/}}
|
|
|
|
{{ define "enttest" }}
|
|
|
|
{{ $pkg := base $.Config.Package }}
|
|
|
|
{{ with extend $ "Package" "enttest" -}}
|
|
{{ template "header" . }}
|
|
{{ end }}
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"{{ $.Config.Package }}"
|
|
{{- if $.SupportMigrate }}
|
|
"github.com/facebookincubator/ent/dialect/sql/schema"
|
|
{{- end }}
|
|
|
|
// required by schema hooks.
|
|
_ "{{ $.Config.Package }}/runtime"
|
|
)
|
|
|
|
type (
|
|
// TestingT is the interface that is shared between
|
|
// testing.T and testing.B and used by enttest.
|
|
TestingT interface {
|
|
FailNow()
|
|
Error(...interface{})
|
|
}
|
|
|
|
// Option configures client creation.
|
|
Option func(*options)
|
|
|
|
options struct {
|
|
opts []{{ $pkg }}.Option
|
|
{{- if $.SupportMigrate }}
|
|
migrateOpts []schema.MigrateOption
|
|
{{- end }}
|
|
}
|
|
)
|
|
|
|
// WithOptions forwards options to client creation.
|
|
func WithOptions(opts ...{{ $pkg }}.Option) Option {
|
|
return func(o *options) {
|
|
o.opts = append(o.opts, opts...)
|
|
}
|
|
}
|
|
|
|
{{- if $.SupportMigrate }}
|
|
// WithMigrateOptions forwards options to auto migration.
|
|
func WithMigrateOptions(opts ...schema.MigrateOption) Option {
|
|
return func(o *options) {
|
|
o.migrateOpts = append(o.migrateOpts, opts...)
|
|
}
|
|
}
|
|
{{- end }}
|
|
|
|
func newOptions(opts []Option) *options {
|
|
o := &options{}
|
|
for _, opt := range opts {
|
|
opt(o)
|
|
}
|
|
return o
|
|
}
|
|
|
|
// Open calls {{ $pkg }}.Open and auto-run migration.
|
|
func Open(t TestingT, driverName, dataSourceName string, opts ...Option) *{{ $pkg }}.Client {
|
|
o := newOptions(opts)
|
|
c, err := {{ $pkg }}.Open(driverName, dataSourceName, o.opts...)
|
|
if err != nil {
|
|
t.Error(err)
|
|
t.FailNow()
|
|
}
|
|
{{- if $.SupportMigrate }}
|
|
if err := c.Schema.Create(context.Background(), o.migrateOpts...); err != nil {
|
|
t.Error(err)
|
|
t.FailNow()
|
|
}
|
|
{{- end }}
|
|
return c
|
|
}
|
|
|
|
// NewClient calls {{ $pkg }}.NewClient and auto-run migration.
|
|
func NewClient(t TestingT, opts ...Option) *{{ $pkg }}.Client {
|
|
o := newOptions(opts)
|
|
c := {{ $pkg }}.NewClient(o.opts...)
|
|
{{- if $.SupportMigrate }}
|
|
if err := c.Schema.Create(context.Background(), o.migrateOpts...); err != nil {
|
|
t.Error(err)
|
|
t.FailNow()
|
|
}
|
|
{{- end }}
|
|
return c
|
|
}
|
|
|
|
{{ end }}
|