2014-07-29 06:59:13 +04:00
package gorm_test
2014-07-29 08:32:58 +04:00
import (
"fmt"
"reflect"
2015-08-01 06:49:34 +03:00
"github.com/jinzhu/gorm"
2014-09-19 17:48:55 +04:00
"github.com/jinzhu/now"
2014-07-29 08:32:58 +04:00
"testing"
"time"
)
2014-07-29 06:59:13 +04:00
func TestFirstAndLast ( t * testing . T ) {
2014-08-28 11:33:43 +04:00
DB . Save ( & User { Name : "user1" , Emails : [ ] Email { { Email : "user1@example.com" } } } )
DB . Save ( & User { Name : "user2" , Emails : [ ] Email { { Email : "user2@example.com" } } } )
2014-07-29 06:59:13 +04:00
var user1 , user2 , user3 , user4 User
2014-08-28 11:33:43 +04:00
DB . First ( & user1 )
2015-03-12 07:01:53 +03:00
DB . Order ( "id" ) . Limit ( 1 ) . Find ( & user2 )
2014-07-29 06:59:13 +04:00
2014-08-28 11:33:43 +04:00
DB . Last ( & user3 )
2015-03-12 07:01:53 +03:00
DB . Order ( "id desc" ) . Limit ( 1 ) . Find ( & user4 )
2014-07-29 06:59:13 +04:00
if user1 . Id != user2 . Id || user3 . Id != user4 . Id {
t . Errorf ( "First and Last should by order by primary key" )
}
var users [ ] User
2014-08-28 11:33:43 +04:00
DB . First ( & users )
2014-07-29 06:59:13 +04:00
if len ( users ) != 1 {
t . Errorf ( "Find first record as slice" )
}
2014-08-28 11:33:43 +04:00
if DB . Joins ( "left join emails on emails.user_id = users.id" ) . First ( & User { } ) . Error != nil {
2014-07-29 06:59:13 +04:00
t . Errorf ( "Should not raise any error when order with Join table" )
}
}
func TestFirstAndLastWithNoStdPrimaryKey ( t * testing . T ) {
2014-08-28 11:33:43 +04:00
DB . Save ( & Animal { Name : "animal1" } )
DB . Save ( & Animal { Name : "animal2" } )
2014-07-29 06:59:13 +04:00
var animal1 , animal2 , animal3 , animal4 Animal
2014-08-28 11:33:43 +04:00
DB . First ( & animal1 )
2015-03-12 07:01:53 +03:00
DB . Order ( "counter" ) . Limit ( 1 ) . Find ( & animal2 )
2014-07-29 06:59:13 +04:00
2014-08-28 11:33:43 +04:00
DB . Last ( & animal3 )
2015-03-12 07:01:53 +03:00
DB . Order ( "counter desc" ) . Limit ( 1 ) . Find ( & animal4 )
2014-07-29 06:59:13 +04:00
if animal1 . Counter != animal2 . Counter || animal3 . Counter != animal4 . Counter {
t . Errorf ( "First and Last should work correctly" )
}
}
2014-10-22 19:33:13 +04:00
func TestUIntPrimaryKey ( t * testing . T ) {
var animal Animal
DB . First ( & animal , uint64 ( 1 ) )
if animal . Counter != 1 {
t . Errorf ( "Fetch a record from with a non-int primary key should work, but failed" )
}
DB . Model ( Animal { } ) . Where ( Animal { Counter : uint64 ( 2 ) } ) . Scan ( & animal )
if animal . Counter != 2 {
t . Errorf ( "Fetch a record from with a non-int primary key should work, but failed" )
}
}
2015-10-16 06:49:47 +03:00
func TestStringPrimaryKeyForNumericValueStartingWithZero ( t * testing . T ) {
type AddressByZipCode struct {
2016-01-16 03:38:56 +03:00
ZipCode string ` gorm:"primary_key" `
Address string
2015-10-16 06:49:47 +03:00
}
DB . AutoMigrate ( & AddressByZipCode { } )
DB . Create ( & AddressByZipCode { ZipCode : "00501" , Address : "Holtsville" } )
var address AddressByZipCode
DB . First ( & address , "00501" )
if address . ZipCode != "00501" {
t . Errorf ( "Fetch a record from with a string primary key for a numeric value starting with zero should work, but failed" )
}
}
2014-07-29 06:59:13 +04:00
func TestFindAsSliceOfPointers ( t * testing . T ) {
2014-08-28 11:33:43 +04:00
DB . Save ( & User { Name : "user" } )
2014-07-29 06:59:13 +04:00
var users [ ] User
2014-08-28 11:33:43 +04:00
DB . Find ( & users )
2014-07-29 06:59:13 +04:00
var userPointers [ ] * User
2014-08-28 11:33:43 +04:00
DB . Find ( & userPointers )
2014-07-29 06:59:13 +04:00
if len ( users ) == 0 || len ( users ) != len ( userPointers ) {
t . Errorf ( "Find slice of pointers" )
}
}
2014-07-29 08:32:58 +04:00
func TestSearchWithPlainSQL ( t * testing . T ) {
user1 := User { Name : "PlainSqlUser1" , Age : 1 , Birthday : now . MustParse ( "2000-1-1" ) }
user2 := User { Name : "PlainSqlUser2" , Age : 10 , Birthday : now . MustParse ( "2010-1-1" ) }
user3 := User { Name : "PlainSqlUser3" , Age : 20 , Birthday : now . MustParse ( "2020-1-1" ) }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 )
scopedb := DB . Where ( "name LIKE ?" , "%PlainSqlUser%" )
2014-07-29 08:32:58 +04:00
2014-08-28 11:33:43 +04:00
if DB . Where ( "name = ?" , user1 . Name ) . First ( & User { } ) . RecordNotFound ( ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Search with plain SQL" )
}
2014-08-28 11:33:43 +04:00
if DB . Where ( "name LIKE ?" , "%" + user1 . Name + "%" ) . First ( & User { } ) . RecordNotFound ( ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Search with plan SQL (regexp)" )
}
var users [ ] User
2014-08-28 11:33:43 +04:00
DB . Find ( & users , "name LIKE ? and age > ?" , "%PlainSqlUser%" , 1 )
2014-07-29 08:32:58 +04:00
if len ( users ) != 2 {
t . Errorf ( "Should found 2 users that age > 1, but got %v" , len ( users ) )
}
2014-08-28 11:33:43 +04:00
DB . Where ( "name LIKE ?" , "%PlainSqlUser%" ) . Where ( "age >= ?" , 1 ) . Find ( & users )
2014-07-29 08:32:58 +04:00
if len ( users ) != 3 {
t . Errorf ( "Should found 3 users that age >= 1, but got %v" , len ( users ) )
}
scopedb . Where ( "age <> ?" , 20 ) . Find ( & users )
if len ( users ) != 2 {
t . Errorf ( "Should found 2 users age != 20, but got %v" , len ( users ) )
}
scopedb . Where ( "birthday > ?" , now . MustParse ( "2000-1-1" ) ) . Find ( & users )
if len ( users ) != 2 {
t . Errorf ( "Should found 2 users's birthday > 2000-1-1, but got %v" , len ( users ) )
}
scopedb . Where ( "birthday > ?" , "2002-10-10" ) . Find ( & users )
if len ( users ) != 2 {
t . Errorf ( "Should found 2 users's birthday >= 2002-10-10, but got %v" , len ( users ) )
}
scopedb . Where ( "birthday >= ?" , "2010-1-1" ) . Where ( "birthday < ?" , "2020-1-1" ) . Find ( & users )
if len ( users ) != 1 {
t . Errorf ( "Should found 1 users's birthday < 2020-1-1 and >= 2010-1-1, but got %v" , len ( users ) )
}
2014-08-28 11:33:43 +04:00
DB . Where ( "name in (?)" , [ ] string { user1 . Name , user2 . Name } ) . Find ( & users )
2014-07-29 08:32:58 +04:00
if len ( users ) != 2 {
t . Errorf ( "Should found 2 users, but got %v" , len ( users ) )
}
2014-08-28 11:33:43 +04:00
DB . Where ( "id in (?)" , [ ] int64 { user1 . Id , user2 . Id , user3 . Id } ) . Find ( & users )
2014-07-29 08:32:58 +04:00
if len ( users ) != 3 {
t . Errorf ( "Should found 3 users, but got %v" , len ( users ) )
}
2014-08-28 11:33:43 +04:00
DB . Where ( "id in (?)" , user1 . Id ) . Find ( & users )
2014-07-29 08:32:58 +04:00
if len ( users ) != 1 {
t . Errorf ( "Should found 1 users, but got %v" , len ( users ) )
}
2014-12-19 06:46:48 +03:00
if DB . Where ( "name = ?" , "none existing" ) . Find ( & [ ] User { } ) . RecordNotFound ( ) {
t . Errorf ( "Should not get RecordNotFound error when looking for none existing records" )
2014-07-29 08:32:58 +04:00
}
}
func TestSearchWithStruct ( t * testing . T ) {
user1 := User { Name : "StructSearchUser1" , Age : 1 , Birthday : now . MustParse ( "2000-1-1" ) }
user2 := User { Name : "StructSearchUser2" , Age : 10 , Birthday : now . MustParse ( "2010-1-1" ) }
user3 := User { Name : "StructSearchUser3" , Age : 20 , Birthday : now . MustParse ( "2020-1-1" ) }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 )
2014-07-29 08:32:58 +04:00
2014-08-28 11:33:43 +04:00
if DB . Where ( user1 . Id ) . First ( & User { } ) . RecordNotFound ( ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Search with primary key" )
}
2014-08-28 11:33:43 +04:00
if DB . First ( & User { } , user1 . Id ) . RecordNotFound ( ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Search with primary key as inline condition" )
}
2014-08-28 11:33:43 +04:00
if DB . First ( & User { } , fmt . Sprintf ( "%v" , user1 . Id ) ) . RecordNotFound ( ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Search with primary key as inline condition" )
}
var users [ ] User
2014-08-28 11:33:43 +04:00
DB . Where ( [ ] int64 { user1 . Id , user2 . Id , user3 . Id } ) . Find ( & users )
2014-07-29 08:32:58 +04:00
if len ( users ) != 3 {
t . Errorf ( "Should found 3 users when search with primary keys, but got %v" , len ( users ) )
}
var user User
2014-08-28 11:33:43 +04:00
DB . First ( & user , & User { Name : user1 . Name } )
2014-07-29 08:32:58 +04:00
if user . Id == 0 || user . Name != user1 . Name {
t . Errorf ( "Search first record with inline pointer of struct" )
}
2014-08-28 11:33:43 +04:00
DB . First ( & user , User { Name : user1 . Name } )
2014-07-29 08:32:58 +04:00
if user . Id == 0 || user . Name != user . Name {
t . Errorf ( "Search first record with inline struct" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : user1 . Name } ) . First ( & user )
2014-07-29 08:32:58 +04:00
if user . Id == 0 || user . Name != user1 . Name {
t . Errorf ( "Search first record with where struct" )
}
2014-08-28 11:33:43 +04:00
DB . Find ( & users , & User { Name : user2 . Name } )
2014-07-29 08:32:58 +04:00
if len ( users ) != 1 {
t . Errorf ( "Search all records with inline struct" )
}
}
func TestSearchWithMap ( t * testing . T ) {
2016-01-16 03:38:56 +03:00
productID := 1
2014-07-29 08:32:58 +04:00
user1 := User { Name : "MapSearchUser1" , Age : 1 , Birthday : now . MustParse ( "2000-1-1" ) }
user2 := User { Name : "MapSearchUser2" , Age : 10 , Birthday : now . MustParse ( "2010-1-1" ) }
user3 := User { Name : "MapSearchUser3" , Age : 20 , Birthday : now . MustParse ( "2020-1-1" ) }
2016-01-16 03:38:56 +03:00
user4 := User { Name : "MapSearchUser4" , Age : 30 , Birthday : now . MustParse ( "2020-1-1" ) , ProductID : & productID }
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 ) . Save ( & user4 )
2014-07-29 08:32:58 +04:00
var user User
2014-08-28 11:33:43 +04:00
DB . First ( & user , map [ string ] interface { } { "name" : user1 . Name } )
2014-07-29 08:32:58 +04:00
if user . Id == 0 || user . Name != user1 . Name {
t . Errorf ( "Search first record with inline map" )
}
user = User { }
2014-08-28 11:33:43 +04:00
DB . Where ( map [ string ] interface { } { "name" : user2 . Name } ) . First ( & user )
2014-07-29 08:32:58 +04:00
if user . Id == 0 || user . Name != user2 . Name {
t . Errorf ( "Search first record with where map" )
}
var users [ ] User
2014-08-28 11:33:43 +04:00
DB . Where ( map [ string ] interface { } { "name" : user3 . Name } ) . Find ( & users )
2014-07-29 08:32:58 +04:00
if len ( users ) != 1 {
t . Errorf ( "Search all records with inline map" )
}
2014-08-28 11:33:43 +04:00
DB . Find ( & users , map [ string ] interface { } { "name" : user3 . Name } )
2014-07-29 08:32:58 +04:00
if len ( users ) != 1 {
t . Errorf ( "Search all records with inline map" )
}
2016-01-16 03:38:56 +03:00
DB . Find ( & users , map [ string ] interface { } { "name" : user4 . Name , "product_id" : nil } )
if len ( users ) != 0 {
t . Errorf ( "Search all records with inline map containing null value finding 0 records" )
}
DB . Find ( & users , map [ string ] interface { } { "name" : user1 . Name , "product_id" : nil } )
if len ( users ) != 1 {
t . Errorf ( "Search all records with inline map containing null value finding 1 record" )
}
DB . Find ( & users , map [ string ] interface { } { "name" : user4 . Name , "product_id" : productID } )
if len ( users ) != 1 {
t . Errorf ( "Search all records with inline multiple value map" )
}
2014-07-29 08:32:58 +04:00
}
2015-01-04 15:47:25 +03:00
func TestSearchWithEmptyChain ( t * testing . T ) {
user1 := User { Name : "ChainSearchUser1" , Age : 1 , Birthday : now . MustParse ( "2000-1-1" ) }
user2 := User { Name : "ChainearchUser2" , Age : 10 , Birthday : now . MustParse ( "2010-1-1" ) }
user3 := User { Name : "ChainearchUser3" , Age : 20 , Birthday : now . MustParse ( "2020-1-1" ) }
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 )
if DB . Where ( "" ) . Where ( "" ) . First ( & User { } ) . Error != nil {
t . Errorf ( "Should not raise any error if searching with empty strings" )
}
if DB . Where ( & User { } ) . Where ( "name = ?" , user1 . Name ) . First ( & User { } ) . Error != nil {
t . Errorf ( "Should not raise any error if searching with empty struct" )
}
if DB . Where ( map [ string ] interface { } { } ) . Where ( "name = ?" , user1 . Name ) . First ( & User { } ) . Error != nil {
t . Errorf ( "Should not raise any error if searching with empty map" )
}
}
2014-07-29 08:32:58 +04:00
func TestSelect ( t * testing . T ) {
user1 := User { Name : "SelectUser1" }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 )
2014-07-29 08:32:58 +04:00
var user User
2014-08-28 11:33:43 +04:00
DB . Where ( "name = ?" , user1 . Name ) . Select ( "name" ) . Find ( & user )
2014-07-29 08:32:58 +04:00
if user . Id != 0 {
t . Errorf ( "Should not have ID because only selected name, %+v" , user . Id )
}
if user . Name != user1 . Name {
t . Errorf ( "Should have user Name when selected it" )
}
}
func TestOrderAndPluck ( t * testing . T ) {
user1 := User { Name : "OrderPluckUser1" , Age : 1 }
user2 := User { Name : "OrderPluckUser2" , Age : 10 }
user3 := User { Name : "OrderPluckUser3" , Age : 20 }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 )
scopedb := DB . Model ( & User { } ) . Where ( "name like ?" , "%OrderPluckUser%" )
2014-07-29 08:32:58 +04:00
var ages [ ] int64
scopedb . Order ( "age desc" ) . Pluck ( "age" , & ages )
2014-09-19 17:48:55 +04:00
if ages [ 0 ] != 20 {
2014-07-29 08:32:58 +04:00
t . Errorf ( "The first age should be 20 when order with age desc" )
}
var ages1 , ages2 [ ] int64
2014-09-16 00:03:14 +04:00
scopedb . Order ( "age desc" ) . Pluck ( "age" , & ages1 ) . Pluck ( "age" , & ages2 )
2014-07-29 08:32:58 +04:00
if ! reflect . DeepEqual ( ages1 , ages2 ) {
2014-09-16 19:32:35 +04:00
t . Errorf ( "The first order is the primary order" )
}
2014-07-29 08:32:58 +04:00
var ages3 , ages4 [ ] int64
scopedb . Model ( & User { } ) . Order ( "age desc" ) . Pluck ( "age" , & ages3 ) . Order ( "age" , true ) . Pluck ( "age" , & ages4 )
if reflect . DeepEqual ( ages3 , ages4 ) {
t . Errorf ( "Reorder should work" )
}
var names [ ] string
var ages5 [ ] int64
scopedb . Model ( User { } ) . Order ( "name" ) . Order ( "age desc" ) . Pluck ( "age" , & ages5 ) . Pluck ( "name" , & names )
2014-09-16 19:32:35 +04:00
if names != nil && ages5 != nil {
if ! ( names [ 0 ] == user1 . Name && names [ 1 ] == user2 . Name && names [ 2 ] == user3 . Name && ages5 [ 2 ] == 20 ) {
t . Errorf ( "Order with multiple orders" )
}
} else {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Order with multiple orders" )
}
2014-08-28 11:33:43 +04:00
DB . Model ( User { } ) . Select ( "name, age" ) . Find ( & [ ] User { } )
2014-07-29 08:32:58 +04:00
}
func TestLimit ( t * testing . T ) {
user1 := User { Name : "LimitUser1" , Age : 1 }
user2 := User { Name : "LimitUser2" , Age : 10 }
user3 := User { Name : "LimitUser3" , Age : 20 }
user4 := User { Name : "LimitUser4" , Age : 10 }
user5 := User { Name : "LimitUser5" , Age : 20 }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 ) . Save ( & user4 ) . Save ( & user5 )
2014-07-29 08:32:58 +04:00
var users1 , users2 , users3 [ ] User
2014-08-28 11:33:43 +04:00
DB . Order ( "age desc" ) . Limit ( 3 ) . Find ( & users1 ) . Limit ( 5 ) . Find ( & users2 ) . Limit ( - 1 ) . Find ( & users3 )
2014-07-29 08:32:58 +04:00
if len ( users1 ) != 3 || len ( users2 ) != 5 || len ( users3 ) <= 5 {
t . Errorf ( "Limit should works" )
}
}
func TestOffset ( t * testing . T ) {
for i := 0 ; i < 20 ; i ++ {
2014-08-28 11:33:43 +04:00
DB . Save ( & User { Name : fmt . Sprintf ( "OffsetUser%v" , i ) } )
2014-07-29 08:32:58 +04:00
}
var users1 , users2 , users3 , users4 [ ] User
2014-08-28 11:33:43 +04:00
DB . Limit ( 100 ) . Order ( "age desc" ) . Find ( & users1 ) . Offset ( 3 ) . Find ( & users2 ) . Offset ( 5 ) . Find ( & users3 ) . Offset ( - 1 ) . Find ( & users4 )
2014-07-29 08:32:58 +04:00
if ( len ( users1 ) != len ( users4 ) ) || ( len ( users1 ) - len ( users2 ) != 3 ) || ( len ( users1 ) - len ( users3 ) != 5 ) {
t . Errorf ( "Offset should work" )
}
}
func TestOr ( t * testing . T ) {
user1 := User { Name : "OrUser1" , Age : 1 }
user2 := User { Name : "OrUser2" , Age : 10 }
user3 := User { Name : "OrUser3" , Age : 20 }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 )
2014-07-29 08:32:58 +04:00
var users [ ] User
2014-08-28 11:33:43 +04:00
DB . Where ( "name = ?" , user1 . Name ) . Or ( "name = ?" , user2 . Name ) . Find ( & users )
2014-07-29 08:32:58 +04:00
if len ( users ) != 2 {
t . Errorf ( "Find users with or" )
}
}
func TestCount ( t * testing . T ) {
user1 := User { Name : "CountUser1" , Age : 1 }
user2 := User { Name : "CountUser2" , Age : 10 }
user3 := User { Name : "CountUser3" , Age : 20 }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 )
2014-07-29 08:32:58 +04:00
var count , count1 , count2 int64
var users [ ] User
2014-08-28 11:33:43 +04:00
if err := DB . Where ( "name = ?" , user1 . Name ) . Or ( "name = ?" , user3 . Name ) . Find ( & users ) . Count ( & count ) . Error ; err != nil {
2014-07-29 08:32:58 +04:00
t . Errorf ( fmt . Sprintf ( "Count should work, but got err %v" , err ) )
}
if count != int64 ( len ( users ) ) {
t . Errorf ( "Count() method should get correct value" )
}
2014-08-28 11:33:43 +04:00
DB . Model ( & User { } ) . Where ( "name = ?" , user1 . Name ) . Count ( & count1 ) . Or ( "name in (?)" , [ ] string { user2 . Name , user3 . Name } ) . Count ( & count2 )
2014-07-29 08:32:58 +04:00
if count1 != 1 || count2 != 3 {
t . Errorf ( "Multiple count in chain" )
}
}
func TestNot ( t * testing . T ) {
2015-03-12 13:01:27 +03:00
DB . Create ( getPreparedUser ( "user1" , "not" ) )
DB . Create ( getPreparedUser ( "user2" , "not" ) )
DB . Create ( getPreparedUser ( "user3" , "not" ) )
2016-01-16 03:38:56 +03:00
user4 := getPreparedUser ( "user4" , "not" )
productID := 1
user4 . ProductID = & productID
DB . Create ( user4 )
2015-03-12 13:01:27 +03:00
DB := DB . Where ( "role = ?" , "not" )
2016-01-16 03:38:56 +03:00
var users1 , users2 , users3 , users4 , users5 , users6 , users7 , users8 , users9 [ ] User
2015-03-12 13:01:27 +03:00
if DB . Find ( & users1 ) . RowsAffected != 4 {
t . Errorf ( "should find 4 not users" )
}
2014-08-28 11:33:43 +04:00
DB . Not ( users1 [ 0 ] . Id ) . Find ( & users2 )
2014-07-29 08:32:58 +04:00
if len ( users1 ) - len ( users2 ) != 1 {
t . Errorf ( "Should ignore the first users with Not" )
}
2014-08-28 11:33:43 +04:00
DB . Not ( [ ] int { } ) . Find ( & users3 )
2014-07-29 08:32:58 +04:00
if len ( users1 ) - len ( users3 ) != 0 {
t . Errorf ( "Should find all users with a blank condition" )
}
var name3Count int64
2015-03-12 13:01:27 +03:00
DB . Table ( "users" ) . Where ( "name = ?" , "user3" ) . Count ( & name3Count )
DB . Not ( "name" , "user3" ) . Find ( & users4 )
2014-07-29 08:32:58 +04:00
if len ( users1 ) - len ( users4 ) != int ( name3Count ) {
t . Errorf ( "Should find all users's name not equal 3" )
}
2015-03-12 13:01:27 +03:00
DB . Not ( "name = ?" , "user3" ) . Find ( & users4 )
2014-07-29 08:32:58 +04:00
if len ( users1 ) - len ( users4 ) != int ( name3Count ) {
t . Errorf ( "Should find all users's name not equal 3" )
}
2015-03-12 13:01:27 +03:00
DB . Not ( "name <> ?" , "user3" ) . Find ( & users4 )
2014-07-29 08:32:58 +04:00
if len ( users4 ) != int ( name3Count ) {
t . Errorf ( "Should find all users's name not equal 3" )
}
2015-03-12 13:01:27 +03:00
DB . Not ( User { Name : "user3" } ) . Find ( & users5 )
2014-07-29 08:32:58 +04:00
if len ( users1 ) - len ( users5 ) != int ( name3Count ) {
t . Errorf ( "Should find all users's name not equal 3" )
}
2015-03-12 13:01:27 +03:00
DB . Not ( map [ string ] interface { } { "name" : "user3" } ) . Find ( & users6 )
2014-07-29 08:32:58 +04:00
if len ( users1 ) - len ( users6 ) != int ( name3Count ) {
t . Errorf ( "Should find all users's name not equal 3" )
}
2016-01-16 03:38:56 +03:00
DB . Not ( map [ string ] interface { } { "name" : "user3" , "product_id" : nil } ) . Find ( & users7 )
if len ( users7 ) != 1 {
t . Errorf ( "Should find all user's name not equal to 3 who do not have product ids" )
}
DB . Not ( "name" , [ ] string { "user3" } ) . Find ( & users8 )
if len ( users1 ) - len ( users8 ) != int ( name3Count ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Should find all users's name not equal 3" )
}
var name2Count int64
2015-03-12 13:01:27 +03:00
DB . Table ( "users" ) . Where ( "name = ?" , "user2" ) . Count ( & name2Count )
2016-01-16 03:38:56 +03:00
DB . Not ( "name" , [ ] string { "user3" , "user2" } ) . Find ( & users9 )
if len ( users1 ) - len ( users9 ) != ( int ( name3Count ) + int ( name2Count ) ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "Should find all users's name not equal 3" )
}
}
func TestFillSmallerStruct ( t * testing . T ) {
user1 := User { Name : "SmallerUser" , Age : 100 }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 )
2014-07-29 08:32:58 +04:00
type SimpleUser struct {
Name string
Id int64
UpdatedAt time . Time
CreatedAt time . Time
}
var simpleUser SimpleUser
2014-08-28 11:33:43 +04:00
DB . Table ( "users" ) . Where ( "name = ?" , user1 . Name ) . First ( & simpleUser )
2014-07-29 08:32:58 +04:00
if simpleUser . Id == 0 || simpleUser . Name == "" {
t . Errorf ( "Should fill data correctly into smaller struct" )
}
}
func TestFindOrInitialize ( t * testing . T ) {
var user1 , user2 , user3 , user4 , user5 , user6 User
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or init" , Age : 33 } ) . FirstOrInit ( & user1 )
2014-07-29 08:32:58 +04:00
if user1 . Name != "find or init" || user1 . Id != 0 || user1 . Age != 33 {
t . Errorf ( "user should be initialized with search value" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( User { Name : "find or init" , Age : 33 } ) . FirstOrInit ( & user2 )
2014-07-29 08:32:58 +04:00
if user2 . Name != "find or init" || user2 . Id != 0 || user2 . Age != 33 {
t . Errorf ( "user should be initialized with search value" )
}
2014-08-28 11:33:43 +04:00
DB . FirstOrInit ( & user3 , map [ string ] interface { } { "name" : "find or init 2" } )
2014-07-29 08:32:58 +04:00
if user3 . Name != "find or init 2" || user3 . Id != 0 {
t . Errorf ( "user should be initialized with inline search value" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or init" } ) . Attrs ( User { Age : 44 } ) . FirstOrInit ( & user4 )
2014-07-29 08:32:58 +04:00
if user4 . Name != "find or init" || user4 . Id != 0 || user4 . Age != 44 {
t . Errorf ( "user should be initialized with search value and attrs" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or init" } ) . Assign ( "age" , 44 ) . FirstOrInit ( & user4 )
2014-07-29 08:32:58 +04:00
if user4 . Name != "find or init" || user4 . Id != 0 || user4 . Age != 44 {
t . Errorf ( "user should be initialized with search value and assign attrs" )
}
2014-08-28 11:33:43 +04:00
DB . Save ( & User { Name : "find or init" , Age : 33 } )
DB . Where ( & User { Name : "find or init" } ) . Attrs ( "age" , 44 ) . FirstOrInit ( & user5 )
2014-07-29 08:32:58 +04:00
if user5 . Name != "find or init" || user5 . Id == 0 || user5 . Age != 33 {
t . Errorf ( "user should be found and not initialized by Attrs" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or init" , Age : 33 } ) . FirstOrInit ( & user6 )
2014-07-29 08:32:58 +04:00
if user6 . Name != "find or init" || user6 . Id == 0 || user6 . Age != 33 {
t . Errorf ( "user should be found with FirstOrInit" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or init" } ) . Assign ( User { Age : 44 } ) . FirstOrInit ( & user6 )
2014-07-29 08:32:58 +04:00
if user6 . Name != "find or init" || user6 . Id == 0 || user6 . Age != 44 {
t . Errorf ( "user should be found and updated with assigned attrs" )
}
}
func TestFindOrCreate ( t * testing . T ) {
var user1 , user2 , user3 , user4 , user5 , user6 , user7 , user8 User
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create" , Age : 33 } ) . FirstOrCreate ( & user1 )
2014-07-29 08:32:58 +04:00
if user1 . Name != "find or create" || user1 . Id == 0 || user1 . Age != 33 {
t . Errorf ( "user should be created with search value" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create" , Age : 33 } ) . FirstOrCreate ( & user2 )
2014-07-29 08:32:58 +04:00
if user1 . Id != user2 . Id || user2 . Name != "find or create" || user2 . Id == 0 || user2 . Age != 33 {
t . Errorf ( "user should be created with search value" )
}
2014-08-28 11:33:43 +04:00
DB . FirstOrCreate ( & user3 , map [ string ] interface { } { "name" : "find or create 2" } )
2014-07-29 08:32:58 +04:00
if user3 . Name != "find or create 2" || user3 . Id == 0 {
t . Errorf ( "user should be created with inline search value" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create 3" } ) . Attrs ( "age" , 44 ) . FirstOrCreate ( & user4 )
2014-07-29 08:32:58 +04:00
if user4 . Name != "find or create 3" || user4 . Id == 0 || user4 . Age != 44 {
t . Errorf ( "user should be created with search value and attrs" )
}
updatedAt1 := user4 . UpdatedAt
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create 3" } ) . Assign ( "age" , 55 ) . FirstOrCreate ( & user4 )
2014-07-29 08:32:58 +04:00
if updatedAt1 . Format ( time . RFC3339Nano ) == user4 . UpdatedAt . Format ( time . RFC3339Nano ) {
t . Errorf ( "UpdateAt should be changed when update values with assign" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create 4" } ) . Assign ( User { Age : 44 } ) . FirstOrCreate ( & user4 )
2014-07-29 08:32:58 +04:00
if user4 . Name != "find or create 4" || user4 . Id == 0 || user4 . Age != 44 {
t . Errorf ( "user should be created with search value and assigned attrs" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create" } ) . Attrs ( "age" , 44 ) . FirstOrInit ( & user5 )
2014-07-29 08:32:58 +04:00
if user5 . Name != "find or create" || user5 . Id == 0 || user5 . Age != 33 {
t . Errorf ( "user should be found and not initialized by Attrs" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create" } ) . Assign ( User { Age : 44 } ) . FirstOrCreate ( & user6 )
2014-07-29 08:32:58 +04:00
if user6 . Name != "find or create" || user6 . Id == 0 || user6 . Age != 44 {
t . Errorf ( "user should be found and updated with assigned attrs" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create" } ) . Find ( & user7 )
2014-07-29 08:32:58 +04:00
if user7 . Name != "find or create" || user7 . Id == 0 || user7 . Age != 44 {
t . Errorf ( "user should be found and updated with assigned attrs" )
}
2014-08-28 11:33:43 +04:00
DB . Where ( & User { Name : "find or create embedded struct" } ) . Assign ( User { Age : 44 , CreditCard : CreditCard { Number : "1231231231" } , Emails : [ ] Email { { Email : "jinzhu@assign_embedded_struct.com" } , { Email : "jinzhu-2@assign_embedded_struct.com" } } } ) . FirstOrCreate ( & user8 )
if DB . Where ( "email = ?" , "jinzhu-2@assign_embedded_struct.com" ) . First ( & Email { } ) . RecordNotFound ( ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "embedded struct email should be saved" )
}
2014-08-28 11:33:43 +04:00
if DB . Where ( "email = ?" , "1231231231" ) . First ( & CreditCard { } ) . RecordNotFound ( ) {
2014-07-29 08:32:58 +04:00
t . Errorf ( "embedded struct credit card should be saved" )
}
}
func TestSelectWithEscapedFieldName ( t * testing . T ) {
user1 := User { Name : "EscapedFieldNameUser" , Age : 1 }
user2 := User { Name : "EscapedFieldNameUser" , Age : 10 }
user3 := User { Name : "EscapedFieldNameUser" , Age : 20 }
2014-08-28 11:33:43 +04:00
DB . Save ( & user1 ) . Save ( & user2 ) . Save ( & user3 )
2014-07-29 08:32:58 +04:00
var names [ ] string
2014-08-28 11:33:43 +04:00
DB . Model ( User { } ) . Where ( & User { Name : "EscapedFieldNameUser" } ) . Pluck ( "\"name\"" , & names )
2014-07-29 08:32:58 +04:00
if len ( names ) != 3 {
t . Errorf ( "Expected 3 name, but got: %d" , len ( names ) )
}
}
2014-11-17 15:12:32 +03:00
func TestSelectWithVariables ( t * testing . T ) {
DB . Save ( & User { Name : "jinzhu" } )
2015-08-01 06:49:34 +03:00
rows , _ := DB . Table ( "users" ) . Select ( "? as fake" , gorm . Expr ( "name" ) ) . Rows ( )
2014-11-17 15:12:32 +03:00
if ! rows . Next ( ) {
t . Errorf ( "Should have returned at least one row" )
2014-11-24 13:16:07 +03:00
} else {
columns , _ := rows . Columns ( )
if ! reflect . DeepEqual ( columns , [ ] string { "fake" } ) {
t . Errorf ( "Should only contains one column" )
}
2014-11-17 15:12:32 +03:00
}
}
func TestSelectWithArrayInput ( t * testing . T ) {
DB . Save ( & User { Name : "jinzhu" , Age : 42 } )
var user User
DB . Select ( [ ] string { "name" , "age" } ) . Where ( "age = 42 AND name = 'jinzhu'" ) . First ( & user )
if user . Name != "jinzhu" || user . Age != 42 {
t . Errorf ( "Should have selected both age and name" )
}
}
2015-08-06 22:37:26 +03:00
func TestCurrentDatabase ( t * testing . T ) {
2015-08-11 18:59:59 +03:00
databaseName := DB . CurrentDatabase ( )
if err := DB . Error ; err != nil {
2015-08-06 22:37:26 +03:00
t . Errorf ( "Problem getting current db name: %s" , err )
}
2015-08-11 18:59:59 +03:00
if databaseName == "" {
2015-08-06 22:37:26 +03:00
t . Errorf ( "Current db name returned empty; this should never happen!" )
}
2015-08-11 18:59:59 +03:00
t . Logf ( "Got current db name: %v" , databaseName )
2015-08-06 22:37:26 +03:00
}