Files
ent/entc/load/load_test.go
Ariel Mashraki 74ce5b36f2 ent/load: safely marshaling schema
Summary: more descriptive output on marshaling failures.

Reviewed By: alexsn

Differential Revision: D16782294

fbshipit-source-id: 4ac3fbb3ef152d59bada257a9f0d8ccde5e95ae9
2019-08-13 03:06:08 -07:00

46 lines
1.1 KiB
Go

package load
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestLoad(t *testing.T) {
cfg := &Config{Path: "./testdata/valid"}
spec, err := cfg.Load()
require.NoError(t, err)
require.Len(t, spec.Schemas, 3)
require.Equal(t, "fbc/ent/entc/load/testdata/valid", spec.PkgPath)
}
func TestLoadWrongPath(t *testing.T) {
cfg := &Config{Path: "./boring"}
plg, err := cfg.Load()
require.Error(t, err)
require.Nil(t, plg)
}
func TestLoadSpecific(t *testing.T) {
cfg := &Config{Path: "./testdata/valid", Names: []string{"User"}}
spec, err := cfg.Load()
require.NoError(t, err)
require.Len(t, spec.Schemas, 1)
require.Equal(t, "User", spec.Schemas[0].Name)
require.Equal(t, "fbc/ent/entc/load/testdata/valid", spec.PkgPath)
}
func TestLoadNoSchema(t *testing.T) {
cfg := &Config{Path: "./testdata/invalid"}
schemas, err := cfg.Load()
require.Error(t, err)
require.Empty(t, schemas)
}
func TestLoadSchemaFailure(t *testing.T) {
cfg := &Config{Path: "./testdata/failure"}
schemas, err := cfg.Load()
require.Error(t, err)
require.Empty(t, schemas)
}