Move fakeDB to MockDriver in drivers package
This commit is contained in:
Patrick O'brien
parent
94d36d7bf7
commit
8d0cd2a53d
3 changed files with 154 additions and 137 deletions
|
|
@ -6,38 +6,39 @@ import (
|
|||
|
||||
"github.com/davecgh/go-spew/spew"
|
||||
"github.com/vattle/sqlboiler/bdb"
|
||||
"github.com/vattle/sqlboiler/bdb/drivers"
|
||||
)
|
||||
|
||||
func TestTextsFromForeignKey(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
tables, err := bdb.Tables(fakeDB(0))
|
||||
tables, err := bdb.Tables(drivers.MockDriver(0))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
videos := bdb.GetTable(tables, "videos")
|
||||
texts := textsFromForeignKey("models", tables, videos, videos.FKeys[0])
|
||||
jets := bdb.GetTable(tables, "jets")
|
||||
texts := textsFromForeignKey("models", tables, jets, jets.FKeys[0])
|
||||
expect := RelationshipToOneTexts{}
|
||||
|
||||
expect.ForeignKey = videos.FKeys[0]
|
||||
expect.ForeignKey = jets.FKeys[0]
|
||||
|
||||
expect.LocalTable.NameGo = "Video"
|
||||
expect.LocalTable.ColumnNameGo = "UserID"
|
||||
expect.LocalTable.NameGo = "Jet"
|
||||
expect.LocalTable.ColumnNameGo = "PilotID"
|
||||
|
||||
expect.ForeignTable.Name = "users"
|
||||
expect.ForeignTable.NameGo = "User"
|
||||
expect.ForeignTable.NamePluralGo = "Users"
|
||||
expect.ForeignTable.Name = "pilots"
|
||||
expect.ForeignTable.NameGo = "Pilot"
|
||||
expect.ForeignTable.NamePluralGo = "Pilots"
|
||||
expect.ForeignTable.ColumnName = "id"
|
||||
expect.ForeignTable.ColumnNameGo = "ID"
|
||||
|
||||
expect.Function.PackageName = "models"
|
||||
expect.Function.Name = "User"
|
||||
expect.Function.Varname = "user"
|
||||
expect.Function.Receiver = "v"
|
||||
expect.Function.Name = "Pilot"
|
||||
expect.Function.Varname = "pilot"
|
||||
expect.Function.Receiver = "j"
|
||||
expect.Function.ReverseInserts = false
|
||||
|
||||
expect.Function.LocalAssignment = "UserID.Int32"
|
||||
expect.Function.LocalAssignment = "PilotID.Int32"
|
||||
expect.Function.ForeignAssignment = "ID"
|
||||
|
||||
if !reflect.DeepEqual(expect, texts) {
|
||||
|
|
@ -48,13 +49,13 @@ func TestTextsFromForeignKey(t *testing.T) {
|
|||
func TestTextsFromOneToOneRelationship(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
tables, err := bdb.Tables(fakeDB(0))
|
||||
tables, err := bdb.Tables(drivers.MockDriver(0))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
users := bdb.GetTable(tables, "users")
|
||||
texts := textsFromOneToOneRelationship("models", tables, users, users.ToManyRelationships[0])
|
||||
pilots := bdb.GetTable(tables, "pilots")
|
||||
texts := textsFromOneToOneRelationship("models", tables, pilots, pilots.ToManyRelationships[0])
|
||||
expect := RelationshipToOneTexts{}
|
||||
|
||||
expect.ForeignKey = bdb.ForeignKey{
|
||||
|
|
@ -63,29 +64,29 @@ func TestTextsFromOneToOneRelationship(t *testing.T) {
|
|||
Nullable: false,
|
||||
Unique: false,
|
||||
|
||||
ForeignTable: "videos",
|
||||
ForeignColumn: "user_id",
|
||||
ForeignTable: "jets",
|
||||
ForeignColumn: "pilot_id",
|
||||
ForeignColumnNullable: true,
|
||||
ForeignColumnUnique: true,
|
||||
}
|
||||
|
||||
expect.LocalTable.NameGo = "User"
|
||||
expect.LocalTable.NameGo = "Pilot"
|
||||
expect.LocalTable.ColumnNameGo = "ID"
|
||||
|
||||
expect.ForeignTable.Name = "videos"
|
||||
expect.ForeignTable.NameGo = "Video"
|
||||
expect.ForeignTable.NamePluralGo = "Videos"
|
||||
expect.ForeignTable.ColumnName = "user_id"
|
||||
expect.ForeignTable.ColumnNameGo = "UserID"
|
||||
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 = "Video"
|
||||
expect.Function.Varname = "video"
|
||||
expect.Function.Receiver = "u"
|
||||
expect.Function.Name = "Jet"
|
||||
expect.Function.Varname = "jet"
|
||||
expect.Function.Receiver = "p"
|
||||
expect.Function.ReverseInserts = true
|
||||
|
||||
expect.Function.LocalAssignment = "ID"
|
||||
expect.Function.ForeignAssignment = "UserID.Int32"
|
||||
expect.Function.ForeignAssignment = "PilotID.Int32"
|
||||
|
||||
if !reflect.DeepEqual(expect, texts) {
|
||||
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
|
||||
|
|
@ -95,65 +96,65 @@ func TestTextsFromOneToOneRelationship(t *testing.T) {
|
|||
func TestTextsFromRelationship(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
tables, err := bdb.Tables(fakeDB(0))
|
||||
tables, err := bdb.Tables(drivers.MockDriver(0))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
users := bdb.GetTable(tables, "users")
|
||||
texts := textsFromRelationship(tables, users, users.ToManyRelationships[0])
|
||||
pilots := bdb.GetTable(tables, "pilots")
|
||||
texts := textsFromRelationship(tables, pilots, pilots.ToManyRelationships[0])
|
||||
expect := RelationshipToManyTexts{}
|
||||
expect.LocalTable.NameGo = "User"
|
||||
expect.LocalTable.NameSingular = "user"
|
||||
expect.LocalTable.NameGo = "Pilot"
|
||||
expect.LocalTable.NameSingular = "pilot"
|
||||
|
||||
expect.ForeignTable.NameGo = "Video"
|
||||
expect.ForeignTable.NameSingular = "video"
|
||||
expect.ForeignTable.NamePluralGo = "Videos"
|
||||
expect.ForeignTable.NameHumanReadable = "videos"
|
||||
expect.ForeignTable.Slice = "VideoSlice"
|
||||
expect.ForeignTable.NameGo = "Jet"
|
||||
expect.ForeignTable.NameSingular = "jet"
|
||||
expect.ForeignTable.NamePluralGo = "Jets"
|
||||
expect.ForeignTable.NameHumanReadable = "jets"
|
||||
expect.ForeignTable.Slice = "JetSlice"
|
||||
|
||||
expect.Function.Name = "Videos"
|
||||
expect.Function.Receiver = "u"
|
||||
expect.Function.Name = "Jets"
|
||||
expect.Function.Receiver = "p"
|
||||
expect.Function.LocalAssignment = "ID"
|
||||
expect.Function.ForeignAssignment = "UserID.Int32"
|
||||
expect.Function.ForeignAssignment = "PilotID.Int32"
|
||||
|
||||
if !reflect.DeepEqual(expect, texts) {
|
||||
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
|
||||
}
|
||||
|
||||
texts = textsFromRelationship(tables, users, users.ToManyRelationships[1])
|
||||
texts = textsFromRelationship(tables, pilots, pilots.ToManyRelationships[1])
|
||||
expect = RelationshipToManyTexts{}
|
||||
expect.LocalTable.NameGo = "User"
|
||||
expect.LocalTable.NameSingular = "user"
|
||||
expect.LocalTable.NameGo = "Pilot"
|
||||
expect.LocalTable.NameSingular = "pilot"
|
||||
|
||||
expect.ForeignTable.NameGo = "Notification"
|
||||
expect.ForeignTable.NameSingular = "notification"
|
||||
expect.ForeignTable.NamePluralGo = "Notifications"
|
||||
expect.ForeignTable.NameHumanReadable = "notifications"
|
||||
expect.ForeignTable.Slice = "NotificationSlice"
|
||||
expect.ForeignTable.NameGo = "License"
|
||||
expect.ForeignTable.NameSingular = "license"
|
||||
expect.ForeignTable.NamePluralGo = "Licenses"
|
||||
expect.ForeignTable.NameHumanReadable = "licenses"
|
||||
expect.ForeignTable.Slice = "LicenseSlice"
|
||||
|
||||
expect.Function.Name = "Notifications"
|
||||
expect.Function.Receiver = "u"
|
||||
expect.Function.Name = "Licenses"
|
||||
expect.Function.Receiver = "p"
|
||||
expect.Function.LocalAssignment = "ID"
|
||||
expect.Function.ForeignAssignment = "UserID"
|
||||
expect.Function.ForeignAssignment = "PilotID"
|
||||
|
||||
if !reflect.DeepEqual(expect, texts) {
|
||||
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
|
||||
}
|
||||
|
||||
texts = textsFromRelationship(tables, users, users.ToManyRelationships[2])
|
||||
texts = textsFromRelationship(tables, pilots, pilots.ToManyRelationships[2])
|
||||
expect = RelationshipToManyTexts{}
|
||||
expect.LocalTable.NameGo = "User"
|
||||
expect.LocalTable.NameSingular = "user"
|
||||
expect.LocalTable.NameGo = "Pilot"
|
||||
expect.LocalTable.NameSingular = "pilot"
|
||||
|
||||
expect.ForeignTable.NameGo = "Notification"
|
||||
expect.ForeignTable.NameSingular = "notification"
|
||||
expect.ForeignTable.NamePluralGo = "Notifications"
|
||||
expect.ForeignTable.NameHumanReadable = "notifications"
|
||||
expect.ForeignTable.Slice = "NotificationSlice"
|
||||
expect.ForeignTable.NameGo = "License"
|
||||
expect.ForeignTable.NameSingular = "license"
|
||||
expect.ForeignTable.NamePluralGo = "Licenses"
|
||||
expect.ForeignTable.NameHumanReadable = "licenses"
|
||||
expect.ForeignTable.Slice = "LicenseSlice"
|
||||
|
||||
expect.Function.Name = "SourceNotifications"
|
||||
expect.Function.Receiver = "u"
|
||||
expect.Function.Name = "SourceLicenses"
|
||||
expect.Function.Receiver = "p"
|
||||
expect.Function.LocalAssignment = "ID"
|
||||
expect.Function.ForeignAssignment = "SourceID.Int32"
|
||||
|
||||
|
|
@ -161,19 +162,19 @@ func TestTextsFromRelationship(t *testing.T) {
|
|||
t.Errorf("Want:\n%s\nGot:\n%s\n", spew.Sdump(expect), spew.Sdump(texts))
|
||||
}
|
||||
|
||||
texts = textsFromRelationship(tables, users, users.ToManyRelationships[3])
|
||||
texts = textsFromRelationship(tables, pilots, pilots.ToManyRelationships[3])
|
||||
expect = RelationshipToManyTexts{}
|
||||
expect.LocalTable.NameGo = "User"
|
||||
expect.LocalTable.NameSingular = "user"
|
||||
expect.LocalTable.NameGo = "Pilot"
|
||||
expect.LocalTable.NameSingular = "pilot"
|
||||
|
||||
expect.ForeignTable.NameGo = "Video"
|
||||
expect.ForeignTable.NameSingular = "video"
|
||||
expect.ForeignTable.NamePluralGo = "Videos"
|
||||
expect.ForeignTable.NameHumanReadable = "videos"
|
||||
expect.ForeignTable.Slice = "VideoSlice"
|
||||
expect.ForeignTable.NameGo = "Jet"
|
||||
expect.ForeignTable.NameSingular = "jet"
|
||||
expect.ForeignTable.NamePluralGo = "Jets"
|
||||
expect.ForeignTable.NameHumanReadable = "jets"
|
||||
expect.ForeignTable.Slice = "JetSlice"
|
||||
|
||||
expect.Function.Name = "Videos"
|
||||
expect.Function.Receiver = "u"
|
||||
expect.Function.Name = "Jets"
|
||||
expect.Function.Receiver = "p"
|
||||
expect.Function.LocalAssignment = "ID"
|
||||
expect.Function.ForeignAssignment = "ID"
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue