fix

latolukasz · Sep 1, 2021 · 797353c · 797353c
1 parent 5ba8eff
commit 797353c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/table_schema.go b/table_schema.go
@@ -90,6 +90,7 @@ type TableSchema interface {
 	GetReferences() []string
 	GetColumns() []string
 	GetSchemaChanges(engine *Engine) (has bool, alters []Alter)
+	GetUsage(registry ValidatedRegistry) map[reflect.Type][]string
 }
 
 type tableSchema struct {

diff --git a/validated_registry_test.go b/validated_registry_test.go
@@ -9,7 +9,8 @@ import (
 
 type validatedRegistryEntity struct {
 	ORM
-	ID uint
+	ID  uint
+	Ref *validatedRegistryEntity
 }
 
 type validatedRegistryNotRegisteredEntity struct {
@@ -71,4 +72,8 @@ func TestValidatedRegistry(t *testing.T) {
 	assert.PanicsWithError(t, "entity 'beeorm.validatedRegistryNotRegisteredEntity' is not registered", func() {
 		validated.GetTableSchemaForEntity(&validatedRegistryNotRegisteredEntity{})
 	})
+
+	usage := validated.GetTableSchemaForEntity(entity).GetUsage(validated)
+	assert.NotNil(t, usage)
+	assert.Len(t, usage, 1)
 }