golang/ent
1
0
Fork 0
mirror of https://github.com/ent/ent.git synced 2026-05-24 09:31:56 +03:00
Code Issues Packages Projects Releases Wiki Activity

entc/gen/features/multischema: pass schema to count query (#1390)

Browse Source 
Fixed #1385
This commit is contained in:
Ariel Mashraki
2021-03-25 13:48:32 +02:00
committed by GitHub
parent d62b810096
commit 2d09c8f846
6 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
entc/integration/multischema/ent/group_query.go
View File

@@ -455,6 +455,8 @@ func (gq *GroupQuery) sqlAll(ctx context.Context) ([]*Group, error) {
func (gq *GroupQuery) sqlCount(ctx context.Context) (int, error) {
_spec := gq.querySpec()
_spec.Node.Schema = gq.schemaConfig.Group
ctx = internal.NewSchemaConfigContext(ctx, gq.schemaConfig)
return sqlgraph.CountNodes(ctx, gq.driver, _spec)
}

2
entc/integration/multischema/ent/pet_query.go
View File

@@ -425,6 +425,8 @@ func (pq *PetQuery) sqlAll(ctx context.Context) ([]*Pet, error) {
func (pq *PetQuery) sqlCount(ctx context.Context) (int, error) {
_spec := pq.querySpec()
_spec.Node.Schema = pq.schemaConfig.Pet
ctx = internal.NewSchemaConfigContext(ctx, pq.schemaConfig)
return sqlgraph.CountNodes(ctx, pq.driver, _spec)
}

2
entc/integration/multischema/ent/user_query.go
View File

@@ -524,6 +524,8 @@ func (uq *UserQuery) sqlAll(ctx context.Context) ([]*User, error) {
func (uq *UserQuery) sqlCount(ctx context.Context) (int, error) {
_spec := uq.querySpec()
_spec.Node.Schema = uq.schemaConfig.User
ctx = internal.NewSchemaConfigContext(ctx, uq.schemaConfig)
return sqlgraph.CountNodes(ctx, uq.driver, _spec)
}

3
entc/integration/multischema/multischema_test.go
View File

@@ -80,6 +80,9 @@ func TestMySQL(t *testing.T) {
pedro = pedro.Update().ClearOwner().SaveX(ctx)
exist = pedro.QueryOwner().ExistX(ctx)
require.False(t, exist)
require.Equal(t, client.User.Query().CountX(ctx), len(client.User.Query().AllX(ctx)))
require.Equal(t, client.Pet.Query().CountX(ctx), len(client.Pet.Query().AllX(ctx)))
}
func setupSchema(t *testing.T, drv *sql.Driver) {