2014-08-18 08:41:43 +00:00
// Copyright 2014 beego Author. All Rights Reserved.
2014-07-03 15:40:21 +00:00
//
2014-08-18 08:41:43 +00:00
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
2014-07-03 15:40:21 +00:00
//
2014-08-18 08:41:43 +00:00
// http://www.apache.org/licenses/LICENSE-2.0
2014-07-03 15:40:21 +00:00
//
2014-08-18 08:41:43 +00:00
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
2018-05-04 08:38:03 +00:00
// +build go1.8
2013-08-07 11:11:44 +00:00
package orm
import (
"bytes"
2018-05-04 08:38:03 +00:00
"context"
2014-02-28 03:53:35 +00:00
"database/sql"
2013-08-07 11:11:44 +00:00
"fmt"
"io/ioutil"
2016-03-18 13:58:11 +00:00
"math"
2013-08-10 16:15:26 +00:00
"os"
2013-08-07 11:11:44 +00:00
"path/filepath"
"reflect"
"runtime"
"strings"
"testing"
"time"
)
2013-08-10 16:15:26 +00:00
var _ = os . PathSeparator
2013-08-13 09:16:12 +00:00
var (
2015-09-12 13:46:43 +00:00
testDate = formatDate + " -0700"
testDateTime = formatDateTime + " -0700"
2016-04-02 17:57:47 +00:00
testTime = formatTime + " -0700"
2013-08-13 09:16:12 +00:00
)
2015-12-15 16:17:56 +00:00
type argAny [ ] interface { }
// get interface by index from interface slice
func ( a argAny ) Get ( i int , args ... interface { } ) ( r interface { } ) {
if i >= 0 && i < len ( a ) {
r = a [ i ]
}
if len ( args ) > 0 {
r = args [ 0 ]
}
return
}
2015-09-12 13:46:43 +00:00
func ValuesCompare ( is bool , a interface { } , args ... interface { } ) ( ok bool , err error ) {
2013-08-07 11:11:44 +00:00
if len ( args ) == 0 {
2015-09-12 13:46:43 +00:00
return false , fmt . Errorf ( "miss args" )
2013-08-07 11:11:44 +00:00
}
b := args [ 0 ]
arg := argAny ( args )
2013-09-12 11:04:39 +00:00
switch v := a . ( type ) {
case reflect . Kind :
ok = reflect . ValueOf ( b ) . Kind ( ) == v
case time . Time :
if v2 , vo := b . ( time . Time ) ; vo {
if arg . Get ( 1 ) != nil {
format := ToStr ( arg . Get ( 1 ) )
a = v . Format ( format )
b = v2 . Format ( format )
ok = a == b
2013-08-07 11:11:44 +00:00
} else {
2013-09-12 11:04:39 +00:00
err = fmt . Errorf ( "compare datetime miss format" )
goto wrongArg
2013-08-07 11:11:44 +00:00
}
}
2013-09-12 11:04:39 +00:00
default :
ok = ToStr ( a ) == ToStr ( b )
}
ok = is && ok || ! is && ! ok
if ! ok {
if is {
2013-10-14 14:31:35 +00:00
err = fmt . Errorf ( "expected: `%v`, get `%v`" , b , a )
2013-08-07 11:11:44 +00:00
} else {
2013-10-14 14:31:35 +00:00
err = fmt . Errorf ( "expected: `%v`, get `%v`" , b , a )
2013-08-07 11:11:44 +00:00
}
}
2013-09-12 11:04:39 +00:00
2013-08-07 11:11:44 +00:00
wrongArg :
if err != nil {
2015-09-12 13:46:43 +00:00
return false , err
2013-08-07 11:11:44 +00:00
}
2015-09-12 13:46:43 +00:00
return true , nil
2013-08-07 11:11:44 +00:00
}
2013-09-12 11:04:39 +00:00
func AssertIs ( a interface { } , args ... interface { } ) error {
2017-03-17 17:24:45 +00:00
if ok , err := ValuesCompare ( true , a , args ... ) ; ! ok {
2013-08-07 11:11:44 +00:00
return err
}
return nil
}
2013-09-12 11:04:39 +00:00
func AssertNot ( a interface { } , args ... interface { } ) error {
2017-03-17 17:24:45 +00:00
if ok , err := ValuesCompare ( false , a , args ... ) ; ! ok {
2013-08-07 11:11:44 +00:00
return err
}
return nil
}
func getCaller ( skip int ) string {
pc , file , line , _ := runtime . Caller ( skip )
fun := runtime . FuncForPC ( pc )
_ , fn := filepath . Split ( file )
data , err := ioutil . ReadFile ( file )
2013-08-10 16:15:26 +00:00
var codes [ ] string
2013-08-07 11:11:44 +00:00
if err == nil {
lines := bytes . Split ( data , [ ] byte { '\n' } )
2013-08-10 16:15:26 +00:00
n := 10
for i := 0 ; i < n ; i ++ {
o := line - n
if o < 0 {
continue
}
cur := o + i + 1
flag := " "
if cur == line {
flag = ">>"
}
2013-08-13 09:16:12 +00:00
code := fmt . Sprintf ( " %s %5d: %s" , flag , cur , strings . Replace ( string ( lines [ o + i ] ) , "\t" , " " , - 1 ) )
2013-08-10 16:15:26 +00:00
if code != "" {
codes = append ( codes , code )
}
}
2013-08-07 11:11:44 +00:00
}
funName := fun . Name ( )
if i := strings . LastIndex ( funName , "." ) ; i > - 1 {
funName = funName [ i + 1 : ]
}
2017-04-28 13:38:08 +00:00
return fmt . Sprintf ( "%s:%s:%d: \n%s" , fn , funName , line , strings . Join ( codes , "\n" ) )
2013-08-07 11:11:44 +00:00
}
func throwFail ( t * testing . T , err error , args ... interface { } ) {
if err != nil {
2013-08-10 16:15:26 +00:00
con := fmt . Sprintf ( "\t\nError: %s\n%s\n" , err . Error ( ) , getCaller ( 2 ) )
if len ( args ) > 0 {
2013-08-13 09:16:12 +00:00
parts := make ( [ ] string , 0 , len ( args ) )
for _ , arg := range args {
parts = append ( parts , fmt . Sprintf ( "%v" , arg ) )
}
con += " " + strings . Join ( parts , ", " )
2013-08-10 16:15:26 +00:00
}
t . Error ( con )
2013-08-07 11:11:44 +00:00
t . Fail ( )
}
}
func throwFailNow ( t * testing . T , err error , args ... interface { } ) {
if err != nil {
2013-08-10 16:15:26 +00:00
con := fmt . Sprintf ( "\t\nError: %s\n%s\n" , err . Error ( ) , getCaller ( 2 ) )
if len ( args ) > 0 {
2013-08-13 09:16:12 +00:00
parts := make ( [ ] string , 0 , len ( args ) )
for _ , arg := range args {
parts = append ( parts , fmt . Sprintf ( "%v" , arg ) )
}
con += " " + strings . Join ( parts , ", " )
2013-08-10 16:15:26 +00:00
}
t . Error ( con )
2013-08-07 11:11:44 +00:00
t . FailNow ( )
}
}
2014-03-10 12:50:54 +00:00
func TestGetDB ( t * testing . T ) {
if db , err := GetDB ( ) ; err != nil {
throwFailNow ( t , err )
} else {
err = db . Ping ( )
throwFailNow ( t , err )
}
}
2013-08-19 14:37:39 +00:00
func TestSyncDb ( t * testing . T ) {
2014-03-13 15:31:47 +00:00
RegisterModel ( new ( Data ) , new ( DataNull ) , new ( DataCustom ) )
2013-08-19 14:37:39 +00:00
RegisterModel ( new ( User ) )
RegisterModel ( new ( Profile ) )
RegisterModel ( new ( Post ) )
RegisterModel ( new ( Tag ) )
RegisterModel ( new ( Comment ) )
2013-08-22 10:35:26 +00:00
RegisterModel ( new ( UserBig ) )
2013-11-24 11:29:48 +00:00
RegisterModel ( new ( PostTags ) )
2015-12-15 16:17:56 +00:00
RegisterModel ( new ( Group ) )
RegisterModel ( new ( Permission ) )
RegisterModel ( new ( GroupPermissions ) )
2016-03-08 14:24:38 +00:00
RegisterModel ( new ( InLine ) )
2016-03-13 13:01:56 +00:00
RegisterModel ( new ( InLineOneToOne ) )
2016-03-18 13:58:11 +00:00
RegisterModel ( new ( IntegerPk ) )
2016-03-23 13:32:52 +00:00
RegisterModel ( new ( UintPk ) )
2017-01-11 10:51:32 +00:00
RegisterModel ( new ( PtrPk ) )
2013-08-19 14:37:39 +00:00
2016-01-15 00:43:02 +00:00
err := RunSyncdb ( "default" , true , Debug )
2013-08-27 04:33:27 +00:00
throwFail ( t , err )
2013-08-19 14:37:39 +00:00
modelCache . clean ( )
}
func TestRegisterModels ( t * testing . T ) {
2014-03-13 15:31:47 +00:00
RegisterModel ( new ( Data ) , new ( DataNull ) , new ( DataCustom ) )
2013-08-19 14:37:39 +00:00
RegisterModel ( new ( User ) )
RegisterModel ( new ( Profile ) )
RegisterModel ( new ( Post ) )
RegisterModel ( new ( Tag ) )
RegisterModel ( new ( Comment ) )
2013-08-30 04:32:05 +00:00
RegisterModel ( new ( UserBig ) )
2013-11-24 11:29:48 +00:00
RegisterModel ( new ( PostTags ) )
2015-12-15 16:17:56 +00:00
RegisterModel ( new ( Group ) )
RegisterModel ( new ( Permission ) )
RegisterModel ( new ( GroupPermissions ) )
2016-03-08 14:24:38 +00:00
RegisterModel ( new ( InLine ) )
2016-03-13 13:01:56 +00:00
RegisterModel ( new ( InLineOneToOne ) )
2016-03-18 13:58:11 +00:00
RegisterModel ( new ( IntegerPk ) )
2016-03-23 13:32:52 +00:00
RegisterModel ( new ( UintPk ) )
2017-01-11 10:51:32 +00:00
RegisterModel ( new ( PtrPk ) )
2013-08-19 14:37:39 +00:00
BootStrap ( )
dORM = NewOrm ( )
2013-09-09 14:33:41 +00:00
dDbBaser = getDbAlias ( "default" ) . DbBaser
2013-08-19 14:37:39 +00:00
}
2013-08-09 12:14:18 +00:00
func TestModelSyntax ( t * testing . T ) {
2013-08-13 09:16:12 +00:00
user := & User { }
ind := reflect . ValueOf ( user ) . Elem ( )
fn := getFullName ( ind . Type ( ) )
2016-08-30 16:07:19 +00:00
mi , ok := modelCache . getByFullName ( fn )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( ok , true ) )
2013-08-13 09:16:12 +00:00
mi , ok = modelCache . get ( "user" )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( ok , true ) )
2013-08-09 12:14:18 +00:00
if ok {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( mi . fields . GetByName ( "ShouldSkip" ) == nil , true ) )
2013-08-09 12:14:18 +00:00
}
}
2015-09-12 13:46:43 +00:00
var DataValues = map [ string ] interface { } {
2013-08-30 04:32:05 +00:00
"Boolean" : true ,
"Char" : "char" ,
"Text" : "text" ,
2016-04-09 02:22:40 +00:00
"JSON" : ` { "name":"json"} ` ,
2016-04-08 14:26:28 +00:00
"Jsonb" : ` { "name": "jsonb"} ` ,
2016-04-02 17:57:47 +00:00
"Time" : time . Now ( ) ,
2013-08-30 04:32:05 +00:00
"Date" : time . Now ( ) ,
"DateTime" : time . Now ( ) ,
"Byte" : byte ( 1 << 8 - 1 ) ,
"Rune" : rune ( 1 << 31 - 1 ) ,
"Int" : int ( 1 << 31 - 1 ) ,
"Int8" : int8 ( 1 << 7 - 1 ) ,
"Int16" : int16 ( 1 << 15 - 1 ) ,
"Int32" : int32 ( 1 << 31 - 1 ) ,
"Int64" : int64 ( 1 << 63 - 1 ) ,
"Uint" : uint ( 1 << 32 - 1 ) ,
"Uint8" : uint8 ( 1 << 8 - 1 ) ,
"Uint16" : uint16 ( 1 << 16 - 1 ) ,
"Uint32" : uint32 ( 1 << 32 - 1 ) ,
"Uint64" : uint64 ( 1 << 63 - 1 ) , // uint64 values with high bit set are not supported
"Float32" : float32 ( 100.1234 ) ,
"Float64" : float64 ( 100.1234 ) ,
"Decimal" : float64 ( 100.1234 ) ,
}
2013-08-13 09:16:12 +00:00
func TestDataTypes ( t * testing . T ) {
d := Data { }
ind := reflect . Indirect ( reflect . ValueOf ( & d ) )
2015-09-12 13:46:43 +00:00
for name , value := range DataValues {
2016-04-09 02:22:40 +00:00
if name == "JSON" {
2016-04-08 14:26:28 +00:00
continue
}
2013-08-13 09:16:12 +00:00
e := ind . FieldByName ( name )
e . Set ( reflect . ValueOf ( value ) )
}
id , err := dORM . Insert ( & d )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 1 ) )
2013-08-13 09:16:12 +00:00
2015-09-12 13:46:43 +00:00
d = Data { ID : 1 }
2013-08-13 09:16:12 +00:00
err = dORM . Read ( & d )
throwFail ( t , err )
ind = reflect . Indirect ( reflect . ValueOf ( & d ) )
2015-09-12 13:46:43 +00:00
for name , value := range DataValues {
2013-08-13 09:16:12 +00:00
e := ind . FieldByName ( name )
vu := e . Interface ( )
switch name {
case "Date" :
2015-09-12 13:46:43 +00:00
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDate )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDate )
2013-08-13 09:16:12 +00:00
case "DateTime" :
2015-09-12 13:46:43 +00:00
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDateTime )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDateTime )
2016-04-02 17:57:47 +00:00
case "Time" :
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testTime )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testTime )
2013-08-13 09:16:12 +00:00
}
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( vu == value , true ) , value , vu )
2013-08-13 09:16:12 +00:00
}
}
func TestNullDataTypes ( t * testing . T ) {
d := DataNull { }
if IsPostgres {
// can removed when this fixed
// https://github.com/lib/pq/pull/125
d . DateTime = time . Now ( )
}
id , err := dORM . Insert ( & d )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 1 ) )
2013-08-13 09:16:12 +00:00
2016-04-08 14:26:28 +00:00
data := ` { "ok":1,"data": { "arr":[1,2],"msg":"gopher"}} `
2016-04-09 02:22:40 +00:00
d = DataNull { ID : 1 , JSON : data }
2016-04-08 14:26:28 +00:00
num , err := dORM . Update ( & d )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
2015-09-12 13:46:43 +00:00
d = DataNull { ID : 1 }
2013-08-13 09:16:12 +00:00
err = dORM . Read ( & d )
throwFail ( t , err )
2016-04-09 02:22:40 +00:00
throwFail ( t , AssertIs ( d . JSON , data ) )
2016-04-08 14:26:28 +00:00
2014-02-28 03:53:35 +00:00
throwFail ( t , AssertIs ( d . NullBool . Valid , false ) )
throwFail ( t , AssertIs ( d . NullString . Valid , false ) )
throwFail ( t , AssertIs ( d . NullInt64 . Valid , false ) )
throwFail ( t , AssertIs ( d . NullFloat64 . Valid , false ) )
2014-06-22 05:15:03 +00:00
throwFail ( t , AssertIs ( d . BooleanPtr , nil ) )
throwFail ( t , AssertIs ( d . CharPtr , nil ) )
throwFail ( t , AssertIs ( d . TextPtr , nil ) )
throwFail ( t , AssertIs ( d . BytePtr , nil ) )
throwFail ( t , AssertIs ( d . RunePtr , nil ) )
throwFail ( t , AssertIs ( d . IntPtr , nil ) )
throwFail ( t , AssertIs ( d . Int8Ptr , nil ) )
throwFail ( t , AssertIs ( d . Int16Ptr , nil ) )
throwFail ( t , AssertIs ( d . Int32Ptr , nil ) )
throwFail ( t , AssertIs ( d . Int64Ptr , nil ) )
throwFail ( t , AssertIs ( d . UintPtr , nil ) )
throwFail ( t , AssertIs ( d . Uint8Ptr , nil ) )
throwFail ( t , AssertIs ( d . Uint16Ptr , nil ) )
throwFail ( t , AssertIs ( d . Uint32Ptr , nil ) )
throwFail ( t , AssertIs ( d . Uint64Ptr , nil ) )
throwFail ( t , AssertIs ( d . Float32Ptr , nil ) )
throwFail ( t , AssertIs ( d . Float64Ptr , nil ) )
throwFail ( t , AssertIs ( d . DecimalPtr , nil ) )
2016-06-22 13:32:37 +00:00
throwFail ( t , AssertIs ( d . TimePtr , nil ) )
throwFail ( t , AssertIs ( d . DatePtr , nil ) )
throwFail ( t , AssertIs ( d . DateTimePtr , nil ) )
2014-06-22 05:15:03 +00:00
2013-08-13 09:16:12 +00:00
_ , err = dORM . Raw ( ` INSERT INTO data_null (boolean) VALUES (?) ` , nil ) . Exec ( )
throwFail ( t , err )
2015-09-12 13:46:43 +00:00
d = DataNull { ID : 2 }
2013-08-13 09:16:12 +00:00
err = dORM . Read ( & d )
throwFail ( t , err )
2014-02-28 03:53:35 +00:00
2014-06-22 05:15:03 +00:00
booleanPtr := true
charPtr := string ( "test" )
textPtr := string ( "test" )
bytePtr := byte ( 't' )
runePtr := rune ( 't' )
intPtr := int ( 42 )
int8Ptr := int8 ( 42 )
int16Ptr := int16 ( 42 )
int32Ptr := int32 ( 42 )
int64Ptr := int64 ( 42 )
uintPtr := uint ( 42 )
uint8Ptr := uint8 ( 42 )
uint16Ptr := uint16 ( 42 )
uint32Ptr := uint32 ( 42 )
uint64Ptr := uint64 ( 42 )
float32Ptr := float32 ( 42.0 )
float64Ptr := float64 ( 42.0 )
decimalPtr := float64 ( 42.0 )
2016-06-22 13:32:37 +00:00
timePtr := time . Now ( )
datePtr := time . Now ( )
dateTimePtr := time . Now ( )
2014-06-22 05:15:03 +00:00
2014-02-28 03:53:35 +00:00
d = DataNull {
DateTime : time . Now ( ) ,
2014-04-11 08:08:43 +00:00
NullString : sql . NullString { String : "test" , Valid : true } ,
NullBool : sql . NullBool { Bool : true , Valid : true } ,
NullInt64 : sql . NullInt64 { Int64 : 42 , Valid : true } ,
NullFloat64 : sql . NullFloat64 { Float64 : 42.42 , Valid : true } ,
2014-06-22 05:15:03 +00:00
BooleanPtr : & booleanPtr ,
CharPtr : & charPtr ,
TextPtr : & textPtr ,
BytePtr : & bytePtr ,
RunePtr : & runePtr ,
IntPtr : & intPtr ,
Int8Ptr : & int8Ptr ,
Int16Ptr : & int16Ptr ,
Int32Ptr : & int32Ptr ,
Int64Ptr : & int64Ptr ,
UintPtr : & uintPtr ,
Uint8Ptr : & uint8Ptr ,
Uint16Ptr : & uint16Ptr ,
Uint32Ptr : & uint32Ptr ,
Uint64Ptr : & uint64Ptr ,
Float32Ptr : & float32Ptr ,
Float64Ptr : & float64Ptr ,
DecimalPtr : & decimalPtr ,
2016-06-22 13:32:37 +00:00
TimePtr : & timePtr ,
DatePtr : & datePtr ,
DateTimePtr : & dateTimePtr ,
2014-02-28 03:53:35 +00:00
}
id , err = dORM . Insert ( & d )
throwFail ( t , err )
throwFail ( t , AssertIs ( id , 3 ) )
2015-09-12 13:46:43 +00:00
d = DataNull { ID : 3 }
2014-02-28 03:53:35 +00:00
err = dORM . Read ( & d )
throwFail ( t , err )
throwFail ( t , AssertIs ( d . NullBool . Valid , true ) )
throwFail ( t , AssertIs ( d . NullBool . Bool , true ) )
throwFail ( t , AssertIs ( d . NullString . Valid , true ) )
throwFail ( t , AssertIs ( d . NullString . String , "test" ) )
throwFail ( t , AssertIs ( d . NullInt64 . Valid , true ) )
throwFail ( t , AssertIs ( d . NullInt64 . Int64 , 42 ) )
throwFail ( t , AssertIs ( d . NullFloat64 . Valid , true ) )
throwFail ( t , AssertIs ( d . NullFloat64 . Float64 , 42.42 ) )
2014-06-22 05:15:03 +00:00
throwFail ( t , AssertIs ( * d . BooleanPtr , booleanPtr ) )
throwFail ( t , AssertIs ( * d . CharPtr , charPtr ) )
throwFail ( t , AssertIs ( * d . TextPtr , textPtr ) )
throwFail ( t , AssertIs ( * d . BytePtr , bytePtr ) )
throwFail ( t , AssertIs ( * d . RunePtr , runePtr ) )
throwFail ( t , AssertIs ( * d . IntPtr , intPtr ) )
throwFail ( t , AssertIs ( * d . Int8Ptr , int8Ptr ) )
throwFail ( t , AssertIs ( * d . Int16Ptr , int16Ptr ) )
throwFail ( t , AssertIs ( * d . Int32Ptr , int32Ptr ) )
throwFail ( t , AssertIs ( * d . Int64Ptr , int64Ptr ) )
throwFail ( t , AssertIs ( * d . UintPtr , uintPtr ) )
throwFail ( t , AssertIs ( * d . Uint8Ptr , uint8Ptr ) )
throwFail ( t , AssertIs ( * d . Uint16Ptr , uint16Ptr ) )
throwFail ( t , AssertIs ( * d . Uint32Ptr , uint32Ptr ) )
throwFail ( t , AssertIs ( * d . Uint64Ptr , uint64Ptr ) )
throwFail ( t , AssertIs ( * d . Float32Ptr , float32Ptr ) )
throwFail ( t , AssertIs ( * d . Float64Ptr , float64Ptr ) )
throwFail ( t , AssertIs ( * d . DecimalPtr , decimalPtr ) )
2018-05-04 08:38:03 +00:00
throwFail ( t , AssertIs ( ( * d . TimePtr ) . UTC ( ) . Format ( testTime ) , timePtr . UTC ( ) . Format ( testTime ) ) )
throwFail ( t , AssertIs ( ( * d . DatePtr ) . UTC ( ) . Format ( testDate ) , datePtr . UTC ( ) . Format ( testDate ) ) )
throwFail ( t , AssertIs ( ( * d . DateTimePtr ) . UTC ( ) . Format ( testDateTime ) , dateTimePtr . UTC ( ) . Format ( testDateTime ) ) )
2013-08-13 09:16:12 +00:00
}
2014-03-13 15:31:47 +00:00
func TestDataCustomTypes ( t * testing . T ) {
d := DataCustom { }
ind := reflect . Indirect ( reflect . ValueOf ( & d ) )
2015-09-12 13:46:43 +00:00
for name , value := range DataValues {
2014-03-13 15:31:47 +00:00
e := ind . FieldByName ( name )
if ! e . IsValid ( ) {
continue
}
e . Set ( reflect . ValueOf ( value ) . Convert ( e . Type ( ) ) )
}
id , err := dORM . Insert ( & d )
throwFail ( t , err )
throwFail ( t , AssertIs ( id , 1 ) )
2015-09-12 13:46:43 +00:00
d = DataCustom { ID : 1 }
2014-03-13 15:31:47 +00:00
err = dORM . Read ( & d )
throwFail ( t , err )
ind = reflect . Indirect ( reflect . ValueOf ( & d ) )
2015-09-12 13:46:43 +00:00
for name , value := range DataValues {
2014-03-13 15:31:47 +00:00
e := ind . FieldByName ( name )
if ! e . IsValid ( ) {
continue
}
vu := e . Interface ( )
value = reflect . ValueOf ( value ) . Convert ( e . Type ( ) ) . Interface ( )
throwFail ( t , AssertIs ( vu == value , true ) , value , vu )
}
}
2013-08-07 11:11:44 +00:00
func TestCRUD ( t * testing . T ) {
profile := NewProfile ( )
profile . Age = 30
profile . Money = 1234.12
id , err := dORM . Insert ( profile )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 1 ) )
2013-08-07 11:11:44 +00:00
user := NewUser ( )
user . UserName = "slene"
user . Email = "vslene@gmail.com"
user . Password = "pass"
user . Status = 3
user . IsStaff = true
user . IsActive = true
id , err = dORM . Insert ( user )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 1 ) )
2013-08-07 11:11:44 +00:00
2015-09-12 13:46:43 +00:00
u := & User { ID : user . ID }
2013-08-07 11:11:44 +00:00
err = dORM . Read ( u )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-08-07 11:11:44 +00:00
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( u . UserName , "slene" ) )
throwFail ( t , AssertIs ( u . Email , "vslene@gmail.com" ) )
throwFail ( t , AssertIs ( u . Password , "pass" ) )
throwFail ( t , AssertIs ( u . Status , 3 ) )
throwFail ( t , AssertIs ( u . IsStaff , true ) )
throwFail ( t , AssertIs ( u . IsActive , true ) )
2015-09-12 13:46:43 +00:00
throwFail ( t , AssertIs ( u . Created . In ( DefaultTimeLoc ) , user . Created . In ( DefaultTimeLoc ) , testDate ) )
throwFail ( t , AssertIs ( u . Updated . In ( DefaultTimeLoc ) , user . Updated . In ( DefaultTimeLoc ) , testDateTime ) )
2013-08-07 11:11:44 +00:00
user . UserName = "astaxie"
user . Profile = profile
num , err := dORM . Update ( user )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
2015-09-12 13:46:43 +00:00
u = & User { ID : user . ID }
2013-08-07 11:11:44 +00:00
err = dORM . Read ( u )
2013-09-12 11:04:39 +00:00
throwFailNow ( t , err )
throwFail ( t , AssertIs ( u . UserName , "astaxie" ) )
2015-09-12 13:46:43 +00:00
throwFail ( t , AssertIs ( u . Profile . ID , profile . ID ) )
2013-09-12 11:04:39 +00:00
u = & User { UserName : "astaxie" , Password : "pass" }
err = dORM . Read ( u , "UserName" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( id , 1 ) )
u . UserName = "QQ"
u . Password = "111"
num , err = dORM . Update ( u , "UserName" )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
2015-09-12 13:46:43 +00:00
u = & User { ID : user . ID }
2013-09-12 11:04:39 +00:00
err = dORM . Read ( u )
throwFailNow ( t , err )
throwFail ( t , AssertIs ( u . UserName , "QQ" ) )
throwFail ( t , AssertIs ( u . Password , "pass" ) )
2013-08-07 11:11:44 +00:00
num , err = dORM . Delete ( profile )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
2015-09-12 13:46:43 +00:00
u = & User { ID : user . ID }
2013-08-07 11:11:44 +00:00
err = dORM . Read ( u )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( true , u . Profile == nil ) )
2013-08-07 11:11:44 +00:00
num , err = dORM . Delete ( user )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
2015-09-12 13:46:43 +00:00
u = & User { ID : 100 }
2013-08-07 11:11:44 +00:00
err = dORM . Read ( u )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( err , ErrNoRows ) )
2013-08-30 04:32:05 +00:00
ub := UserBig { }
ub . Name = "name"
id , err = dORM . Insert ( & ub )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 1 ) )
2013-08-30 04:32:05 +00:00
2015-09-12 13:46:43 +00:00
ub = UserBig { ID : 1 }
2013-08-30 04:32:05 +00:00
err = dORM . Read ( & ub )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( ub . Name , "name" ) )
2016-09-12 13:13:21 +00:00
num , err = dORM . Delete ( & ub , "name" )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
}
func TestInsertTestData ( t * testing . T ) {
var users [ ] * User
profile := NewProfile ( )
profile . Age = 28
profile . Money = 1234.12
id , err := dORM . Insert ( profile )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 2 ) )
2013-08-07 11:11:44 +00:00
user := NewUser ( )
user . UserName = "slene"
user . Email = "vslene@gmail.com"
user . Password = "pass"
user . Status = 1
user . IsStaff = false
user . IsActive = true
user . Profile = profile
users = append ( users , user )
id , err = dORM . Insert ( user )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 2 ) )
2013-08-07 11:11:44 +00:00
profile = NewProfile ( )
profile . Age = 30
profile . Money = 4321.09
id , err = dORM . Insert ( profile )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 3 ) )
2013-08-07 11:11:44 +00:00
user = NewUser ( )
user . UserName = "astaxie"
user . Email = "astaxie@gmail.com"
user . Password = "password"
user . Status = 2
user . IsStaff = true
user . IsActive = false
user . Profile = profile
users = append ( users , user )
id , err = dORM . Insert ( user )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 3 ) )
2013-08-07 11:11:44 +00:00
user = NewUser ( )
user . UserName = "nobody"
user . Email = "nobody@gmail.com"
user . Password = "nobody"
user . Status = 3
user . IsStaff = false
user . IsActive = false
users = append ( users , user )
id , err = dORM . Insert ( user )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 4 ) )
2013-08-07 11:11:44 +00:00
tags := [ ] * Tag {
2015-09-12 13:46:43 +00:00
{ Name : "golang" , BestPost : & Post { ID : 2 } } ,
2015-03-20 05:29:01 +00:00
{ Name : "example" } ,
{ Name : "format" } ,
{ Name : "c++" } ,
2013-08-07 11:11:44 +00:00
}
posts := [ ] * Post {
2015-03-20 05:29:01 +00:00
{ User : users [ 0 ] , Tags : [ ] * Tag { tags [ 0 ] } , Title : "Introduction" , Content : ` Go is a new language . Although it borrows ideas from existing languages , it has unusual properties that make effective Go programs different in character from programs written in its relatives . A straightforward translation of a C ++ or Java program into Go is unlikely to produce a satisfactory result — Java programs are written in Java , not Go . On the other hand , thinking about the problem from a Go perspective could produce a successful but quite different program . In other words , to write Go well , it ' s important to understand its properties and idioms . It ' s also important to know the established conventions for programming in Go , such as naming , formatting , program construction , and so on , so that programs you write will be easy for other Go programmers to understand .
2013-08-07 11:11:44 +00:00
This document gives tips for writing clear , idiomatic Go code . It augments the language specification , the Tour of Go , and How to Write Go Code , all of which you should read first . ` } ,
2015-03-20 05:29:01 +00:00
{ User : users [ 1 ] , Tags : [ ] * Tag { tags [ 0 ] , tags [ 1 ] } , Title : "Examples" , Content : ` The Go package sources are intended to serve not only as the core library but also as examples of how to use the language. Moreover, many of the packages contain working, self-contained executable examples you can run directly from the golang.org web site, such as this one (click on the word "Example" to open it up). If you have a question about how to approach a problem or how something might be implemented, the documentation, code and examples in the library can provide answers, ideas and background. ` } ,
{ User : users [ 1 ] , Tags : [ ] * Tag { tags [ 0 ] , tags [ 2 ] } , Title : "Formatting" , Content : ` Formatting issues are the most contentious but the least consequential . People can adapt to different formatting styles but it ' s better if they don ' t have to , and less time is devoted to the topic if everyone adheres to the same style . The problem is how to approach this Utopia without a long prescriptive style guide .
2013-08-07 11:11:44 +00:00
With Go we take an unusual approach and let the machine take care of most formatting issues . The gofmt program ( also available as go fmt , which operates at the package level rather than source file level ) reads a Go program and emits the source in a standard style of indentation and vertical alignment , retaining and if necessary reformatting comments . If you want to know how to handle some new layout situation , run gofmt ; if the answer doesn ' t seem right , rearrange your program ( or file a bug about gofmt ) , don ' t work around it . ` } ,
2015-03-20 05:29:01 +00:00
{ User : users [ 2 ] , Tags : [ ] * Tag { tags [ 3 ] } , Title : "Commentary" , Content : ` Go provides C - style /* */ block comments and C ++ - style // line comments. Line comments are the norm; block comments appear mostly as package comments, but are useful within an expression or to disable large swaths of code.
2013-08-07 11:11:44 +00:00
The program — and web server — godoc processes Go source files to extract documentation about the contents of the package . Comments that appear before top - level declarations , with no intervening newlines , are extracted along with the declaration to serve as explanatory text for the item . The nature and style of these comments determines the quality of the documentation godoc produces . ` } ,
}
comments := [ ] * Comment {
2015-03-20 05:29:01 +00:00
{ Post : posts [ 0 ] , Content : "a comment" } ,
{ Post : posts [ 1 ] , Content : "yes" } ,
{ Post : posts [ 1 ] } ,
{ Post : posts [ 1 ] } ,
{ Post : posts [ 2 ] } ,
{ Post : posts [ 2 ] } ,
2013-08-07 11:11:44 +00:00
}
for _ , tag := range tags {
id , err := dORM . Insert ( tag )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id > 0 , true ) )
2013-08-07 11:11:44 +00:00
}
for _ , post := range posts {
id , err := dORM . Insert ( post )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id > 0 , true ) )
2013-10-14 14:31:35 +00:00
num := len ( post . Tags )
if num > 0 {
nums , err := dORM . QueryM2M ( post , "tags" ) . Add ( post . Tags )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( nums , num ) )
}
2013-08-07 11:11:44 +00:00
}
for _ , comment := range comments {
id , err := dORM . Insert ( comment )
2013-08-10 16:15:26 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id > 0 , true ) )
2013-08-07 11:11:44 +00:00
}
2015-12-15 16:17:56 +00:00
permissions := [ ] * Permission {
{ Name : "writePosts" } ,
{ Name : "readComments" } ,
{ Name : "readPosts" } ,
}
groups := [ ] * Group {
{
Name : "admins" ,
Permissions : [ ] * Permission { permissions [ 0 ] , permissions [ 1 ] , permissions [ 2 ] } ,
} ,
{
Name : "users" ,
Permissions : [ ] * Permission { permissions [ 1 ] , permissions [ 2 ] } ,
} ,
}
for _ , permission := range permissions {
id , err := dORM . Insert ( permission )
throwFail ( t , err )
throwFail ( t , AssertIs ( id > 0 , true ) )
}
for _ , group := range groups {
2016-01-15 00:43:02 +00:00
_ , err := dORM . Insert ( group )
2015-12-15 16:17:56 +00:00
throwFail ( t , err )
2016-01-15 06:36:45 +00:00
throwFail ( t , AssertIs ( id > 0 , true ) )
2015-12-15 16:17:56 +00:00
num := len ( group . Permissions )
if num > 0 {
nums , err := dORM . QueryM2M ( group , "permissions" ) . Add ( group . Permissions )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( nums , num ) )
}
}
2013-08-07 11:11:44 +00:00
}
2013-11-26 10:32:12 +00:00
func TestCustomField ( t * testing . T ) {
2015-09-12 13:46:43 +00:00
user := User { ID : 2 }
2013-11-26 10:32:12 +00:00
err := dORM . Read ( & user )
throwFailNow ( t , err )
user . Langs = append ( user . Langs , "zh-CN" , "en-US" )
2013-11-26 13:34:40 +00:00
user . Extra . Name = "beego"
user . Extra . Data = "orm"
_ , err = dORM . Update ( & user , "Langs" , "Extra" )
2013-11-26 10:32:12 +00:00
throwFailNow ( t , err )
2015-09-12 13:46:43 +00:00
user = User { ID : 2 }
2013-11-26 10:32:12 +00:00
err = dORM . Read ( & user )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( len ( user . Langs ) , 2 ) )
throwFailNow ( t , AssertIs ( user . Langs [ 0 ] , "zh-CN" ) )
throwFailNow ( t , AssertIs ( user . Langs [ 1 ] , "en-US" ) )
2013-11-26 13:34:40 +00:00
throwFailNow ( t , AssertIs ( user . Extra . Name , "beego" ) )
throwFailNow ( t , AssertIs ( user . Extra . Data , "orm" ) )
2013-11-26 10:32:12 +00:00
}
2013-08-07 11:11:44 +00:00
func TestExpr ( t * testing . T ) {
2013-08-13 09:16:12 +00:00
user := & User { }
qs := dORM . QueryTable ( user )
2013-11-24 06:26:32 +00:00
qs = dORM . QueryTable ( ( * User ) ( nil ) )
2013-08-13 09:16:12 +00:00
qs = dORM . QueryTable ( "User" )
2013-08-07 11:11:44 +00:00
qs = dORM . QueryTable ( "user" )
num , err := qs . Filter ( "UserName" , "slene" ) . Filter ( "user_name" , "slene" ) . Filter ( "profile__Age" , 28 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-11 14:27:45 +00:00
num , err = qs . Filter ( "created" , time . Now ( ) ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-13 09:16:12 +00:00
2013-11-06 13:08:12 +00:00
// num, err = qs.Filter("created", time.Now().Format(format_Date)).Count()
// throwFail(t, err)
// throwFail(t, AssertIs(num, 3))
2013-08-07 11:11:44 +00:00
}
func TestOperators ( t * testing . T ) {
qs := dORM . QueryTable ( "user" )
num , err := qs . Filter ( "user_name" , "slene" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
2014-03-13 15:31:47 +00:00
num , err = qs . Filter ( "user_name__exact" , String ( "slene" ) ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__exact" , "slene" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__iexact" , "Slene" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__contains" , "e" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
2013-08-10 16:15:26 +00:00
var shouldNum int
2015-09-10 08:31:53 +00:00
if IsSqlite || IsTidb {
2013-08-10 16:15:26 +00:00
shouldNum = 2
} else {
shouldNum = 0
}
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__contains" , "E" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , shouldNum ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__icontains" , "E" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__icontains" , "E" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "status__gt" , 1 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "status__gte" , 1 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-07 11:11:44 +00:00
2014-03-13 15:31:47 +00:00
num , err = qs . Filter ( "status__lt" , Uint ( 3 ) ) . Count ( )
2013-08-07 11:11:44 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
2014-03-13 15:31:47 +00:00
num , err = qs . Filter ( "status__lte" , Int ( 3 ) ) . Count ( )
2013-08-07 11:11:44 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__startswith" , "s" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
2015-09-10 08:31:53 +00:00
if IsSqlite || IsTidb {
2013-08-10 16:15:26 +00:00
shouldNum = 1
} else {
shouldNum = 0
}
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__startswith" , "S" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , shouldNum ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__istartswith" , "S" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__endswith" , "e" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
2015-09-10 08:31:53 +00:00
if IsSqlite || IsTidb {
2013-08-10 16:15:26 +00:00
shouldNum = 2
} else {
shouldNum = 0
}
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__endswith" , "E" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , shouldNum ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__iendswith" , "E" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "profile__isnull" , true ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "status__in" , 1 , 2 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-08 14:34:18 +00:00
num , err = qs . Filter ( "status__in" , [ ] int { 1 , 2 } ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-08 14:34:18 +00:00
n1 , n2 := 1 , 2
num , err = qs . Filter ( "status__in" , [ ] * int { & n1 } , & n2 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2014-03-10 12:18:57 +00:00
num , err = qs . Filter ( "id__between" , 2 , 3 ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 2 ) )
num , err = qs . Filter ( "id__between" , [ ] int { 2 , 3 } ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
}
2013-10-14 14:31:35 +00:00
func TestSetCond ( t * testing . T ) {
cond := NewCondition ( )
cond1 := cond . And ( "profile__isnull" , false ) . AndNot ( "status__in" , 1 ) . Or ( "profile__age__gt" , 2000 )
qs := dORM . QueryTable ( "user" )
num , err := qs . SetCond ( cond1 ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
cond2 := cond . AndCond ( cond1 ) . OrCond ( cond . And ( "user_name" , "slene" ) )
num , err = qs . SetCond ( cond2 ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 2 ) )
2016-11-28 08:49:06 +00:00
cond3 := cond . AndNotCond ( cond . And ( "status__in" , 1 ) )
num , err = qs . SetCond ( cond3 ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 2 ) )
cond4 := cond . And ( "user_name" , "slene" ) . OrNotCond ( cond . And ( "user_name" , "slene" ) )
num , err = qs . SetCond ( cond4 ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 3 ) )
2013-10-14 14:31:35 +00:00
}
func TestLimit ( t * testing . T ) {
var posts [ ] * Post
qs := dORM . QueryTable ( "post" )
num , err := qs . Limit ( 1 ) . All ( & posts )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = qs . Limit ( - 1 ) . All ( & posts )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 4 ) )
num , err = qs . Limit ( - 1 , 2 ) . All ( & posts )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 2 ) )
num , err = qs . Limit ( 0 , 2 ) . All ( & posts )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 2 ) )
}
func TestOffset ( t * testing . T ) {
var posts [ ] * Post
qs := dORM . QueryTable ( "post" )
num , err := qs . Limit ( 1 ) . Offset ( 2 ) . All ( & posts )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = qs . Offset ( 2 ) . All ( & posts )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 2 ) )
}
func TestOrderBy ( t * testing . T ) {
qs := dORM . QueryTable ( "user" )
num , err := qs . OrderBy ( "-status" ) . Filter ( "user_name" , "nobody" ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = qs . OrderBy ( "status" ) . Filter ( "user_name" , "slene" ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = qs . OrderBy ( "-profile__age" ) . Filter ( "user_name" , "astaxie" ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
}
2013-08-07 11:11:44 +00:00
func TestAll ( t * testing . T ) {
var users [ ] * User
qs := dORM . QueryTable ( "user" )
2013-09-09 14:33:41 +00:00
num , err := qs . OrderBy ( "Id" ) . All ( & users )
2013-08-07 11:11:44 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFailNow ( t , AssertIs ( num , 3 ) )
2013-09-09 14:33:41 +00:00
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( users [ 0 ] . UserName , "slene" ) )
throwFail ( t , AssertIs ( users [ 1 ] . UserName , "astaxie" ) )
throwFail ( t , AssertIs ( users [ 2 ] . UserName , "nobody" ) )
2013-09-09 14:33:41 +00:00
var users2 [ ] User
qs = dORM . QueryTable ( "user" )
num , err = qs . OrderBy ( "Id" ) . All ( & users2 )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFailNow ( t , AssertIs ( num , 3 ) )
throwFailNow ( t , AssertIs ( users2 [ 0 ] . UserName , "slene" ) )
throwFailNow ( t , AssertIs ( users2 [ 1 ] . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( users2 [ 2 ] . UserName , "nobody" ) )
2013-09-09 14:33:41 +00:00
2013-09-12 11:04:39 +00:00
qs = dORM . QueryTable ( "user" )
num , err = qs . OrderBy ( "Id" ) . RelatedSel ( ) . All ( & users2 , "UserName" )
throwFail ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
throwFailNow ( t , AssertIs ( len ( users2 ) , 3 ) )
throwFailNow ( t , AssertIs ( users2 [ 0 ] . UserName , "slene" ) )
throwFailNow ( t , AssertIs ( users2 [ 1 ] . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( users2 [ 2 ] . UserName , "nobody" ) )
2015-09-12 13:46:43 +00:00
throwFailNow ( t , AssertIs ( users2 [ 0 ] . ID , 0 ) )
throwFailNow ( t , AssertIs ( users2 [ 1 ] . ID , 0 ) )
throwFailNow ( t , AssertIs ( users2 [ 2 ] . ID , 0 ) )
2013-09-12 11:04:39 +00:00
throwFailNow ( t , AssertIs ( users2 [ 0 ] . Profile == nil , false ) )
throwFailNow ( t , AssertIs ( users2 [ 1 ] . Profile == nil , false ) )
throwFailNow ( t , AssertIs ( users2 [ 2 ] . Profile == nil , true ) )
2013-08-07 11:11:44 +00:00
qs = dORM . QueryTable ( "user" )
num , err = qs . Filter ( "user_name" , "nothing" ) . All ( & users )
2013-11-08 14:19:01 +00:00
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 0 ) )
var users3 [ ] * User
qs = dORM . QueryTable ( "user" )
num , err = qs . Filter ( "user_name" , "nothing" ) . All ( & users3 )
2017-04-28 13:38:08 +00:00
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 0 ) )
2013-11-08 14:19:01 +00:00
throwFailNow ( t , AssertIs ( users3 == nil , false ) )
2013-08-07 11:11:44 +00:00
}
func TestOne ( t * testing . T ) {
var user User
qs := dORM . QueryTable ( "user" )
err := qs . One ( & user )
2016-04-13 02:36:12 +00:00
throwFail ( t , err )
2013-08-07 11:11:44 +00:00
2013-09-09 14:33:41 +00:00
user = User { }
err = qs . OrderBy ( "Id" ) . Limit ( 1 ) . One ( & user )
throwFailNow ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( user . UserName , "slene" ) )
2016-04-25 07:51:12 +00:00
throwFail ( t , AssertNot ( err , ErrMultiRows ) )
2013-09-09 14:33:41 +00:00
2016-04-13 02:36:12 +00:00
user = User { }
err = qs . OrderBy ( "-Id" ) . Limit ( 100 ) . One ( & user )
throwFailNow ( t , err )
throwFail ( t , AssertIs ( user . UserName , "nobody" ) )
2016-04-25 07:51:12 +00:00
throwFail ( t , AssertNot ( err , ErrMultiRows ) )
2013-09-09 14:33:41 +00:00
2013-08-07 11:11:44 +00:00
err = qs . Filter ( "user_name" , "nothing" ) . One ( & user )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( err , ErrNoRows ) )
2013-09-09 14:33:41 +00:00
2013-08-07 11:11:44 +00:00
}
func TestValues ( t * testing . T ) {
var maps [ ] Params
qs := dORM . QueryTable ( "user" )
2013-11-26 10:32:12 +00:00
num , err := qs . OrderBy ( "Id" ) . Values ( & maps )
2013-08-07 11:11:44 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-07 11:11:44 +00:00
if num == 3 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( maps [ 0 ] [ "UserName" ] , "slene" ) )
throwFail ( t , AssertIs ( maps [ 2 ] [ "Profile" ] , nil ) )
2013-08-07 11:11:44 +00:00
}
2013-11-26 10:32:12 +00:00
num , err = qs . OrderBy ( "Id" ) . Values ( & maps , "UserName" , "Profile__Age" )
2013-08-07 11:11:44 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-07 11:11:44 +00:00
if num == 3 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( maps [ 0 ] [ "UserName" ] , "slene" ) )
throwFail ( t , AssertIs ( maps [ 0 ] [ "Profile__Age" ] , 28 ) )
throwFail ( t , AssertIs ( maps [ 2 ] [ "Profile__Age" ] , nil ) )
2013-08-07 11:11:44 +00:00
}
2013-12-17 08:46:14 +00:00
num , err = qs . Filter ( "UserName" , "slene" ) . Values ( & maps )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
}
func TestValuesList ( t * testing . T ) {
var list [ ] ParamsList
qs := dORM . QueryTable ( "user" )
2013-11-26 10:32:12 +00:00
num , err := qs . OrderBy ( "Id" ) . ValuesList ( & list )
2013-08-07 11:11:44 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-07 11:11:44 +00:00
if num == 3 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( list [ 0 ] [ 1 ] , "slene" ) )
throwFail ( t , AssertIs ( list [ 2 ] [ 9 ] , nil ) )
2013-08-07 11:11:44 +00:00
}
2013-11-26 10:32:12 +00:00
num , err = qs . OrderBy ( "Id" ) . ValuesList ( & list , "UserName" , "Profile__Age" )
2013-08-07 11:11:44 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-07 11:11:44 +00:00
if num == 3 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( list [ 0 ] [ 0 ] , "slene" ) )
throwFail ( t , AssertIs ( list [ 0 ] [ 1 ] , 28 ) )
throwFail ( t , AssertIs ( list [ 2 ] [ 1 ] , nil ) )
2013-08-07 11:11:44 +00:00
}
}
func TestValuesFlat ( t * testing . T ) {
var list ParamsList
qs := dORM . QueryTable ( "user" )
num , err := qs . OrderBy ( "id" ) . ValuesFlat ( & list , "UserName" )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-07 11:11:44 +00:00
if num == 3 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( list [ 0 ] , "slene" ) )
throwFail ( t , AssertIs ( list [ 1 ] , "astaxie" ) )
throwFail ( t , AssertIs ( list [ 2 ] , "nobody" ) )
2013-08-07 11:11:44 +00:00
}
}
func TestRelatedSel ( t * testing . T ) {
2015-09-10 08:31:53 +00:00
if IsTidb {
// Skip it. TiDB does not support relation now.
return
}
2013-08-07 11:11:44 +00:00
qs := dORM . QueryTable ( "user" )
num , err := qs . Filter ( "profile__age" , 28 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "profile__age__gt" , 28 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "profile__user__profile__age__gt" , 28 ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
var user User
err = qs . Filter ( "user_name" , "slene" ) . RelatedSel ( "profile" ) . One ( & user )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
throwFail ( t , AssertNot ( user . Profile , nil ) )
2013-08-07 11:11:44 +00:00
if user . Profile != nil {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( user . Profile . Age , 28 ) )
2013-08-07 11:11:44 +00:00
}
err = qs . Filter ( "user_name" , "slene" ) . RelatedSel ( ) . One ( & user )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
throwFail ( t , AssertNot ( user . Profile , nil ) )
2013-08-07 11:11:44 +00:00
if user . Profile != nil {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( user . Profile . Age , 28 ) )
2013-08-07 11:11:44 +00:00
}
err = qs . Filter ( "user_name" , "nobody" ) . RelatedSel ( "profile" ) . One ( & user )
2017-04-28 14:36:28 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
throwFail ( t , AssertIs ( user . Profile , nil ) )
2013-08-07 11:11:44 +00:00
qs = dORM . QueryTable ( "user_profile" )
num , err = qs . Filter ( "user__username" , "slene" ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-09-22 11:20:40 +00:00
var posts [ ] * Post
qs = dORM . QueryTable ( "post" )
num , err = qs . RelatedSel ( ) . All ( & posts )
throwFail ( t , err )
throwFailNow ( t , AssertIs ( num , 4 ) )
throwFailNow ( t , AssertIs ( posts [ 0 ] . User . UserName , "slene" ) )
throwFailNow ( t , AssertIs ( posts [ 1 ] . User . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( posts [ 2 ] . User . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( posts [ 3 ] . User . UserName , "nobody" ) )
2013-08-07 11:11:44 +00:00
}
2013-10-14 14:31:35 +00:00
func TestReverseQuery ( t * testing . T ) {
var profile Profile
err := dORM . QueryTable ( "user_profile" ) . Filter ( "User" , 3 ) . One ( & profile )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( profile . Age , 30 ) )
2013-08-07 11:11:44 +00:00
2013-10-14 14:31:35 +00:00
profile = Profile { }
err = dORM . QueryTable ( "user_profile" ) . Filter ( "User__UserName" , "astaxie" ) . One ( & profile )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( profile . Age , 30 ) )
2013-08-07 11:11:44 +00:00
2013-10-14 14:31:35 +00:00
var user User
err = dORM . QueryTable ( "user" ) . Filter ( "Posts__Title" , "Examples" ) . One ( & user )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( user . UserName , "astaxie" ) )
user = User { }
err = dORM . QueryTable ( "user" ) . Filter ( "Posts__User__UserName" , "astaxie" ) . Limit ( 1 ) . One ( & user )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( user . UserName , "astaxie" ) )
user = User { }
err = dORM . QueryTable ( "user" ) . Filter ( "Posts__User__UserName" , "astaxie" ) . RelatedSel ( ) . Limit ( 1 ) . One ( & user )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( user . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( user . Profile == nil , false ) )
throwFailNow ( t , AssertIs ( user . Profile . Age , 30 ) )
2013-08-07 11:11:44 +00:00
var posts [ ] * Post
2013-10-14 14:31:35 +00:00
num , err := dORM . QueryTable ( "post" ) . Filter ( "Tags__Tag__Name" , "golang" ) . All ( & posts )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
throwFailNow ( t , AssertIs ( posts [ 0 ] . Title , "Introduction" ) )
2013-08-07 11:11:44 +00:00
2013-10-14 14:31:35 +00:00
posts = [ ] * Post { }
num , err = dORM . QueryTable ( "post" ) . Filter ( "Tags__Tag__Name" , "golang" ) . Filter ( "User__UserName" , "slene" ) . All ( & posts )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( posts [ 0 ] . Title , "Introduction" ) )
2013-08-07 11:11:44 +00:00
2013-10-14 14:31:35 +00:00
posts = [ ] * Post { }
num , err = dORM . QueryTable ( "post" ) . Filter ( "Tags__Tag__Name" , "golang" ) .
Filter ( "User__UserName" , "slene" ) . RelatedSel ( ) . All ( & posts )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( posts [ 0 ] . User == nil , false ) )
throwFailNow ( t , AssertIs ( posts [ 0 ] . User . UserName , "slene" ) )
2013-08-07 11:11:44 +00:00
2013-10-14 14:31:35 +00:00
var tags [ ] * Tag
num , err = dORM . QueryTable ( "tag" ) . Filter ( "Posts__Post__Title" , "Introduction" ) . All ( & tags )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( tags [ 0 ] . Name , "golang" ) )
tags = [ ] * Tag { }
num , err = dORM . QueryTable ( "tag" ) . Filter ( "Posts__Post__Title" , "Introduction" ) .
Filter ( "BestPost__User__UserName" , "astaxie" ) . All ( & tags )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( tags [ 0 ] . Name , "golang" ) )
tags = [ ] * Tag { }
num , err = dORM . QueryTable ( "tag" ) . Filter ( "Posts__Post__Title" , "Introduction" ) .
Filter ( "BestPost__User__UserName" , "astaxie" ) . RelatedSel ( ) . All ( & tags )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( tags [ 0 ] . Name , "golang" ) )
throwFailNow ( t , AssertIs ( tags [ 0 ] . BestPost == nil , false ) )
throwFailNow ( t , AssertIs ( tags [ 0 ] . BestPost . Title , "Examples" ) )
throwFailNow ( t , AssertIs ( tags [ 0 ] . BestPost . User == nil , false ) )
throwFailNow ( t , AssertIs ( tags [ 0 ] . BestPost . User . UserName , "astaxie" ) )
2013-08-07 11:11:44 +00:00
}
2013-10-14 14:31:35 +00:00
func TestLoadRelated ( t * testing . T ) {
// load reverse foreign key
2015-09-12 13:46:43 +00:00
user := User { ID : 3 }
2013-08-07 11:11:44 +00:00
2013-10-14 14:31:35 +00:00
err := dORM . Read ( & user )
throwFailNow ( t , err )
num , err := dORM . LoadRelated ( & user , "Posts" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
throwFailNow ( t , AssertIs ( len ( user . Posts ) , 2 ) )
2015-09-12 13:46:43 +00:00
throwFailNow ( t , AssertIs ( user . Posts [ 0 ] . User . ID , 3 ) )
2013-10-14 14:31:35 +00:00
num , err = dORM . LoadRelated ( & user , "Posts" , true )
throwFailNow ( t , err )
2017-04-28 14:36:28 +00:00
throwFailNow ( t , AssertIs ( num , 2 ) )
2013-10-14 14:31:35 +00:00
throwFailNow ( t , AssertIs ( len ( user . Posts ) , 2 ) )
throwFailNow ( t , AssertIs ( user . Posts [ 0 ] . User . UserName , "astaxie" ) )
num , err = dORM . LoadRelated ( & user , "Posts" , true , 1 )
throwFailNow ( t , err )
2017-04-28 14:36:28 +00:00
throwFailNow ( t , AssertIs ( num , 1 ) )
2013-10-14 14:31:35 +00:00
throwFailNow ( t , AssertIs ( len ( user . Posts ) , 1 ) )
num , err = dORM . LoadRelated ( & user , "Posts" , true , 0 , 0 , "-Id" )
throwFailNow ( t , err )
2017-04-28 14:36:28 +00:00
throwFailNow ( t , AssertIs ( num , 2 ) )
2013-10-14 14:31:35 +00:00
throwFailNow ( t , AssertIs ( len ( user . Posts ) , 2 ) )
throwFailNow ( t , AssertIs ( user . Posts [ 0 ] . Title , "Formatting" ) )
num , err = dORM . LoadRelated ( & user , "Posts" , true , 1 , 1 , "Id" )
throwFailNow ( t , err )
2017-04-28 14:36:28 +00:00
throwFailNow ( t , AssertIs ( num , 1 ) )
2013-10-14 14:31:35 +00:00
throwFailNow ( t , AssertIs ( len ( user . Posts ) , 1 ) )
throwFailNow ( t , AssertIs ( user . Posts [ 0 ] . Title , "Formatting" ) )
// load reverse one to one
2015-09-12 13:46:43 +00:00
profile := Profile { ID : 3 }
profile . BestPost = & Post { ID : 2 }
2013-10-14 14:31:35 +00:00
num , err = dORM . Update ( & profile , "BestPost" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
err = dORM . Read ( & profile )
throwFailNow ( t , err )
num , err = dORM . LoadRelated ( & profile , "User" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( profile . User == nil , false ) )
throwFailNow ( t , AssertIs ( profile . User . UserName , "astaxie" ) )
num , err = dORM . LoadRelated ( & profile , "User" , true )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( profile . User == nil , false ) )
throwFailNow ( t , AssertIs ( profile . User . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( profile . User . Profile . Age , profile . Age ) )
// load rel one to one
err = dORM . Read ( & user )
throwFailNow ( t , err )
num , err = dORM . LoadRelated ( & user , "Profile" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( user . Profile == nil , false ) )
throwFailNow ( t , AssertIs ( user . Profile . Age , 30 ) )
num , err = dORM . LoadRelated ( & user , "Profile" , true )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( user . Profile == nil , false ) )
throwFailNow ( t , AssertIs ( user . Profile . Age , 30 ) )
throwFailNow ( t , AssertIs ( user . Profile . BestPost == nil , false ) )
throwFailNow ( t , AssertIs ( user . Profile . BestPost . Title , "Examples" ) )
2015-09-12 13:46:43 +00:00
post := Post { ID : 2 }
2013-10-14 14:31:35 +00:00
// load rel foreign key
err = dORM . Read ( & post )
throwFailNow ( t , err )
num , err = dORM . LoadRelated ( & post , "User" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( post . User == nil , false ) )
throwFailNow ( t , AssertIs ( post . User . UserName , "astaxie" ) )
num , err = dORM . LoadRelated ( & post , "User" , true )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( post . User == nil , false ) )
throwFailNow ( t , AssertIs ( post . User . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( post . User . Profile == nil , false ) )
throwFailNow ( t , AssertIs ( post . User . Profile . Age , 30 ) )
// load rel m2m
2015-09-12 13:46:43 +00:00
post = Post { ID : 2 }
2013-10-14 14:31:35 +00:00
err = dORM . Read ( & post )
throwFailNow ( t , err )
num , err = dORM . LoadRelated ( & post , "Tags" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
throwFailNow ( t , AssertIs ( len ( post . Tags ) , 2 ) )
throwFailNow ( t , AssertIs ( post . Tags [ 0 ] . Name , "golang" ) )
num , err = dORM . LoadRelated ( & post , "Tags" , true )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
throwFailNow ( t , AssertIs ( len ( post . Tags ) , 2 ) )
throwFailNow ( t , AssertIs ( post . Tags [ 0 ] . Name , "golang" ) )
throwFailNow ( t , AssertIs ( post . Tags [ 0 ] . BestPost == nil , false ) )
throwFailNow ( t , AssertIs ( post . Tags [ 0 ] . BestPost . User . UserName , "astaxie" ) )
// load reverse m2m
2015-09-12 13:46:43 +00:00
tag := Tag { ID : 1 }
2013-10-14 14:31:35 +00:00
err = dORM . Read ( & tag )
throwFailNow ( t , err )
num , err = dORM . LoadRelated ( & tag , "Posts" )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
throwFailNow ( t , AssertIs ( tag . Posts [ 0 ] . Title , "Introduction" ) )
2015-09-12 13:46:43 +00:00
throwFailNow ( t , AssertIs ( tag . Posts [ 0 ] . User . ID , 2 ) )
2013-10-14 14:31:35 +00:00
throwFailNow ( t , AssertIs ( tag . Posts [ 0 ] . User . Profile == nil , true ) )
num , err = dORM . LoadRelated ( & tag , "Posts" , true )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
throwFailNow ( t , AssertIs ( tag . Posts [ 0 ] . Title , "Introduction" ) )
2015-09-12 13:46:43 +00:00
throwFailNow ( t , AssertIs ( tag . Posts [ 0 ] . User . ID , 2 ) )
2013-10-14 14:31:35 +00:00
throwFailNow ( t , AssertIs ( tag . Posts [ 0 ] . User . UserName , "slene" ) )
2013-08-07 11:11:44 +00:00
}
2013-10-14 14:31:35 +00:00
func TestQueryM2M ( t * testing . T ) {
2015-09-12 13:46:43 +00:00
post := Post { ID : 4 }
2013-10-14 14:31:35 +00:00
m2m := dORM . QueryM2M ( & post , "Tags" )
2013-08-07 11:11:44 +00:00
2015-03-20 05:29:01 +00:00
tag1 := [ ] * Tag { { Name : "TestTag1" } , { Name : "TestTag2" } }
2013-10-14 14:31:35 +00:00
tag2 := & Tag { Name : "TestTag3" }
tag3 := [ ] interface { } { & Tag { Name : "TestTag4" } }
2013-08-07 11:11:44 +00:00
2013-10-14 14:31:35 +00:00
tags := [ ] interface { } { tag1 [ 0 ] , tag1 [ 1 ] , tag2 , tag3 [ 0 ] }
for _ , tag := range tags {
_ , err := dORM . Insert ( tag )
throwFailNow ( t , err )
}
num , err := m2m . Add ( tag1 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
num , err = m2m . Add ( tag2 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
num , err = m2m . Add ( tag3 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 5 ) )
num , err = m2m . Remove ( tag3 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 4 ) )
exist := m2m . Exist ( tag2 )
throwFailNow ( t , AssertIs ( exist , true ) )
num , err = m2m . Remove ( tag2 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
exist = m2m . Exist ( tag2 )
throwFailNow ( t , AssertIs ( exist , false ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
num , err = m2m . Clear ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 0 ) )
2013-11-06 13:08:12 +00:00
tag := Tag { Name : "test" }
_ , err = dORM . Insert ( & tag )
throwFailNow ( t , err )
m2m = dORM . QueryM2M ( & tag , "Posts" )
2015-03-20 05:29:01 +00:00
post1 := [ ] * Post { { Title : "TestPost1" } , { Title : "TestPost2" } }
2013-11-06 13:08:12 +00:00
post2 := & Post { Title : "TestPost3" }
post3 := [ ] interface { } { & Post { Title : "TestPost4" } }
posts := [ ] interface { } { post1 [ 0 ] , post1 [ 1 ] , post2 , post3 [ 0 ] }
for _ , post := range posts {
p := post . ( * Post )
2015-09-12 13:46:43 +00:00
p . User = & User { ID : 1 }
2013-11-06 13:08:12 +00:00
_ , err := dORM . Insert ( post )
throwFailNow ( t , err )
}
num , err = m2m . Add ( post1 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
num , err = m2m . Add ( post2 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
num , err = m2m . Add ( post3 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 4 ) )
num , err = m2m . Remove ( post3 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
exist = m2m . Exist ( post2 )
throwFailNow ( t , AssertIs ( exist , true ) )
num , err = m2m . Remove ( post2 )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
exist = m2m . Exist ( post2 )
throwFailNow ( t , AssertIs ( exist , false ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
num , err = m2m . Clear ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
num , err = m2m . Count ( )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 0 ) )
num , err = dORM . Delete ( & tag )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 1 ) )
2013-10-14 14:31:35 +00:00
}
func TestQueryRelate ( t * testing . T ) {
// post := &Post{Id: 2}
// qs := dORM.QueryRelate(post, "Tags")
// num, err := qs.Count()
// throwFailNow(t, err)
// throwFailNow(t, AssertIs(num, 2))
// var tags []*Tag
// num, err = qs.All(&tags)
// throwFailNow(t, err)
// throwFailNow(t, AssertIs(num, 2))
// throwFailNow(t, AssertIs(tags[0].Name, "golang"))
// num, err = dORM.QueryTable("Tag").Filter("Posts__Post", 2).Count()
// throwFailNow(t, err)
// throwFailNow(t, AssertIs(num, 2))
2013-08-07 11:11:44 +00:00
}
2015-12-15 16:17:56 +00:00
func TestPkManyRelated ( t * testing . T ) {
permission := & Permission { Name : "readPosts" }
err := dORM . Read ( permission , "Name" )
throwFailNow ( t , err )
var groups [ ] * Group
qs := dORM . QueryTable ( "Group" )
num , err := qs . Filter ( "Permissions__Permission" , permission . ID ) . All ( & groups )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
}
2013-08-07 11:11:44 +00:00
func TestPrepareInsert ( t * testing . T ) {
qs := dORM . QueryTable ( "user" )
i , err := qs . PrepareInsert ( )
2013-08-10 16:15:26 +00:00
throwFailNow ( t , err )
2013-08-07 11:11:44 +00:00
var user User
user . UserName = "testing1"
num , err := i . Insert ( & user )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num > 0 , true ) )
2013-08-07 11:11:44 +00:00
user . UserName = "testing2"
num , err = i . Insert ( & user )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num > 0 , true ) )
2013-08-07 11:11:44 +00:00
num , err = qs . Filter ( "user_name__in" , "testing1" , "testing2" ) . Delete ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 2 ) )
2013-08-07 11:11:44 +00:00
err = i . Close ( )
throwFail ( t , err )
err = i . Close ( )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( err , ErrStmtClosed ) )
2013-08-07 11:11:44 +00:00
}
2013-09-09 14:33:41 +00:00
func TestRawExec ( t * testing . T ) {
Q := dDbBaser . TableQuote ( )
query := fmt . Sprintf ( "UPDATE %suser%s SET %suser_name%s = ? WHERE %suser_name%s = ?" , Q , Q , Q , Q , Q , Q )
res , err := dORM . Raw ( query , "testing" , "slene" ) . Exec ( )
throwFail ( t , err )
num , err := res . RowsAffected ( )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) , err )
2013-08-30 04:32:05 +00:00
2013-09-09 14:33:41 +00:00
res , err = dORM . Raw ( query , "slene" , "testing" ) . Exec ( )
throwFail ( t , err )
num , err = res . RowsAffected ( )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) , err )
2013-08-30 04:32:05 +00:00
}
2013-09-09 14:33:41 +00:00
func TestRawQueryRow ( t * testing . T ) {
var (
Boolean bool
Char string
Text string
2016-04-02 17:57:47 +00:00
Time time . Time
2013-09-09 14:33:41 +00:00
Date time . Time
DateTime time . Time
Byte byte
Rune rune
Int int
Int8 int
Int16 int16
Int32 int32
Int64 int64
Uint uint
Uint8 uint8
Uint16 uint16
Uint32 uint32
Uint64 uint64
Float32 float32
Float64 float64
Decimal float64
)
2015-09-12 13:46:43 +00:00
dataValues := make ( map [ string ] interface { } , len ( DataValues ) )
2013-09-09 14:33:41 +00:00
2015-09-12 13:46:43 +00:00
for k , v := range DataValues {
dataValues [ strings . ToLower ( k ) ] = v
2013-09-09 14:33:41 +00:00
}
2013-08-30 04:32:05 +00:00
2013-09-09 14:33:41 +00:00
Q := dDbBaser . TableQuote ( )
2013-08-30 04:32:05 +00:00
2013-09-09 14:33:41 +00:00
cols := [ ] string {
2016-04-02 17:57:47 +00:00
"id" , "boolean" , "char" , "text" , "time" , "date" , "datetime" , "byte" , "rune" , "int" , "int8" , "int16" , "int32" ,
2013-09-09 14:33:41 +00:00
"int64" , "uint" , "uint8" , "uint16" , "uint32" , "uint64" , "float32" , "float64" , "decimal" ,
}
sep := fmt . Sprintf ( "%s, %s" , Q , Q )
query := fmt . Sprintf ( "SELECT %s%s%s FROM data WHERE id = ?" , Q , strings . Join ( cols , sep ) , Q )
var id int
values := [ ] interface { } {
2016-04-02 17:57:47 +00:00
& id , & Boolean , & Char , & Text , & Time , & Date , & DateTime , & Byte , & Rune , & Int , & Int8 , & Int16 , & Int32 ,
2013-09-09 14:33:41 +00:00
& Int64 , & Uint , & Uint8 , & Uint16 , & Uint32 , & Uint64 , & Float32 , & Float64 , & Decimal ,
}
err := dORM . Raw ( query , 1 ) . QueryRow ( values ... )
throwFailNow ( t , err )
for i , col := range cols {
vu := values [ i ]
v := reflect . ValueOf ( vu ) . Elem ( ) . Interface ( )
switch col {
case "id" :
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , 1 ) )
2016-04-02 17:57:47 +00:00
case "time" :
v = v . ( time . Time ) . In ( DefaultTimeLoc )
value := dataValues [ col ] . ( time . Time ) . In ( DefaultTimeLoc )
throwFail ( t , AssertIs ( v , value , testTime ) )
2013-09-09 14:33:41 +00:00
case "date" :
v = v . ( time . Time ) . In ( DefaultTimeLoc )
2015-09-12 13:46:43 +00:00
value := dataValues [ col ] . ( time . Time ) . In ( DefaultTimeLoc )
throwFail ( t , AssertIs ( v , value , testDate ) )
2013-09-09 14:33:41 +00:00
case "datetime" :
v = v . ( time . Time ) . In ( DefaultTimeLoc )
2015-09-12 13:46:43 +00:00
value := dataValues [ col ] . ( time . Time ) . In ( DefaultTimeLoc )
throwFail ( t , AssertIs ( v , value , testDateTime ) )
2013-09-09 14:33:41 +00:00
default :
2015-09-12 13:46:43 +00:00
throwFail ( t , AssertIs ( v , dataValues [ col ] ) )
2013-09-09 14:33:41 +00:00
}
}
2013-08-10 16:15:26 +00:00
2013-09-09 14:33:41 +00:00
var (
uid int
status * int
pid * int
)
2013-08-11 14:27:45 +00:00
2013-09-09 14:33:41 +00:00
cols = [ ] string {
2014-03-19 02:00:26 +00:00
"id" , "Status" , "profile_id" ,
2013-09-09 14:33:41 +00:00
}
query = fmt . Sprintf ( "SELECT %s%s%s FROM %suser%s WHERE id = ?" , Q , strings . Join ( cols , sep ) , Q , Q , Q )
err = dORM . Raw ( query , 4 ) . QueryRow ( & uid , & status , & pid )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( uid , 4 ) )
throwFail ( t , AssertIs ( * status , 3 ) )
throwFail ( t , AssertIs ( pid , nil ) )
2013-09-09 14:33:41 +00:00
}
2013-08-11 14:27:45 +00:00
2017-05-14 04:03:34 +00:00
// user_profile table
type userProfile struct {
User
Age int
Money float64
}
2013-09-09 14:33:41 +00:00
func TestQueryRows ( t * testing . T ) {
Q := dDbBaser . TableQuote ( )
2013-08-11 14:27:45 +00:00
2013-09-09 14:33:41 +00:00
var datas [ ] * Data
2013-12-30 15:04:13 +00:00
query := fmt . Sprintf ( "SELECT * FROM %sdata%s" , Q , Q )
num , err := dORM . Raw ( query ) . QueryRows ( & datas )
2013-09-09 14:33:41 +00:00
throwFailNow ( t , err )
2013-09-12 11:04:39 +00:00
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( len ( datas ) , 1 ) )
2013-09-09 14:33:41 +00:00
ind := reflect . Indirect ( reflect . ValueOf ( datas [ 0 ] ) )
2015-09-12 13:46:43 +00:00
for name , value := range DataValues {
2013-09-09 14:33:41 +00:00
e := ind . FieldByName ( name )
vu := e . Interface ( )
switch name {
2016-04-02 17:57:47 +00:00
case "Time" :
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testTime )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testTime )
2013-09-09 14:33:41 +00:00
case "Date" :
2015-09-12 13:46:43 +00:00
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDate )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDate )
2013-09-09 14:33:41 +00:00
case "DateTime" :
2015-09-12 13:46:43 +00:00
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDateTime )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDateTime )
2013-08-11 14:27:45 +00:00
}
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( vu == value , true ) , value , vu )
2013-09-09 14:33:41 +00:00
}
2013-08-11 14:27:45 +00:00
2013-12-30 15:04:13 +00:00
var datas2 [ ] Data
2013-09-09 14:33:41 +00:00
2013-12-30 15:04:13 +00:00
query = fmt . Sprintf ( "SELECT * FROM %sdata%s" , Q , Q )
num , err = dORM . Raw ( query ) . QueryRows ( & datas2 )
2013-09-09 14:33:41 +00:00
throwFailNow ( t , err )
2013-12-30 15:04:13 +00:00
throwFailNow ( t , AssertIs ( num , 1 ) )
throwFailNow ( t , AssertIs ( len ( datas2 ) , 1 ) )
2013-09-09 14:33:41 +00:00
2013-12-30 15:04:13 +00:00
ind = reflect . Indirect ( reflect . ValueOf ( datas2 [ 0 ] ) )
2013-08-11 14:27:45 +00:00
2015-09-12 13:46:43 +00:00
for name , value := range DataValues {
2013-12-30 15:04:13 +00:00
e := ind . FieldByName ( name )
vu := e . Interface ( )
switch name {
2016-04-02 17:57:47 +00:00
case "Time" :
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testTime )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testTime )
2013-12-30 15:04:13 +00:00
case "Date" :
2015-09-12 13:46:43 +00:00
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDate )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDate )
2013-12-30 15:04:13 +00:00
case "DateTime" :
2015-09-12 13:46:43 +00:00
vu = vu . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDateTime )
value = value . ( time . Time ) . In ( DefaultTimeLoc ) . Format ( testDateTime )
2013-08-30 04:32:05 +00:00
}
2013-12-30 15:04:13 +00:00
throwFail ( t , AssertIs ( vu == value , true ) , value , vu )
2013-08-30 04:32:05 +00:00
}
2013-09-09 14:33:41 +00:00
2013-12-30 15:04:13 +00:00
var ids [ ] int
var usernames [ ] string
2013-12-31 01:55:29 +00:00
query = fmt . Sprintf ( "SELECT %sid%s, %suser_name%s FROM %suser%s ORDER BY %sid%s ASC" , Q , Q , Q , Q , Q , Q , Q , Q )
num , err = dORM . Raw ( query ) . QueryRows ( & ids , & usernames )
2013-12-30 15:04:13 +00:00
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 3 ) )
throwFailNow ( t , AssertIs ( len ( ids ) , 3 ) )
throwFailNow ( t , AssertIs ( ids [ 0 ] , 2 ) )
throwFailNow ( t , AssertIs ( usernames [ 0 ] , "slene" ) )
throwFailNow ( t , AssertIs ( ids [ 1 ] , 3 ) )
throwFailNow ( t , AssertIs ( usernames [ 1 ] , "astaxie" ) )
throwFailNow ( t , AssertIs ( ids [ 2 ] , 4 ) )
throwFailNow ( t , AssertIs ( usernames [ 2 ] , "nobody" ) )
2017-05-14 04:03:34 +00:00
//test query rows by nested struct
var l [ ] userProfile
query = fmt . Sprintf ( "SELECT * FROM %suser_profile%s LEFT JOIN %suser%s ON %suser_profile%s.%sid%s = %suser%s.%sid%s" , Q , Q , Q , Q , Q , Q , Q , Q , Q , Q , Q , Q )
num , err = dORM . Raw ( query ) . QueryRows ( & l )
throwFailNow ( t , err )
throwFailNow ( t , AssertIs ( num , 2 ) )
throwFailNow ( t , AssertIs ( len ( l ) , 2 ) )
throwFailNow ( t , AssertIs ( l [ 0 ] . UserName , "slene" ) )
throwFailNow ( t , AssertIs ( l [ 0 ] . Age , 28 ) )
throwFailNow ( t , AssertIs ( l [ 1 ] . UserName , "astaxie" ) )
throwFailNow ( t , AssertIs ( l [ 1 ] . Age , 30 ) )
2013-09-09 14:33:41 +00:00
}
func TestRawValues ( t * testing . T ) {
Q := dDbBaser . TableQuote ( )
var maps [ ] Params
2014-03-19 02:00:26 +00:00
query := fmt . Sprintf ( "SELECT %suser_name%s FROM %suser%s WHERE %sStatus%s = ?" , Q , Q , Q , Q , Q , Q )
2013-09-09 14:33:41 +00:00
num , err := dORM . Raw ( query , 1 ) . Values ( & maps )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-09-09 14:33:41 +00:00
if num == 1 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( maps [ 0 ] [ "user_name" ] , "slene" ) )
2013-09-09 14:33:41 +00:00
}
var lists [ ] ParamsList
num , err = dORM . Raw ( query , 1 ) . ValuesList ( & lists )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-09-09 14:33:41 +00:00
if num == 1 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( lists [ 0 ] [ 0 ] , "slene" ) )
2013-09-09 14:33:41 +00:00
}
query = fmt . Sprintf ( "SELECT %sprofile_id%s FROM %suser%s ORDER BY %sid%s ASC" , Q , Q , Q , Q , Q , Q )
var list ParamsList
num , err = dORM . Raw ( query ) . ValuesFlat ( & list )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-09-09 14:33:41 +00:00
if num == 3 {
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( list [ 0 ] , "2" ) )
throwFail ( t , AssertIs ( list [ 1 ] , "3" ) )
throwFail ( t , AssertIs ( list [ 2 ] , nil ) )
2013-09-09 14:33:41 +00:00
}
2013-08-30 04:32:05 +00:00
}
func TestRawPrepare ( t * testing . T ) {
switch {
case IsMysql || IsSqlite :
pre , err := dORM . Raw ( "INSERT INTO tag (name) VALUES (?)" ) . Prepare ( )
throwFail ( t , err )
if pre != nil {
r , err := pre . Exec ( "name1" )
throwFail ( t , err )
tid , err := r . LastInsertId ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( tid > 0 , true ) )
2013-08-30 04:32:05 +00:00
r , err = pre . Exec ( "name2" )
throwFail ( t , err )
id , err := r . LastInsertId ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , tid + 1 ) )
2013-08-30 04:32:05 +00:00
r , err = pre . Exec ( "name3" )
throwFail ( t , err )
id , err = r . LastInsertId ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id , tid + 2 ) )
2013-08-30 04:32:05 +00:00
err = pre . Close ( )
throwFail ( t , err )
res , err := dORM . Raw ( "DELETE FROM tag WHERE name IN (?, ?, ?)" , [ ] string { "name1" , "name2" , "name3" } ) . Exec ( )
throwFail ( t , err )
num , err := res . RowsAffected ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-11 14:27:45 +00:00
}
2013-08-30 04:32:05 +00:00
case IsPostgres :
2013-08-11 14:27:45 +00:00
pre , err := dORM . Raw ( ` INSERT INTO "tag" ("name") VALUES (?) RETURNING "id" ` ) . Prepare ( )
throwFail ( t , err )
if pre != nil {
_ , err := pre . Exec ( "name1" )
throwFail ( t , err )
_ , err = pre . Exec ( "name2" )
throwFail ( t , err )
_ , err = pre . Exec ( "name3" )
throwFail ( t , err )
err = pre . Close ( )
throwFail ( t , err )
res , err := dORM . Raw ( ` DELETE FROM "tag" WHERE "name" IN (?, ?, ?) ` , [ ] string { "name1" , "name2" , "name3" } ) . Exec ( )
throwFail ( t , err )
2013-08-13 09:16:12 +00:00
if err == nil {
num , err := res . RowsAffected ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 3 ) )
2013-08-13 09:16:12 +00:00
}
2013-08-11 14:27:45 +00:00
}
2013-08-07 11:11:44 +00:00
}
}
func TestUpdate ( t * testing . T ) {
qs := dORM . QueryTable ( "user" )
2013-08-10 16:15:26 +00:00
num , err := qs . Filter ( "user_name" , "slene" ) . Filter ( "is_staff" , false ) . Update ( Params {
2013-11-01 10:32:03 +00:00
"is_staff" : true ,
"is_active" : true ,
2013-08-07 11:11:44 +00:00
} )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-10 16:15:26 +00:00
// with join
num , err = qs . Filter ( "user_name" , "slene" ) . Filter ( "profile__age" , 28 ) . Filter ( "is_staff" , true ) . Update ( Params {
"is_staff" : false ,
} )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-10-09 12:28:54 +00:00
num , err = qs . Filter ( "user_name" , "slene" ) . Update ( Params {
2015-09-12 13:46:43 +00:00
"Nums" : ColValue ( ColAdd , 100 ) ,
2013-10-09 12:28:54 +00:00
} )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = qs . Filter ( "user_name" , "slene" ) . Update ( Params {
2015-09-12 13:46:43 +00:00
"Nums" : ColValue ( ColMinus , 50 ) ,
2013-10-09 12:28:54 +00:00
} )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = qs . Filter ( "user_name" , "slene" ) . Update ( Params {
2015-09-12 13:46:43 +00:00
"Nums" : ColValue ( ColMultiply , 3 ) ,
2013-10-09 12:28:54 +00:00
} )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = qs . Filter ( "user_name" , "slene" ) . Update ( Params {
2015-09-12 13:46:43 +00:00
"Nums" : ColValue ( ColExcept , 5 ) ,
2013-10-09 12:28:54 +00:00
} )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
user := User { UserName : "slene" }
err = dORM . Read ( & user , "UserName" )
throwFail ( t , err )
throwFail ( t , AssertIs ( user . Nums , 30 ) )
2013-08-07 11:11:44 +00:00
}
func TestDelete ( t * testing . T ) {
qs := dORM . QueryTable ( "user_profile" )
num , err := qs . Filter ( "user__user_name" , "slene" ) . Delete ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
qs = dORM . QueryTable ( "user" )
num , err = qs . Filter ( "user_name" , "slene" ) . Filter ( "profile__isnull" , true ) . Count ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2014-01-04 14:30:17 +00:00
qs = dORM . QueryTable ( "comment" )
num , err = qs . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 6 ) )
qs = dORM . QueryTable ( "post" )
num , err = qs . Filter ( "Id" , 3 ) . Delete ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
qs = dORM . QueryTable ( "comment" )
num , err = qs . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 4 ) )
qs = dORM . QueryTable ( "comment" )
num , err = qs . Filter ( "Post__User" , 3 ) . Delete ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 3 ) )
qs = dORM . QueryTable ( "comment" )
num , err = qs . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
}
func TestTransaction ( t * testing . T ) {
2013-08-10 16:15:26 +00:00
// this test worked when database support transaction
2013-08-08 14:34:18 +00:00
o := NewOrm ( )
err := o . Begin ( )
throwFail ( t , err )
var names = [ ] string { "1" , "2" , "3" }
2013-08-10 16:15:26 +00:00
var tag Tag
tag . Name = names [ 0 ]
id , err := o . Insert ( & tag )
2013-08-08 14:34:18 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id > 0 , true ) )
2013-08-08 14:34:18 +00:00
2013-08-10 16:15:26 +00:00
num , err := o . QueryTable ( "tag" ) . Filter ( "name" , "golang" ) . Update ( Params { "name" : names [ 1 ] } )
2013-08-08 14:34:18 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-08 14:34:18 +00:00
2013-08-10 16:15:26 +00:00
switch {
case IsMysql || IsSqlite :
res , err := o . Raw ( "INSERT INTO tag (name) VALUES (?)" , names [ 2 ] ) . Exec ( )
2013-08-08 14:34:18 +00:00
throwFail ( t , err )
2013-08-10 16:15:26 +00:00
if err == nil {
id , err = res . LastInsertId ( )
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id > 0 , true ) )
2013-08-10 16:15:26 +00:00
}
2013-08-08 14:34:18 +00:00
}
err = o . Rollback ( )
throwFail ( t , err )
2013-08-10 16:15:26 +00:00
num , err = o . QueryTable ( "tag" ) . Filter ( "name__in" , names ) . Count ( )
2013-08-08 14:34:18 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 0 ) )
2013-08-08 14:34:18 +00:00
err = o . Begin ( )
throwFail ( t , err )
2013-08-10 16:15:26 +00:00
tag . Name = "commit"
id , err = o . Insert ( & tag )
2013-08-08 14:34:18 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( id > 0 , true ) )
2013-08-08 14:34:18 +00:00
o . Commit ( )
throwFail ( t , err )
2013-08-10 16:15:26 +00:00
num , err = o . QueryTable ( "tag" ) . Filter ( "name" , "commit" ) . Delete ( )
2013-08-08 14:34:18 +00:00
throwFail ( t , err )
2013-09-12 11:04:39 +00:00
throwFail ( t , AssertIs ( num , 1 ) )
2013-08-07 11:11:44 +00:00
}
2014-01-22 05:58:57 +00:00
2018-05-04 08:38:03 +00:00
func TestTransactionIsolationLevel ( t * testing . T ) {
// this test worked when database support transaction isolation level
if IsSqlite {
return
}
o1 := NewOrm ( )
o2 := NewOrm ( )
// start two transaction with isolation level repeatable read
err := o1 . BeginTx ( context . Background ( ) , & sql . TxOptions { Isolation : sql . LevelRepeatableRead } )
throwFail ( t , err )
err = o2 . BeginTx ( context . Background ( ) , & sql . TxOptions { Isolation : sql . LevelRepeatableRead } )
throwFail ( t , err )
// o1 insert tag
var tag Tag
tag . Name = "test-transaction"
id , err := o1 . Insert ( & tag )
throwFail ( t , err )
throwFail ( t , AssertIs ( id > 0 , true ) )
// o2 query tag table, no result
num , err := o2 . QueryTable ( "tag" ) . Filter ( "name" , "test-transaction" ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 0 ) )
// o1 commit
o1 . Commit ( )
// o2 query tag table, still no result
num , err = o2 . QueryTable ( "tag" ) . Filter ( "name" , "test-transaction" ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 0 ) )
// o2 commit and query tag table, get the result
o2 . Commit ( )
num , err = o2 . QueryTable ( "tag" ) . Filter ( "name" , "test-transaction" ) . Count ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
num , err = o1 . QueryTable ( "tag" ) . Filter ( "name" , "test-transaction" ) . Delete ( )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
}
func TestBeginTxWithContextCanceled ( t * testing . T ) {
o := NewOrm ( )
ctx , cancel := context . WithCancel ( context . Background ( ) )
o . BeginTx ( ctx , nil )
id , err := o . Insert ( & Tag { Name : "test-context" } )
throwFail ( t , err )
throwFail ( t , AssertIs ( id > 0 , true ) )
// cancel the context before commit to make it error
cancel ( )
err = o . Commit ( )
throwFail ( t , AssertIs ( err , context . Canceled ) )
}
2014-01-22 05:58:57 +00:00
func TestReadOrCreate ( t * testing . T ) {
u := & User {
UserName : "Kyle" ,
2014-02-28 03:53:35 +00:00
Email : "kylemcc@gmail.com" ,
2014-01-22 05:58:57 +00:00
Password : "other_pass" ,
2014-02-28 03:53:35 +00:00
Status : 7 ,
IsStaff : false ,
2014-01-22 05:58:57 +00:00
IsActive : true ,
}
created , pk , err := dORM . ReadOrCreate ( u , "UserName" )
throwFail ( t , err )
throwFail ( t , AssertIs ( created , true ) )
2017-04-28 14:36:28 +00:00
throwFail ( t , AssertIs ( u . ID , pk ) )
2014-01-22 05:58:57 +00:00
throwFail ( t , AssertIs ( u . UserName , "Kyle" ) )
throwFail ( t , AssertIs ( u . Email , "kylemcc@gmail.com" ) )
throwFail ( t , AssertIs ( u . Password , "other_pass" ) )
throwFail ( t , AssertIs ( u . Status , 7 ) )
throwFail ( t , AssertIs ( u . IsStaff , false ) )
throwFail ( t , AssertIs ( u . IsActive , true ) )
2015-09-12 13:46:43 +00:00
throwFail ( t , AssertIs ( u . Created . In ( DefaultTimeLoc ) , u . Created . In ( DefaultTimeLoc ) , testDate ) )
throwFail ( t , AssertIs ( u . Updated . In ( DefaultTimeLoc ) , u . Updated . In ( DefaultTimeLoc ) , testDateTime ) )
2014-01-22 05:58:57 +00:00
nu := & User { UserName : u . UserName , Email : "someotheremail@gmail.com" }
created , pk , err = dORM . ReadOrCreate ( nu , "UserName" )
throwFail ( t , err )
throwFail ( t , AssertIs ( created , false ) )
2015-09-12 13:46:43 +00:00
throwFail ( t , AssertIs ( nu . ID , u . ID ) )
throwFail ( t , AssertIs ( pk , u . ID ) )
2014-01-22 05:58:57 +00:00
throwFail ( t , AssertIs ( nu . UserName , u . UserName ) )
throwFail ( t , AssertIs ( nu . Email , u . Email ) ) // should contain the value in the table, not the one specified above
throwFail ( t , AssertIs ( nu . Password , u . Password ) )
throwFail ( t , AssertIs ( nu . Status , u . Status ) )
throwFail ( t , AssertIs ( nu . IsStaff , u . IsStaff ) )
throwFail ( t , AssertIs ( nu . IsActive , u . IsActive ) )
dORM . Delete ( u )
}
2016-03-08 14:24:38 +00:00
func TestInLine ( t * testing . T ) {
name := "inline"
email := "hello@go.com"
inline := NewInLine ( )
inline . Name = name
inline . Email = email
id , err := dORM . Insert ( inline )
throwFail ( t , err )
throwFail ( t , AssertIs ( id , 1 ) )
il := NewInLine ( )
2016-03-10 14:21:21 +00:00
il . ID = 1
2016-03-08 14:24:38 +00:00
err = dORM . Read ( il )
throwFail ( t , err )
throwFail ( t , AssertIs ( il . Name , name ) )
throwFail ( t , AssertIs ( il . Email , email ) )
throwFail ( t , AssertIs ( il . Created . In ( DefaultTimeLoc ) , inline . Created . In ( DefaultTimeLoc ) , testDate ) )
throwFail ( t , AssertIs ( il . Updated . In ( DefaultTimeLoc ) , inline . Updated . In ( DefaultTimeLoc ) , testDateTime ) )
}
2016-03-13 13:01:56 +00:00
func TestInLineOneToOne ( t * testing . T ) {
name := "121"
email := "121@go.com"
inline := NewInLine ( )
inline . Name = name
inline . Email = email
id , err := dORM . Insert ( inline )
throwFail ( t , err )
throwFail ( t , AssertIs ( id , 2 ) )
note := "one2one"
il121 := NewInLineOneToOne ( )
il121 . Note = note
il121 . InLine = inline
_ , err = dORM . Insert ( il121 )
throwFail ( t , err )
throwFail ( t , AssertIs ( il121 . ID , 1 ) )
il := NewInLineOneToOne ( )
err = dORM . QueryTable ( il ) . Filter ( "Id" , 1 ) . RelatedSel ( ) . One ( il )
throwFail ( t , err )
throwFail ( t , AssertIs ( il . Note , note ) )
throwFail ( t , AssertIs ( il . InLine . ID , id ) )
throwFail ( t , AssertIs ( il . InLine . Name , name ) )
throwFail ( t , AssertIs ( il . InLine . Email , email ) )
rinline := NewInLine ( )
err = dORM . QueryTable ( rinline ) . Filter ( "InLineOneToOne__Id" , 1 ) . One ( rinline )
throwFail ( t , err )
throwFail ( t , AssertIs ( rinline . ID , id ) )
throwFail ( t , AssertIs ( rinline . Name , name ) )
throwFail ( t , AssertIs ( rinline . Email , email ) )
}
2016-03-18 13:58:11 +00:00
func TestIntegerPk ( t * testing . T ) {
its := [ ] IntegerPk {
2016-04-09 02:22:40 +00:00
{ ID : math . MinInt64 , Value : "-" } ,
{ ID : 0 , Value : "0" } ,
{ ID : math . MaxInt64 , Value : "+" } ,
2016-03-18 13:58:11 +00:00
}
num , err := dORM . InsertMulti ( len ( its ) , its )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , len ( its ) ) )
for _ , intPk := range its {
2016-04-09 02:22:40 +00:00
out := IntegerPk { ID : intPk . ID }
2016-03-18 13:58:11 +00:00
err = dORM . Read ( & out )
throwFail ( t , err )
throwFail ( t , AssertIs ( out . Value , intPk . Value ) )
}
2016-04-13 12:20:49 +00:00
2016-08-17 14:49:30 +00:00
num , err = dORM . InsertMulti ( 1 , [ ] * IntegerPk { {
2016-04-13 12:20:49 +00:00
ID : 1 , Value : "ok" ,
} } )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
2016-03-18 13:58:11 +00:00
}
2016-03-23 12:28:22 +00:00
func TestInsertAuto ( t * testing . T ) {
u := & User {
UserName : "autoPre" ,
Email : "autoPre@gmail.com" ,
}
id , err := dORM . Insert ( u )
throwFail ( t , err )
id += 100
su := & User {
ID : int ( id ) ,
UserName : "auto" ,
Email : "auto@gmail.com" ,
}
2016-03-26 13:51:05 +00:00
nid , err := dORM . Insert ( su )
2016-03-23 12:28:22 +00:00
throwFail ( t , err )
2016-03-26 13:51:05 +00:00
throwFail ( t , AssertIs ( nid , id ) )
users := [ ] User {
{ ID : int ( id + 100 ) , UserName : "auto_100" } ,
{ ID : int ( id + 110 ) , UserName : "auto_110" } ,
{ ID : int ( id + 120 ) , UserName : "auto_120" } ,
}
num , err := dORM . InsertMulti ( 100 , users )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 3 ) )
u = & User {
UserName : "auto_121" ,
}
nid , err = dORM . Insert ( u )
throwFail ( t , err )
throwFail ( t , AssertIs ( nid , id + 120 + 1 ) )
2016-03-23 12:28:22 +00:00
}
2016-03-26 13:16:52 +00:00
2016-03-23 13:32:52 +00:00
func TestUintPk ( t * testing . T ) {
name := "go"
u := & UintPk {
2016-04-09 02:22:40 +00:00
ID : 8 ,
2016-03-23 13:32:52 +00:00
Name : name ,
}
2017-04-28 14:58:17 +00:00
created , _ , err := dORM . ReadOrCreate ( u , "ID" )
2016-03-23 13:32:52 +00:00
throwFail ( t , err )
throwFail ( t , AssertIs ( created , true ) )
throwFail ( t , AssertIs ( u . Name , name ) )
2016-04-09 02:22:40 +00:00
nu := & UintPk { ID : 8 }
2017-04-30 14:55:39 +00:00
created , pk , err := dORM . ReadOrCreate ( nu , "ID" )
2016-03-23 13:32:52 +00:00
throwFail ( t , err )
throwFail ( t , AssertIs ( created , false ) )
2016-04-09 02:22:40 +00:00
throwFail ( t , AssertIs ( nu . ID , u . ID ) )
throwFail ( t , AssertIs ( pk , u . ID ) )
2016-03-23 13:32:52 +00:00
throwFail ( t , AssertIs ( nu . Name , name ) )
dORM . Delete ( u )
}
2016-06-03 14:06:43 +00:00
2017-01-11 10:51:32 +00:00
func TestPtrPk ( t * testing . T ) {
parent := & IntegerPk { ID : 10 , Value : "10" }
id , _ := dORM . Insert ( parent )
if ! IsMysql {
// MySql does not support last_insert_id in this case: see #2382
throwFail ( t , AssertIs ( id , 10 ) )
}
ptr := PtrPk { ID : parent , Positive : true }
num , err := dORM . InsertMulti ( 2 , [ ] PtrPk { ptr } )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
throwFail ( t , AssertIs ( ptr . ID , parent ) )
nptr := & PtrPk { ID : parent }
created , pk , err := dORM . ReadOrCreate ( nptr , "ID" )
throwFail ( t , err )
throwFail ( t , AssertIs ( created , false ) )
throwFail ( t , AssertIs ( pk , 10 ) )
throwFail ( t , AssertIs ( nptr . ID , parent ) )
throwFail ( t , AssertIs ( nptr . Positive , true ) )
nptr = & PtrPk { Positive : true }
created , pk , err = dORM . ReadOrCreate ( nptr , "Positive" )
throwFail ( t , err )
throwFail ( t , AssertIs ( created , false ) )
throwFail ( t , AssertIs ( pk , 10 ) )
throwFail ( t , AssertIs ( nptr . ID , parent ) )
nptr . Positive = false
num , err = dORM . Update ( nptr )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
throwFail ( t , AssertIs ( nptr . ID , parent ) )
throwFail ( t , AssertIs ( nptr . Positive , false ) )
num , err = dORM . Delete ( nptr )
throwFail ( t , err )
throwFail ( t , AssertIs ( num , 1 ) )
}
2016-06-03 14:06:43 +00:00
func TestSnake ( t * testing . T ) {
cases := map [ string ] string {
"i" : "i" ,
"I" : "i" ,
2016-06-04 02:41:55 +00:00
"iD" : "i_d" ,
2016-08-29 16:15:02 +00:00
"ID" : "i_d" ,
"NO" : "n_o" ,
"NOO" : "n_o_o" ,
"NOOooOOoo" : "n_o_ooo_o_ooo" ,
"OrderNO" : "order_n_o" ,
2016-06-04 02:41:55 +00:00
"tagName" : "tag_name" ,
2016-08-29 16:15:02 +00:00
"tag_Name" : "tag__name" ,
2016-06-03 14:06:43 +00:00
"tag_name" : "tag_name" ,
"_tag_name" : "_tag_name" ,
"tag_666name" : "tag_666name" ,
"tag_666Name" : "tag_666_name" ,
}
for name , want := range cases {
got := snakeString ( name )
throwFail ( t , AssertIs ( got , want ) )
}
}
func TestIgnoreCaseTag ( t * testing . T ) {
type testTagModel struct {
ID int ` orm:"pk" `
NOO string ` orm:"column(n)" `
Name01 string ` orm:"NULL" `
Name02 string ` orm:"COLUMN(Name)" `
Name03 string ` orm:"Column(name)" `
}
2016-06-04 02:41:55 +00:00
modelCache . clean ( )
2016-06-03 14:06:43 +00:00
RegisterModel ( & testTagModel { } )
info , ok := modelCache . get ( "test_tag_model" )
throwFail ( t , AssertIs ( ok , true ) )
throwFail ( t , AssertNot ( info , nil ) )
if t == nil {
return
}
throwFail ( t , AssertIs ( info . fields . GetByName ( "NOO" ) . column , "n" ) )
throwFail ( t , AssertIs ( info . fields . GetByName ( "Name01" ) . null , true ) )
throwFail ( t , AssertIs ( info . fields . GetByName ( "Name02" ) . column , "Name" ) )
throwFail ( t , AssertIs ( info . fields . GetByName ( "Name03" ) . column , "name" ) )
}
2018-05-04 08:38:03 +00:00
2016-07-20 06:37:05 +00:00
func TestInsertOrUpdate ( t * testing . T ) {
2016-07-20 07:13:18 +00:00
RegisterModel ( new ( User ) )
2016-07-20 06:37:05 +00:00
user := User { UserName : "unique_username133" , Status : 1 , Password : "o" }
user1 := User { UserName : "unique_username133" , Status : 2 , Password : "o" }
user2 := User { UserName : "unique_username133" , Status : 3 , Password : "oo" }
dORM . Insert ( & user )
2016-07-20 07:13:18 +00:00
test := User { UserName : "unique_username133" }
2016-07-20 06:37:05 +00:00
fmt . Println ( dORM . Driver ( ) . Name ( ) )
if dORM . Driver ( ) . Name ( ) == "sqlite3" {
fmt . Println ( "sqlite3 is nonsupport" )
return
}
2016-07-20 09:28:26 +00:00
//test1
2016-07-20 08:52:14 +00:00
_ , err := dORM . InsertOrUpdate ( & user1 , "user_name" )
2016-07-26 04:19:06 +00:00
if err != nil {
2016-07-20 06:37:05 +00:00
fmt . Println ( err )
2016-07-26 04:19:06 +00:00
if err . Error ( ) == "postgres version must 9.5 or higher" || err . Error ( ) == "`sqlite3` nonsupport InsertOrUpdate in beego" {
2016-07-26 04:44:16 +00:00
} else {
throwFailNow ( t , err )
2016-07-26 04:19:06 +00:00
}
2016-07-20 06:37:05 +00:00
} else {
2016-07-20 08:52:14 +00:00
dORM . Read ( & test , "user_name" )
2016-07-20 07:13:18 +00:00
throwFailNow ( t , AssertIs ( user1 . Status , test . Status ) )
2016-07-20 06:37:05 +00:00
}
2016-07-20 09:28:26 +00:00
//test2
2016-07-20 08:52:14 +00:00
_ , err = dORM . InsertOrUpdate ( & user2 , "user_name" )
2016-07-26 04:19:06 +00:00
if err != nil {
2016-07-20 06:37:05 +00:00
fmt . Println ( err )
2016-07-26 04:19:06 +00:00
if err . Error ( ) == "postgres version must 9.5 or higher" || err . Error ( ) == "`sqlite3` nonsupport InsertOrUpdate in beego" {
2016-07-26 04:44:16 +00:00
} else {
throwFailNow ( t , err )
2016-07-26 04:19:06 +00:00
}
2016-07-20 06:37:05 +00:00
} else {
2016-07-20 08:52:14 +00:00
dORM . Read ( & test , "user_name" )
2016-07-20 06:37:05 +00:00
throwFailNow ( t , AssertIs ( user2 . Status , test . Status ) )
throwFailNow ( t , AssertIs ( user2 . Password , strings . TrimSpace ( test . Password ) ) )
}
2018-05-04 08:15:50 +00:00
//postgres ON CONFLICT DO UPDATE SET can`t use colu=colu+values
if IsPostgres {
return
}
2016-07-20 09:28:26 +00:00
//test3 +
2016-07-26 04:19:06 +00:00
_ , err = dORM . InsertOrUpdate ( & user2 , "user_name" , "status=status+1" )
if err != nil {
2016-07-20 06:37:05 +00:00
fmt . Println ( err )
2016-07-26 04:19:06 +00:00
if err . Error ( ) == "postgres version must 9.5 or higher" || err . Error ( ) == "`sqlite3` nonsupport InsertOrUpdate in beego" {
2016-07-26 04:44:16 +00:00
} else {
throwFailNow ( t , err )
2016-07-26 04:19:06 +00:00
}
2016-07-20 06:37:05 +00:00
} else {
2016-07-20 08:52:14 +00:00
dORM . Read ( & test , "user_name" )
2016-07-20 06:37:05 +00:00
throwFailNow ( t , AssertIs ( user2 . Status + 1 , test . Status ) )
}
2016-07-20 09:28:26 +00:00
//test4 -
2016-07-26 04:19:06 +00:00
_ , err = dORM . InsertOrUpdate ( & user2 , "user_name" , "status=status-1" )
if err != nil {
2016-07-20 06:37:05 +00:00
fmt . Println ( err )
2016-07-26 04:19:06 +00:00
if err . Error ( ) == "postgres version must 9.5 or higher" || err . Error ( ) == "`sqlite3` nonsupport InsertOrUpdate in beego" {
2016-07-26 04:44:16 +00:00
} else {
throwFailNow ( t , err )
2016-07-26 04:19:06 +00:00
}
2016-07-20 06:37:05 +00:00
} else {
2016-07-20 08:52:14 +00:00
dORM . Read ( & test , "user_name" )
2016-07-20 06:37:05 +00:00
throwFailNow ( t , AssertIs ( ( user2 . Status + 1 ) - 1 , test . Status ) )
}
2016-07-20 09:28:26 +00:00
//test5 *
2016-07-26 04:19:06 +00:00
_ , err = dORM . InsertOrUpdate ( & user2 , "user_name" , "status=status*3" )
if err != nil {
2016-07-20 06:37:05 +00:00
fmt . Println ( err )
2016-07-26 04:19:06 +00:00
if err . Error ( ) == "postgres version must 9.5 or higher" || err . Error ( ) == "`sqlite3` nonsupport InsertOrUpdate in beego" {
2016-07-26 04:44:16 +00:00
} else {
throwFailNow ( t , err )
2016-07-26 04:19:06 +00:00
}
2016-07-20 06:37:05 +00:00
} else {
2016-07-20 08:52:14 +00:00
dORM . Read ( & test , "user_name" )
2016-07-20 06:37:05 +00:00
throwFailNow ( t , AssertIs ( ( ( user2 . Status + 1 ) - 1 ) * 3 , test . Status ) )
}
2016-07-20 09:28:26 +00:00
//test6 /
2016-07-20 08:52:14 +00:00
_ , err = dORM . InsertOrUpdate ( & user2 , "user_name" , "Status=Status/3" )
2016-07-26 04:19:06 +00:00
if err != nil {
2016-07-20 06:37:05 +00:00
fmt . Println ( err )
2016-07-26 04:19:06 +00:00
if err . Error ( ) == "postgres version must 9.5 or higher" || err . Error ( ) == "`sqlite3` nonsupport InsertOrUpdate in beego" {
2016-07-26 04:44:16 +00:00
} else {
throwFailNow ( t , err )
2016-07-26 04:19:06 +00:00
}
2016-07-20 06:37:05 +00:00
} else {
2016-07-20 08:52:14 +00:00
dORM . Read ( & test , "user_name" )
2016-07-20 06:37:05 +00:00
throwFailNow ( t , AssertIs ( ( ( ( user2 . Status + 1 ) - 1 ) * 3 ) / 3 , test . Status ) )
}
}