mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
76f5b6ec5f
Summary: Pull Request resolved: https://github.com/facebookincubator/ent/pull/40 Reviewed By: alexsn Differential Revision: D17480104 fbshipit-source-id: 5223430e3b2223b8471a85bd1d85b445f23acfce
43 lines
1.4 KiB
Go
43 lines
1.4 KiB
Go
// 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.
|
|
|
|
package template
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/facebookincubator/ent/dialect/sql"
|
|
"github.com/facebookincubator/ent/entc/integration/config/ent"
|
|
"github.com/facebookincubator/ent/entc/integration/config/ent/migrate"
|
|
"github.com/facebookincubator/ent/entc/integration/config/ent/schema"
|
|
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCustomTemplate(t *testing.T) {
|
|
drv, err := sql.Open("sqlite3", "file:ent?mode=memory&cache=shared&_fk=1")
|
|
require.NoError(t, err)
|
|
defer drv.Close()
|
|
ctx := context.Background()
|
|
client := ent.NewClient(ent.Driver(drv))
|
|
require.NoError(t, client.Schema.Create(ctx, migrate.WithGlobalUniqueID(true)))
|
|
client.User.Create().SaveX(ctx)
|
|
|
|
// check that the table was created with the given custom name.
|
|
table := schema.User{}.Config().Table
|
|
query, args := sql.Select().Count().
|
|
From(sql.Table("sqlite_master")).
|
|
Where(sql.EQ("type", "table").And().EQ("name", table)).
|
|
Query()
|
|
rows := &sql.Rows{}
|
|
require.NoError(t, drv.Query(ctx, query, args, rows))
|
|
defer rows.Close()
|
|
require.True(t, rows.Next(), "no rows returned")
|
|
var n int
|
|
require.NoError(t, rows.Scan(&n), "scanning count")
|
|
require.Equalf(t, 1, n, "expecting table %q to be exist", table)
|
|
}
|