1471 lines
35 KiB
Go
1471 lines
35 KiB
Go
// Code generated by SQLBoiler 4.19.5 (https://github.com/aarondl/sqlboiler). DO NOT EDIT.
|
|
// This file is meant to be re-generated in place and/or deleted at any time.
|
|
|
|
package models
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/aarondl/randomize"
|
|
"github.com/aarondl/sqlboiler/v4/boil"
|
|
"github.com/aarondl/sqlboiler/v4/queries"
|
|
"github.com/aarondl/strmangle"
|
|
)
|
|
|
|
var (
|
|
// Relationships sometimes use the reflection helper queries.Equal/queries.Assign
|
|
// so force a package dependency in case they don't.
|
|
_ = queries.Equal
|
|
)
|
|
|
|
func testUsers(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
query := Users()
|
|
|
|
if query.Query == nil {
|
|
t.Error("expected a query, got nothing")
|
|
}
|
|
}
|
|
|
|
func testUsersDelete(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if rowsAff, err := o.Delete(ctx, tx); err != nil {
|
|
t.Error(err)
|
|
} else if rowsAff != 1 {
|
|
t.Error("should only have deleted one row, but affected:", rowsAff)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 0 {
|
|
t.Error("want zero records, got:", count)
|
|
}
|
|
}
|
|
|
|
func testUsersQueryDeleteAll(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if rowsAff, err := Users().DeleteAll(ctx, tx); err != nil {
|
|
t.Error(err)
|
|
} else if rowsAff != 1 {
|
|
t.Error("should only have deleted one row, but affected:", rowsAff)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 0 {
|
|
t.Error("want zero records, got:", count)
|
|
}
|
|
}
|
|
|
|
func testUsersSliceDeleteAll(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
slice := UserSlice{o}
|
|
|
|
if rowsAff, err := slice.DeleteAll(ctx, tx); err != nil {
|
|
t.Error(err)
|
|
} else if rowsAff != 1 {
|
|
t.Error("should only have deleted one row, but affected:", rowsAff)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 0 {
|
|
t.Error("want zero records, got:", count)
|
|
}
|
|
}
|
|
|
|
func testUsersExists(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
e, err := UserExists(ctx, tx, o.ID)
|
|
if err != nil {
|
|
t.Errorf("Unable to check if User exists: %s", err)
|
|
}
|
|
if !e {
|
|
t.Errorf("Expected UserExists to return true, but got false.")
|
|
}
|
|
}
|
|
|
|
func testUsersFind(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
userFound, err := FindUser(ctx, tx, o.ID)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if userFound == nil {
|
|
t.Error("want a record, got nil")
|
|
}
|
|
}
|
|
|
|
func testUsersBind(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err = Users().Bind(ctx, tx, o); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func testUsersOne(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if x, err := Users().One(ctx, tx); err != nil {
|
|
t.Error(err)
|
|
} else if x == nil {
|
|
t.Error("expected to get a non nil record")
|
|
}
|
|
}
|
|
|
|
func testUsersAll(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
userOne := &User{}
|
|
userTwo := &User{}
|
|
if err = randomize.Struct(seed, userOne, userDBTypes, false, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
if err = randomize.Struct(seed, userTwo, userDBTypes, false, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = userOne.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = userTwo.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
slice, err := Users().All(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(slice) != 2 {
|
|
t.Error("want 2 records, got:", len(slice))
|
|
}
|
|
}
|
|
|
|
func testUsersCount(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var err error
|
|
seed := randomize.NewSeed()
|
|
userOne := &User{}
|
|
userTwo := &User{}
|
|
if err = randomize.Struct(seed, userOne, userDBTypes, false, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
if err = randomize.Struct(seed, userTwo, userDBTypes, false, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = userOne.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = userTwo.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 2 {
|
|
t.Error("want 2 records, got:", count)
|
|
}
|
|
}
|
|
|
|
func testUsersInsert(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 1 {
|
|
t.Error("want one record, got:", count)
|
|
}
|
|
}
|
|
|
|
func testUsersInsertWhitelist(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Whitelist(strmangle.SetMerge(userPrimaryKeyColumns, userColumnsWithoutDefault)...)); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 1 {
|
|
t.Error("want one record, got:", count)
|
|
}
|
|
}
|
|
|
|
func testUserOneToOneAppUserProfileUsingAppUserProfile(t *testing.T) {
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var foreign AppUserProfile
|
|
var local User
|
|
|
|
seed := randomize.NewSeed()
|
|
if err := randomize.Struct(seed, &foreign, appUserProfileDBTypes, true, appUserProfileColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize AppUserProfile struct: %s", err)
|
|
}
|
|
if err := randomize.Struct(seed, &local, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if err := local.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
foreign.UserID = local.ID
|
|
if err := foreign.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
check, err := local.AppUserProfile().One(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if check.UserID != foreign.UserID {
|
|
t.Errorf("want: %v, got %v", foreign.UserID, check.UserID)
|
|
}
|
|
|
|
slice := UserSlice{&local}
|
|
if err = local.L.LoadAppUserProfile(ctx, tx, false, (*[]*User)(&slice), nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if local.R.AppUserProfile == nil {
|
|
t.Error("struct should have been eager loaded")
|
|
}
|
|
|
|
local.R.AppUserProfile = nil
|
|
if err = local.L.LoadAppUserProfile(ctx, tx, true, &local, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if local.R.AppUserProfile == nil {
|
|
t.Error("struct should have been eager loaded")
|
|
}
|
|
|
|
}
|
|
|
|
func testUserOneToOneSetOpAppUserProfileUsingAppUserProfile(t *testing.T) {
|
|
var err error
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c AppUserProfile
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, false, strmangle.SetComplement(userPrimaryKeyColumns, userColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = randomize.Struct(seed, &b, appUserProfileDBTypes, false, strmangle.SetComplement(appUserProfilePrimaryKeyColumns, appUserProfileColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = randomize.Struct(seed, &c, appUserProfileDBTypes, false, strmangle.SetComplement(appUserProfilePrimaryKeyColumns, appUserProfileColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
for i, x := range []*AppUserProfile{&b, &c} {
|
|
err = a.SetAppUserProfile(ctx, tx, i != 0, x)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if a.R.AppUserProfile != x {
|
|
t.Error("relationship struct not set to correct value")
|
|
}
|
|
if x.R.User != &a {
|
|
t.Error("failed to append to foreign relationship struct")
|
|
}
|
|
|
|
if a.ID != x.UserID {
|
|
t.Error("foreign key was wrong value", a.ID)
|
|
}
|
|
|
|
if exists, err := AppUserProfileExists(ctx, tx, x.UserID); err != nil {
|
|
t.Fatal(err)
|
|
} else if !exists {
|
|
t.Error("want 'x' to exist")
|
|
}
|
|
|
|
if a.ID != x.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, x.UserID)
|
|
}
|
|
|
|
if _, err = x.Delete(ctx, tx); err != nil {
|
|
t.Fatal("failed to delete x", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func testUserToManyAccessTokens(t *testing.T) {
|
|
var err error
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c AccessToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err = randomize.Struct(seed, &b, accessTokenDBTypes, false, accessTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = randomize.Struct(seed, &c, accessTokenDBTypes, false, accessTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
b.UserID = a.ID
|
|
c.UserID = a.ID
|
|
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
check, err := a.AccessTokens().All(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bFound, cFound := false, false
|
|
for _, v := range check {
|
|
if v.UserID == b.UserID {
|
|
bFound = true
|
|
}
|
|
if v.UserID == c.UserID {
|
|
cFound = true
|
|
}
|
|
}
|
|
|
|
if !bFound {
|
|
t.Error("expected to find b")
|
|
}
|
|
if !cFound {
|
|
t.Error("expected to find c")
|
|
}
|
|
|
|
slice := UserSlice{&a}
|
|
if err = a.L.LoadAccessTokens(ctx, tx, false, (*[]*User)(&slice), nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.AccessTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
a.R.AccessTokens = nil
|
|
if err = a.L.LoadAccessTokens(ctx, tx, true, &a, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.AccessTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
if t.Failed() {
|
|
t.Logf("%#v", check)
|
|
}
|
|
}
|
|
|
|
func testUserToManyConfirmationTokens(t *testing.T) {
|
|
var err error
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c ConfirmationToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err = randomize.Struct(seed, &b, confirmationTokenDBTypes, false, confirmationTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = randomize.Struct(seed, &c, confirmationTokenDBTypes, false, confirmationTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
b.UserID = a.ID
|
|
c.UserID = a.ID
|
|
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
check, err := a.ConfirmationTokens().All(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bFound, cFound := false, false
|
|
for _, v := range check {
|
|
if v.UserID == b.UserID {
|
|
bFound = true
|
|
}
|
|
if v.UserID == c.UserID {
|
|
cFound = true
|
|
}
|
|
}
|
|
|
|
if !bFound {
|
|
t.Error("expected to find b")
|
|
}
|
|
if !cFound {
|
|
t.Error("expected to find c")
|
|
}
|
|
|
|
slice := UserSlice{&a}
|
|
if err = a.L.LoadConfirmationTokens(ctx, tx, false, (*[]*User)(&slice), nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.ConfirmationTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
a.R.ConfirmationTokens = nil
|
|
if err = a.L.LoadConfirmationTokens(ctx, tx, true, &a, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.ConfirmationTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
if t.Failed() {
|
|
t.Logf("%#v", check)
|
|
}
|
|
}
|
|
|
|
func testUserToManyPasswordResetTokens(t *testing.T) {
|
|
var err error
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c PasswordResetToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err = randomize.Struct(seed, &b, passwordResetTokenDBTypes, false, passwordResetTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = randomize.Struct(seed, &c, passwordResetTokenDBTypes, false, passwordResetTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
b.UserID = a.ID
|
|
c.UserID = a.ID
|
|
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
check, err := a.PasswordResetTokens().All(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bFound, cFound := false, false
|
|
for _, v := range check {
|
|
if v.UserID == b.UserID {
|
|
bFound = true
|
|
}
|
|
if v.UserID == c.UserID {
|
|
cFound = true
|
|
}
|
|
}
|
|
|
|
if !bFound {
|
|
t.Error("expected to find b")
|
|
}
|
|
if !cFound {
|
|
t.Error("expected to find c")
|
|
}
|
|
|
|
slice := UserSlice{&a}
|
|
if err = a.L.LoadPasswordResetTokens(ctx, tx, false, (*[]*User)(&slice), nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.PasswordResetTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
a.R.PasswordResetTokens = nil
|
|
if err = a.L.LoadPasswordResetTokens(ctx, tx, true, &a, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.PasswordResetTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
if t.Failed() {
|
|
t.Logf("%#v", check)
|
|
}
|
|
}
|
|
|
|
func testUserToManyPushTokens(t *testing.T) {
|
|
var err error
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c PushToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err = randomize.Struct(seed, &b, pushTokenDBTypes, false, pushTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = randomize.Struct(seed, &c, pushTokenDBTypes, false, pushTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
b.UserID = a.ID
|
|
c.UserID = a.ID
|
|
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
check, err := a.PushTokens().All(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bFound, cFound := false, false
|
|
for _, v := range check {
|
|
if v.UserID == b.UserID {
|
|
bFound = true
|
|
}
|
|
if v.UserID == c.UserID {
|
|
cFound = true
|
|
}
|
|
}
|
|
|
|
if !bFound {
|
|
t.Error("expected to find b")
|
|
}
|
|
if !cFound {
|
|
t.Error("expected to find c")
|
|
}
|
|
|
|
slice := UserSlice{&a}
|
|
if err = a.L.LoadPushTokens(ctx, tx, false, (*[]*User)(&slice), nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.PushTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
a.R.PushTokens = nil
|
|
if err = a.L.LoadPushTokens(ctx, tx, true, &a, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.PushTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
if t.Failed() {
|
|
t.Logf("%#v", check)
|
|
}
|
|
}
|
|
|
|
func testUserToManyRefreshTokens(t *testing.T) {
|
|
var err error
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c RefreshToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err = randomize.Struct(seed, &b, refreshTokenDBTypes, false, refreshTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = randomize.Struct(seed, &c, refreshTokenDBTypes, false, refreshTokenColumnsWithDefault...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
b.UserID = a.ID
|
|
c.UserID = a.ID
|
|
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
check, err := a.RefreshTokens().All(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
bFound, cFound := false, false
|
|
for _, v := range check {
|
|
if v.UserID == b.UserID {
|
|
bFound = true
|
|
}
|
|
if v.UserID == c.UserID {
|
|
cFound = true
|
|
}
|
|
}
|
|
|
|
if !bFound {
|
|
t.Error("expected to find b")
|
|
}
|
|
if !cFound {
|
|
t.Error("expected to find c")
|
|
}
|
|
|
|
slice := UserSlice{&a}
|
|
if err = a.L.LoadRefreshTokens(ctx, tx, false, (*[]*User)(&slice), nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.RefreshTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
a.R.RefreshTokens = nil
|
|
if err = a.L.LoadRefreshTokens(ctx, tx, true, &a, nil); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if got := len(a.R.RefreshTokens); got != 2 {
|
|
t.Error("number of eager loaded records wrong, got:", got)
|
|
}
|
|
|
|
if t.Failed() {
|
|
t.Logf("%#v", check)
|
|
}
|
|
}
|
|
|
|
func testUserToManyAddOpAccessTokens(t *testing.T) {
|
|
var err error
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c, d, e AccessToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, false, strmangle.SetComplement(userPrimaryKeyColumns, userColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
foreigners := []*AccessToken{&b, &c, &d, &e}
|
|
for _, x := range foreigners {
|
|
if err = randomize.Struct(seed, x, accessTokenDBTypes, false, strmangle.SetComplement(accessTokenPrimaryKeyColumns, accessTokenColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
foreignersSplitByInsertion := [][]*AccessToken{
|
|
{&b, &c},
|
|
{&d, &e},
|
|
}
|
|
|
|
for i, x := range foreignersSplitByInsertion {
|
|
err = a.AddAccessTokens(ctx, tx, i != 0, x...)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
first := x[0]
|
|
second := x[1]
|
|
|
|
if a.ID != first.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, first.UserID)
|
|
}
|
|
if a.ID != second.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, second.UserID)
|
|
}
|
|
|
|
if first.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
if second.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
|
|
if a.R.AccessTokens[i*2] != first {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
if a.R.AccessTokens[i*2+1] != second {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
|
|
count, err := a.AccessTokens().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want := int64((i + 1) * 2); count != want {
|
|
t.Error("want", want, "got", count)
|
|
}
|
|
}
|
|
}
|
|
func testUserToManyAddOpConfirmationTokens(t *testing.T) {
|
|
var err error
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c, d, e ConfirmationToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, false, strmangle.SetComplement(userPrimaryKeyColumns, userColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
foreigners := []*ConfirmationToken{&b, &c, &d, &e}
|
|
for _, x := range foreigners {
|
|
if err = randomize.Struct(seed, x, confirmationTokenDBTypes, false, strmangle.SetComplement(confirmationTokenPrimaryKeyColumns, confirmationTokenColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
foreignersSplitByInsertion := [][]*ConfirmationToken{
|
|
{&b, &c},
|
|
{&d, &e},
|
|
}
|
|
|
|
for i, x := range foreignersSplitByInsertion {
|
|
err = a.AddConfirmationTokens(ctx, tx, i != 0, x...)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
first := x[0]
|
|
second := x[1]
|
|
|
|
if a.ID != first.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, first.UserID)
|
|
}
|
|
if a.ID != second.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, second.UserID)
|
|
}
|
|
|
|
if first.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
if second.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
|
|
if a.R.ConfirmationTokens[i*2] != first {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
if a.R.ConfirmationTokens[i*2+1] != second {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
|
|
count, err := a.ConfirmationTokens().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want := int64((i + 1) * 2); count != want {
|
|
t.Error("want", want, "got", count)
|
|
}
|
|
}
|
|
}
|
|
func testUserToManyAddOpPasswordResetTokens(t *testing.T) {
|
|
var err error
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c, d, e PasswordResetToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, false, strmangle.SetComplement(userPrimaryKeyColumns, userColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
foreigners := []*PasswordResetToken{&b, &c, &d, &e}
|
|
for _, x := range foreigners {
|
|
if err = randomize.Struct(seed, x, passwordResetTokenDBTypes, false, strmangle.SetComplement(passwordResetTokenPrimaryKeyColumns, passwordResetTokenColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
foreignersSplitByInsertion := [][]*PasswordResetToken{
|
|
{&b, &c},
|
|
{&d, &e},
|
|
}
|
|
|
|
for i, x := range foreignersSplitByInsertion {
|
|
err = a.AddPasswordResetTokens(ctx, tx, i != 0, x...)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
first := x[0]
|
|
second := x[1]
|
|
|
|
if a.ID != first.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, first.UserID)
|
|
}
|
|
if a.ID != second.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, second.UserID)
|
|
}
|
|
|
|
if first.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
if second.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
|
|
if a.R.PasswordResetTokens[i*2] != first {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
if a.R.PasswordResetTokens[i*2+1] != second {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
|
|
count, err := a.PasswordResetTokens().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want := int64((i + 1) * 2); count != want {
|
|
t.Error("want", want, "got", count)
|
|
}
|
|
}
|
|
}
|
|
func testUserToManyAddOpPushTokens(t *testing.T) {
|
|
var err error
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c, d, e PushToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, false, strmangle.SetComplement(userPrimaryKeyColumns, userColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
foreigners := []*PushToken{&b, &c, &d, &e}
|
|
for _, x := range foreigners {
|
|
if err = randomize.Struct(seed, x, pushTokenDBTypes, false, strmangle.SetComplement(pushTokenPrimaryKeyColumns, pushTokenColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
foreignersSplitByInsertion := [][]*PushToken{
|
|
{&b, &c},
|
|
{&d, &e},
|
|
}
|
|
|
|
for i, x := range foreignersSplitByInsertion {
|
|
err = a.AddPushTokens(ctx, tx, i != 0, x...)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
first := x[0]
|
|
second := x[1]
|
|
|
|
if a.ID != first.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, first.UserID)
|
|
}
|
|
if a.ID != second.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, second.UserID)
|
|
}
|
|
|
|
if first.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
if second.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
|
|
if a.R.PushTokens[i*2] != first {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
if a.R.PushTokens[i*2+1] != second {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
|
|
count, err := a.PushTokens().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want := int64((i + 1) * 2); count != want {
|
|
t.Error("want", want, "got", count)
|
|
}
|
|
}
|
|
}
|
|
func testUserToManyAddOpRefreshTokens(t *testing.T) {
|
|
var err error
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
|
|
var a User
|
|
var b, c, d, e RefreshToken
|
|
|
|
seed := randomize.NewSeed()
|
|
if err = randomize.Struct(seed, &a, userDBTypes, false, strmangle.SetComplement(userPrimaryKeyColumns, userColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
foreigners := []*RefreshToken{&b, &c, &d, &e}
|
|
for _, x := range foreigners {
|
|
if err = randomize.Struct(seed, x, refreshTokenDBTypes, false, strmangle.SetComplement(refreshTokenPrimaryKeyColumns, refreshTokenColumnsWithoutDefault)...); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
if err := a.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = b.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err = c.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
foreignersSplitByInsertion := [][]*RefreshToken{
|
|
{&b, &c},
|
|
{&d, &e},
|
|
}
|
|
|
|
for i, x := range foreignersSplitByInsertion {
|
|
err = a.AddRefreshTokens(ctx, tx, i != 0, x...)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
first := x[0]
|
|
second := x[1]
|
|
|
|
if a.ID != first.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, first.UserID)
|
|
}
|
|
if a.ID != second.UserID {
|
|
t.Error("foreign key was wrong value", a.ID, second.UserID)
|
|
}
|
|
|
|
if first.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
if second.R.User != &a {
|
|
t.Error("relationship was not added properly to the foreign slice")
|
|
}
|
|
|
|
if a.R.RefreshTokens[i*2] != first {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
if a.R.RefreshTokens[i*2+1] != second {
|
|
t.Error("relationship struct slice not set to correct value")
|
|
}
|
|
|
|
count, err := a.RefreshTokens().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if want := int64((i + 1) * 2); count != want {
|
|
t.Error("want", want, "got", count)
|
|
}
|
|
}
|
|
}
|
|
|
|
func testUsersReload(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err = o.Reload(ctx, tx); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func testUsersReloadAll(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
slice := UserSlice{o}
|
|
|
|
if err = slice.ReloadAll(ctx, tx); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func testUsersSelect(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
slice, err := Users().All(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(slice) != 1 {
|
|
t.Error("want one record, got:", len(slice))
|
|
}
|
|
}
|
|
|
|
var (
|
|
userDBTypes = map[string]string{`ID`: `uuid`, `Username`: `character varying`, `Password`: `text`, `IsActive`: `boolean`, `Scopes`: `ARRAYtext`, `LastAuthenticatedAt`: `timestamp with time zone`, `CreatedAt`: `timestamp with time zone`, `UpdatedAt`: `timestamp with time zone`, `RequiresConfirmation`: `boolean`}
|
|
_ = bytes.MinRead
|
|
)
|
|
|
|
func testUsersUpdate(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if 0 == len(userPrimaryKeyColumns) {
|
|
t.Skip("Skipping table with no primary key columns")
|
|
}
|
|
if len(userAllColumns) == len(userPrimaryKeyColumns) {
|
|
t.Skip("Skipping table with only primary key columns")
|
|
}
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 1 {
|
|
t.Error("want one record, got:", count)
|
|
}
|
|
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userPrimaryKeyColumns...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if rowsAff, err := o.Update(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
} else if rowsAff != 1 {
|
|
t.Error("should only affect one row but affected", rowsAff)
|
|
}
|
|
}
|
|
|
|
func testUsersSliceUpdateAll(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if len(userAllColumns) == len(userPrimaryKeyColumns) {
|
|
t.Skip("Skipping table with only primary key columns")
|
|
}
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
o := &User{}
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userColumnsWithDefault...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Insert(ctx, tx, boil.Infer()); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if count != 1 {
|
|
t.Error("want one record, got:", count)
|
|
}
|
|
|
|
if err = randomize.Struct(seed, o, userDBTypes, true, userPrimaryKeyColumns...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
// Remove Primary keys and unique columns from what we plan to update
|
|
var fields []string
|
|
if strmangle.StringSliceMatch(userAllColumns, userPrimaryKeyColumns) {
|
|
fields = userAllColumns
|
|
} else {
|
|
fields = strmangle.SetComplement(
|
|
userAllColumns,
|
|
userPrimaryKeyColumns,
|
|
)
|
|
}
|
|
|
|
value := reflect.Indirect(reflect.ValueOf(o))
|
|
typ := reflect.TypeOf(o).Elem()
|
|
n := typ.NumField()
|
|
|
|
updateMap := M{}
|
|
for _, col := range fields {
|
|
for i := 0; i < n; i++ {
|
|
f := typ.Field(i)
|
|
if f.Tag.Get("boil") == col {
|
|
updateMap[col] = value.Field(i).Interface()
|
|
}
|
|
}
|
|
}
|
|
|
|
slice := UserSlice{o}
|
|
if rowsAff, err := slice.UpdateAll(ctx, tx, updateMap); err != nil {
|
|
t.Error(err)
|
|
} else if rowsAff != 1 {
|
|
t.Error("wanted one record updated but got", rowsAff)
|
|
}
|
|
}
|
|
|
|
func testUsersUpsert(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
if len(userAllColumns) == len(userPrimaryKeyColumns) {
|
|
t.Skip("Skipping table with only primary key columns")
|
|
}
|
|
|
|
seed := randomize.NewSeed()
|
|
var err error
|
|
// Attempt the INSERT side of an UPSERT
|
|
o := User{}
|
|
if err = randomize.Struct(seed, &o, userDBTypes, true); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
tx := MustTx(boil.BeginTx(ctx, nil))
|
|
defer func() { _ = tx.Rollback() }()
|
|
if err = o.Upsert(ctx, tx, false, nil, boil.Infer(), boil.Infer()); err != nil {
|
|
t.Errorf("Unable to upsert User: %s", err)
|
|
}
|
|
|
|
count, err := Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if count != 1 {
|
|
t.Error("want one record, got:", count)
|
|
}
|
|
|
|
// Attempt the UPDATE side of an UPSERT
|
|
if err = randomize.Struct(seed, &o, userDBTypes, false, userPrimaryKeyColumns...); err != nil {
|
|
t.Errorf("Unable to randomize User struct: %s", err)
|
|
}
|
|
|
|
if err = o.Upsert(ctx, tx, true, nil, boil.Infer(), boil.Infer()); err != nil {
|
|
t.Errorf("Unable to upsert User: %s", err)
|
|
}
|
|
|
|
count, err = Users().Count(ctx, tx)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if count != 1 {
|
|
t.Error("want one record, got:", count)
|
|
}
|
|
}
|