sqlboiler/text_helpers_test.go

208 lines
6 KiB
Go
Raw Normal View History

2016-07-13 05:05:33 +02:00
package main
import (
"reflect"
"testing"
"github.com/davecgh/go-spew/spew"
2016-08-09 09:59:30 +02:00
"github.com/vattle/sqlboiler/bdb"
"github.com/vattle/sqlboiler/bdb/drivers"
2016-07-13 05:05:33 +02:00
)
2016-07-15 06:18:49 +02:00
func TestTextsFromForeignKey(t *testing.T) {
t.Parallel()
tables, err := bdb.Tables(&drivers.MockDriver{})
2016-07-15 06:18:49 +02:00
if err != nil {
t.Fatal(err)
}
jets := bdb.GetTable(tables, "jets")
texts := textsFromForeignKey("models", tables, jets, jets.FKeys[0])
2016-07-15 06:18:49 +02:00
expect := RelationshipToOneTexts{}
expect.ForeignKey = jets.FKeys[0]
expect.LocalTable.NameGo = "Jet"
expect.LocalTable.ColumnNameGo = "PilotID"
2016-07-15 06:18:49 +02:00
expect.ForeignTable.Name = "pilots"
expect.ForeignTable.NameGo = "Pilot"
expect.ForeignTable.NamePluralGo = "Pilots"
expect.ForeignTable.ColumnName = "id"
2016-07-15 06:18:49 +02:00
expect.ForeignTable.ColumnNameGo = "ID"
expect.Function.PackageName = "models"
expect.Function.Name = "Pilot"
2016-08-28 08:01:26 +02:00
expect.Function.ForeignName = "Jet"
expect.Function.Varname = "pilot"
expect.Function.Receiver = "j"
expect.Function.OneToOne = false
2016-07-15 06:18:49 +02:00
expect.Function.LocalAssignment = "PilotID.Int"
2016-07-15 06:18:49 +02:00
expect.Function.ForeignAssignment = "ID"
if !reflect.DeepEqual(expect, texts) {
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
}
2016-08-28 08:01:26 +02:00
texts = textsFromForeignKey("models", tables, jets, jets.FKeys[1])
expect = RelationshipToOneTexts{}
expect.ForeignKey = jets.FKeys[1]
expect.LocalTable.NameGo = "Jet"
expect.LocalTable.ColumnNameGo = "AirportID"
expect.ForeignTable.Name = "airports"
expect.ForeignTable.NameGo = "Airport"
expect.ForeignTable.NamePluralGo = "Airports"
expect.ForeignTable.ColumnName = "id"
expect.ForeignTable.ColumnNameGo = "ID"
expect.Function.PackageName = "models"
expect.Function.Name = "Airport"
expect.Function.ForeignName = "Jets"
expect.Function.Varname = "airport"
expect.Function.Receiver = "j"
expect.Function.OneToOne = false
2016-08-28 08:01:26 +02:00
expect.Function.LocalAssignment = "AirportID"
expect.Function.ForeignAssignment = "ID"
if !reflect.DeepEqual(expect, texts) {
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
}
if !reflect.DeepEqual(expect, texts) {
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
}
}
func TestTextsFromOneToOneRelationship(t *testing.T) {
t.Parallel()
tables, err := bdb.Tables(&drivers.MockDriver{})
if err != nil {
t.Fatal(err)
}
pilots := bdb.GetTable(tables, "pilots")
texts := textsFromOneToOneRelationship("models", tables, pilots, pilots.ToManyRelationships[0])
expect := RelationshipToOneTexts{}
expect.ForeignKey = bdb.ForeignKey{
Table: "pilots",
Name: "none",
Column: "id",
Nullable: false,
Unique: false,
ForeignTable: "jets",
ForeignColumn: "pilot_id",
ForeignColumnNullable: true,
ForeignColumnUnique: true,
}
expect.LocalTable.NameGo = "Pilot"
expect.LocalTable.ColumnNameGo = "ID"
expect.ForeignTable.Name = "jets"
expect.ForeignTable.NameGo = "Jet"
expect.ForeignTable.NamePluralGo = "Jets"
expect.ForeignTable.ColumnName = "pilot_id"
expect.ForeignTable.ColumnNameGo = "PilotID"
expect.Function.PackageName = "models"
expect.Function.Name = "Jet"
expect.Function.ForeignName = "Pilot"
expect.Function.Varname = "jet"
expect.Function.Receiver = "p"
expect.Function.OneToOne = true
expect.Function.LocalAssignment = "ID"
expect.Function.ForeignAssignment = "PilotID.Int"
2016-07-15 06:18:49 +02:00
if !reflect.DeepEqual(expect, texts) {
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
}
}
func TestTextsFromRelationship(t *testing.T) {
2016-07-13 05:05:33 +02:00
t.Parallel()
tables, err := bdb.Tables(&drivers.MockDriver{})
2016-07-13 05:05:33 +02:00
if err != nil {
t.Fatal(err)
}
pilots := bdb.GetTable(tables, "pilots")
texts := textsFromRelationship(tables, pilots, pilots.ToManyRelationships[0])
2016-07-13 05:05:33 +02:00
expect := RelationshipToManyTexts{}
expect.LocalTable.NameGo = "Pilot"
expect.LocalTable.NameSingular = "pilot"
2016-08-27 21:00:42 +02:00
expect.LocalTable.ColumnNameGo = "ID"
2016-07-13 05:05:33 +02:00
expect.ForeignTable.NameGo = "Jet"
expect.ForeignTable.NameSingular = "jet"
expect.ForeignTable.NamePluralGo = "Jets"
expect.ForeignTable.NameHumanReadable = "jets"
2016-08-27 21:00:42 +02:00
expect.ForeignTable.ColumnNameGo = "PilotID"
expect.ForeignTable.Slice = "JetSlice"
2016-07-13 05:05:33 +02:00
expect.Function.Name = "Jets"
expect.Function.ForeignName = "Pilot"
expect.Function.Receiver = "p"
2016-07-13 05:05:33 +02:00
expect.Function.LocalAssignment = "ID"
expect.Function.ForeignAssignment = "PilotID.Int"
2016-07-13 05:05:33 +02:00
if !reflect.DeepEqual(expect, texts) {
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
}
2016-07-14 05:29:29 +02:00
texts = textsFromRelationship(tables, pilots, pilots.ToManyRelationships[1])
2016-07-14 05:29:29 +02:00
expect = RelationshipToManyTexts{}
expect.LocalTable.NameGo = "Pilot"
expect.LocalTable.NameSingular = "pilot"
2016-08-27 21:00:42 +02:00
expect.LocalTable.ColumnNameGo = "ID"
2016-07-14 05:29:29 +02:00
expect.ForeignTable.NameGo = "License"
expect.ForeignTable.NameSingular = "license"
expect.ForeignTable.NamePluralGo = "Licenses"
expect.ForeignTable.NameHumanReadable = "licenses"
2016-08-27 21:00:42 +02:00
expect.ForeignTable.ColumnNameGo = "PilotID"
expect.ForeignTable.Slice = "LicenseSlice"
2016-07-14 05:29:29 +02:00
expect.Function.Name = "Licenses"
expect.Function.ForeignName = "Pilot"
expect.Function.Receiver = "p"
2016-07-14 05:29:29 +02:00
expect.Function.LocalAssignment = "ID"
expect.Function.ForeignAssignment = "PilotID"
2016-07-14 05:29:29 +02:00
if !reflect.DeepEqual(expect, texts) {
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
}
texts = textsFromRelationship(tables, pilots, pilots.ToManyRelationships[2])
2016-07-14 05:29:29 +02:00
expect = RelationshipToManyTexts{}
expect.LocalTable.NameGo = "Pilot"
expect.LocalTable.NameSingular = "pilot"
2016-08-27 21:00:42 +02:00
expect.LocalTable.ColumnNameGo = "ID"
2016-07-14 05:29:29 +02:00
expect.ForeignTable.NameGo = "Language"
expect.ForeignTable.NameSingular = "language"
expect.ForeignTable.NamePluralGo = "Languages"
expect.ForeignTable.NameHumanReadable = "languages"
2016-08-27 21:00:42 +02:00
expect.ForeignTable.ColumnNameGo = "ID"
expect.ForeignTable.Slice = "LanguageSlice"
2016-07-14 05:29:29 +02:00
expect.Function.Name = "Languages"
expect.Function.ForeignName = "Pilots"
expect.Function.Receiver = "p"
2016-07-14 05:29:29 +02:00
expect.Function.LocalAssignment = "ID"
expect.Function.ForeignAssignment = "ID"
if !reflect.DeepEqual(expect, texts) {
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
}
2016-07-13 05:05:33 +02:00
}