func schema(sn string) *pqt.Schema {
title := pqt.NewColumn("title", pqt.TypeText(), pqt.WithNotNull(), pqt.WithUnique())
lead := pqt.NewColumn("lead", pqt.TypeText())
news := pqt.NewTable("news", pqt.WithTableIfNotExists()).
AddColumn(pqt.NewColumn("id", pqt.TypeSerialBig(), pqt.WithPrimaryKey())).
AddColumn(title).
AddColumn(lead).
AddColumn(pqt.NewColumn("continue", pqt.TypeBool(), pqt.WithNotNull(), pqt.WithDefault("false"))).
AddColumn(pqt.NewColumn("content", pqt.TypeText(), pqt.WithNotNull())).
AddUnique(title, lead)
comment := pqt.NewTable("comment", pqt.WithTableIfNotExists()).
AddColumn(pqt.NewColumn("id", pqt.TypeSerialBig())).
AddColumn(pqt.NewColumn("content", pqt.TypeText(), pqt.WithNotNull())).
AddColumn(pqt.NewColumn(
"news_title",
pqt.TypeText(),
pqt.WithNotNull(),
pqt.WithReference(title, pqt.WithBidirectional(), pqt.WithOwnerName("comments_by_news_title"), pqt.WithInversedName("news_by_title")),
))
category := pqt.NewTable("category", pqt.WithTableIfNotExists()).
AddColumn(pqt.NewColumn("id", pqt.TypeSerialBig(), pqt.WithPrimaryKey())).
AddColumn(pqt.NewColumn("name", pqt.TypeText(), pqt.WithNotNull())).
AddColumn(pqt.NewColumn("content", pqt.TypeText(), pqt.WithNotNull())).
AddRelationship(
pqt.OneToMany(
pqt.SelfReference(),
pqt.WithBidirectional(),
pqt.WithInversedName("child_category"),
pqt.WithOwnerName("parent_category"),
pqt.WithColumnName("parent_id"),
),
)
pkg := pqt.NewTable("package", pqt.WithTableIfNotExists()).
AddColumn(pqt.NewColumn("id", pqt.TypeSerialBig(), pqt.WithPrimaryKey())).
AddColumn(pqt.NewColumn("break", pqt.TypeText())).
AddRelationship(pqt.ManyToOne(
category,
pqt.WithBidirectional(),
))
timestampable(news)
timestampable(comment)
timestampable(category)
timestampable(pkg)
comment.AddRelationship(pqt.ManyToOne(news, pqt.WithBidirectional(), pqt.WithInversedName("news_by_id")), pqt.WithNotNull())
pqt.ManyToMany(category, news, pqt.WithBidirectional())
return pqt.NewSchema(sn, pqt.WithSchemaIfNotExists()).
AddTable(category).
AddTable(pkg).
AddTable(news).
AddTable(comment)
}
func TestTable_AddRelationship_oneToOneBidirectional(t *testing.T) {
user := pqt.NewTable("user").AddColumn(pqt.NewColumn("id", pqt.TypeSerial(), pqt.WithPrimaryKey()))
userDetail := pqt.NewTable("user_detail").AddColumn(pqt.NewColumn("id", pqt.TypeSerial(), pqt.WithPrimaryKey()))
user.AddRelationship(pqt.OneToOne(
userDetail,
pqt.WithInversedName("details"),
pqt.WithOwnerName("user"),
pqt.WithBidirectional(),
))
if len(user.OwnedRelationships) != 1 {
t.Fatalf("user should have 1 relationship, but has %d", len(user.OwnedRelationships))
}
if user.OwnedRelationships[0].OwnerName != "user" {
t.Errorf("user relationship to user_detail should be mapped by user, but is %s", user.OwnedRelationships[0].OwnerName)
}
if user.OwnedRelationships[0].OwnerTable != user {
t.Errorf("user relationship to user_detail should be mapped by user table, but is %v", user.OwnedRelationships[0].OwnerTable)
}
if user.OwnedRelationships[0].Type != pqt.RelationshipTypeOneToOne {
t.Error("user relationship to user_detail should be one to one bidirectional")
}
if len(userDetail.InversedRelationships) != 1 {
t.Fatalf("user_detail should have 1 relationship, but has %d", len(userDetail.InversedRelationships))
}
if userDetail.InversedRelationships[0].InversedName != "details" {
t.Error("user_detail relationship to user should be mapped by user")
}
if userDetail.InversedRelationships[0].InversedTable != userDetail {
t.Error("user_detail relationship to user should be mapped by user_detail table")
}
if userDetail.InversedRelationships[0].Type != pqt.RelationshipTypeOneToOne {
t.Errorf("user_detail relationship to user should be %d, but is %d", pqt.RelationshipTypeOneToOne, userDetail.InversedRelationships[0].Type)
}
}
func TestTable_AddRelationship_oneToMany(t *testing.T) {
user := pqt.NewTable("user").AddColumn(pqt.NewColumn("id", pqt.TypeSerial(), pqt.WithPrimaryKey()))
comment := pqt.NewTable("comment").AddColumn(pqt.NewColumn("id", pqt.TypeSerial(), pqt.WithPrimaryKey()))
user.AddRelationship(pqt.OneToMany(
comment,
pqt.WithBidirectional(),
pqt.WithInversedName("author"),
pqt.WithOwnerName("comments"),
))
if len(user.InversedRelationships) != 1 {
t.Fatalf("user should have 1 inversed relationship, but has %d", len(user.InversedRelationships))
}
if user.InversedRelationships[0].OwnerName != "comments" {
t.Errorf("user inversed relationship to comment should be mapped by comments")
}
if user.InversedRelationships[0].OwnerTable != comment {
t.Errorf("user inversed relationship to comment should be mapped by comment table")
}
if user.InversedRelationships[0].Type != pqt.RelationshipTypeOneToMany {
t.Errorf("user inversed relationship to comment should be one to many")
}
if len(comment.OwnedRelationships) != 1 {
t.Fatalf("comment should have 1 owned relationship, but has %d", len(comment.OwnedRelationships))
}
if comment.OwnedRelationships[0].InversedName != "author" {
t.Errorf("comment relationship to user should be mapped by author")
}
if comment.OwnedRelationships[0].InversedTable != user {
t.Errorf("comment relationship to user should be mapped by user table")
}
if comment.OwnedRelationships[0].Type != pqt.RelationshipTypeOneToMany {
t.Errorf("comment relationship to user should be %d, but is %d", pqt.RelationshipTypeOneToMany, comment.OwnedRelationships[0].Type)
}
}