session_update_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110
  1. // Copyright 2017 The Xorm Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style
  3. // license that can be found in the LICENSE file.
  4. package xorm
  5. import (
  6. "errors"
  7. "fmt"
  8. "sync"
  9. "testing"
  10. "time"
  11. "github.com/stretchr/testify/assert"
  12. "github.com/xormplus/core"
  13. )
  14. func TestUpdateMap(t *testing.T) {
  15. assert.NoError(t, prepareEngine())
  16. type UpdateTable struct {
  17. Id int64
  18. Name string
  19. Age int
  20. }
  21. assert.NoError(t, testEngine.Sync2(new(UpdateTable)))
  22. var tb = UpdateTable{
  23. Name: "test",
  24. Age: 35,
  25. }
  26. _, err := testEngine.Insert(&tb)
  27. assert.NoError(t, err)
  28. cnt, err := testEngine.Table("update_table").Where("id = ?", tb.Id).Update(map[string]interface{}{
  29. "name": "test2",
  30. "age": 36,
  31. })
  32. assert.NoError(t, err)
  33. assert.EqualValues(t, 1, cnt)
  34. }
  35. func TestUpdateLimit(t *testing.T) {
  36. assert.NoError(t, prepareEngine())
  37. type UpdateTable struct {
  38. Id int64
  39. Name string
  40. Age int
  41. }
  42. assert.NoError(t, testEngine.Sync2(new(UpdateTable)))
  43. var tb = UpdateTable{
  44. Name: "test1",
  45. Age: 35,
  46. }
  47. cnt, err := testEngine.Insert(&tb)
  48. assert.NoError(t, err)
  49. assert.EqualValues(t, 1, cnt)
  50. tb.Name = "test2"
  51. tb.Id = 0
  52. cnt, err = testEngine.Insert(&tb)
  53. assert.NoError(t, err)
  54. assert.EqualValues(t, 1, cnt)
  55. cnt, err = testEngine.OrderBy("name desc").Limit(1).Update(&UpdateTable{
  56. Age: 30,
  57. })
  58. assert.NoError(t, err)
  59. assert.EqualValues(t, 1, cnt)
  60. var uts []UpdateTable
  61. err = testEngine.Find(&uts)
  62. assert.NoError(t, err)
  63. assert.EqualValues(t, 2, len(uts))
  64. assert.EqualValues(t, 35, uts[0].Age)
  65. assert.EqualValues(t, 30, uts[1].Age)
  66. }
  67. type ForUpdate struct {
  68. Id int64 `xorm:"pk"`
  69. Name string
  70. }
  71. func setupForUpdate(engine *Engine) error {
  72. v := new(ForUpdate)
  73. err := testEngine.DropTables(v)
  74. if err != nil {
  75. return err
  76. }
  77. err = testEngine.CreateTables(v)
  78. if err != nil {
  79. return err
  80. }
  81. list := []ForUpdate{
  82. {1, "data1"},
  83. {2, "data2"},
  84. {3, "data3"},
  85. }
  86. for _, f := range list {
  87. _, err = testEngine.Insert(f)
  88. if err != nil {
  89. return err
  90. }
  91. }
  92. return nil
  93. }
  94. func TestForUpdate(t *testing.T) {
  95. if testEngine.DriverName() != "mysql" && testEngine.DriverName() != "mymysql" {
  96. return
  97. }
  98. err := setupForUpdate(testEngine)
  99. if err != nil {
  100. t.Error(err)
  101. return
  102. }
  103. session1 := testEngine.NewSession()
  104. session2 := testEngine.NewSession()
  105. session3 := testEngine.NewSession()
  106. defer session1.Close()
  107. defer session2.Close()
  108. defer session3.Close()
  109. // start transaction
  110. err = session1.Begin()
  111. if err != nil {
  112. t.Error(err)
  113. return
  114. }
  115. // use lock
  116. fList := make([]ForUpdate, 0)
  117. session1.ForUpdate()
  118. session1.Where("(id) = ?", 1)
  119. err = session1.Find(&fList)
  120. switch {
  121. case err != nil:
  122. t.Error(err)
  123. return
  124. case len(fList) != 1:
  125. t.Errorf("find not returned single row")
  126. return
  127. case fList[0].Name != "data1":
  128. t.Errorf("for_update.name must be `data1`")
  129. return
  130. }
  131. // wait for lock
  132. wg := &sync.WaitGroup{}
  133. // lock is used
  134. wg.Add(1)
  135. go func() {
  136. f2 := new(ForUpdate)
  137. session2.Where("(id) = ?", 1).ForUpdate()
  138. has, err := session2.Get(f2) // wait release lock
  139. switch {
  140. case err != nil:
  141. t.Error(err)
  142. case !has:
  143. t.Errorf("cannot find target row. for_update.id = 1")
  144. case f2.Name != "updated by session1":
  145. t.Errorf("read lock failed")
  146. }
  147. wg.Done()
  148. }()
  149. // lock is NOT used
  150. wg.Add(1)
  151. go func() {
  152. f3 := new(ForUpdate)
  153. session3.Where("(id) = ?", 1)
  154. has, err := session3.Get(f3) // wait release lock
  155. switch {
  156. case err != nil:
  157. t.Error(err)
  158. case !has:
  159. t.Errorf("cannot find target row. for_update.id = 1")
  160. case f3.Name != "data1":
  161. t.Errorf("read lock failed")
  162. }
  163. wg.Done()
  164. }()
  165. // wait for go rountines
  166. time.Sleep(50 * time.Millisecond)
  167. f := new(ForUpdate)
  168. f.Name = "updated by session1"
  169. session1.Where("(id) = ?", 1)
  170. session1.Update(f)
  171. // release lock
  172. err = session1.Commit()
  173. if err != nil {
  174. t.Error(err)
  175. return
  176. }
  177. wg.Wait()
  178. }
  179. func TestWithIn(t *testing.T) {
  180. type temp3 struct {
  181. Id int64 `xorm:"Id pk autoincr"`
  182. Name string `xorm:"Name"`
  183. Test bool `xorm:"Test"`
  184. }
  185. assert.NoError(t, prepareEngine())
  186. assert.NoError(t, testEngine.Sync(new(temp3)))
  187. testEngine.Insert(&[]temp3{
  188. {
  189. Name: "user1",
  190. },
  191. {
  192. Name: "user1",
  193. },
  194. {
  195. Name: "user1",
  196. },
  197. })
  198. cnt, err := testEngine.In("Id", 1, 2, 3, 4).Update(&temp3{Name: "aa"}, &temp3{Name: "user1"})
  199. assert.NoError(t, err)
  200. assert.EqualValues(t, 3, cnt)
  201. }
  202. type Condi map[string]interface{}
  203. type UpdateAllCols struct {
  204. Id int64
  205. Bool bool
  206. String string
  207. Ptr *string
  208. }
  209. type UpdateMustCols struct {
  210. Id int64
  211. Bool bool
  212. String string
  213. }
  214. type UpdateIncr struct {
  215. Id int64
  216. Cnt int
  217. }
  218. type Article struct {
  219. Id int32 `xorm:"pk INT autoincr"`
  220. Name string `xorm:"VARCHAR(45)"`
  221. Img string `xorm:"VARCHAR(100)"`
  222. Aside string `xorm:"VARCHAR(200)"`
  223. Desc string `xorm:"VARCHAR(200)"`
  224. Content string `xorm:"TEXT"`
  225. Status int8 `xorm:"TINYINT(4)"`
  226. }
  227. func TestUpdateMap2(t *testing.T) {
  228. assert.NoError(t, prepareEngine())
  229. assertSync(t, new(UpdateMustCols))
  230. _, err := testEngine.Table("update_must_cols").Where("id =?", 1).Update(map[string]interface{}{
  231. "bool": true,
  232. })
  233. if err != nil {
  234. t.Error(err)
  235. panic(err)
  236. }
  237. }
  238. func TestUpdate1(t *testing.T) {
  239. assert.NoError(t, prepareEngine())
  240. assertSync(t, new(Userinfo))
  241. _, err := testEngine.Insert(&Userinfo{
  242. Username: "user1",
  243. })
  244. var ori Userinfo
  245. has, err := testEngine.Get(&ori)
  246. if err != nil {
  247. t.Error(err)
  248. panic(err)
  249. }
  250. if !has {
  251. t.Error(errors.New("not exist"))
  252. panic(errors.New("not exist"))
  253. }
  254. // update by id
  255. user := Userinfo{Username: "xxx", Height: 1.2}
  256. cnt, err := testEngine.Id(ori.Uid).Update(&user)
  257. if err != nil {
  258. t.Error(err)
  259. panic(err)
  260. }
  261. if cnt != 1 {
  262. err = errors.New("update not returned 1")
  263. t.Error(err)
  264. panic(err)
  265. return
  266. }
  267. condi := Condi{"username": "zzz", "departname": ""}
  268. cnt, err = testEngine.Table(&user).Id(ori.Uid).Update(&condi)
  269. if err != nil {
  270. t.Error(err)
  271. panic(err)
  272. }
  273. if cnt != 1 {
  274. err = errors.New("update not returned 1")
  275. t.Error(err)
  276. panic(err)
  277. return
  278. }
  279. cnt, err = testEngine.Update(&Userinfo{Username: "yyy"}, &user)
  280. if err != nil {
  281. t.Error(err)
  282. panic(err)
  283. }
  284. total, err := testEngine.Count(&user)
  285. if err != nil {
  286. t.Error(err)
  287. panic(err)
  288. }
  289. if cnt != total {
  290. err = errors.New("insert not returned 1")
  291. t.Error(err)
  292. panic(err)
  293. return
  294. }
  295. // nullable update
  296. {
  297. user := &Userinfo{Username: "not null data", Height: 180.5}
  298. _, err := testEngine.Insert(user)
  299. if err != nil {
  300. t.Error(err)
  301. panic(err)
  302. }
  303. userID := user.Uid
  304. has, err := testEngine.Id(userID).
  305. And("username = ?", user.Username).
  306. And("height = ?", user.Height).
  307. And("departname = ?", "").
  308. And("detail_id = ?", 0).
  309. And("is_man = ?", 0).
  310. Get(&Userinfo{})
  311. if err != nil {
  312. t.Error(err)
  313. panic(err)
  314. }
  315. if !has {
  316. err = errors.New("cannot insert properly")
  317. t.Error(err)
  318. panic(err)
  319. }
  320. updatedUser := &Userinfo{Username: "null data"}
  321. cnt, err = testEngine.Id(userID).
  322. Nullable("height", "departname", "is_man", "created").
  323. Update(updatedUser)
  324. if err != nil {
  325. t.Error(err)
  326. panic(err)
  327. }
  328. if cnt != 1 {
  329. err = errors.New("update not returned 1")
  330. t.Error(err)
  331. panic(err)
  332. }
  333. has, err = testEngine.Id(userID).
  334. And("username = ?", updatedUser.Username).
  335. And("height IS NULL").
  336. And("departname IS NULL").
  337. And("is_man IS NULL").
  338. And("created IS NULL").
  339. And("detail_id = ?", 0).
  340. Get(&Userinfo{})
  341. if err != nil {
  342. t.Error(err)
  343. panic(err)
  344. }
  345. if !has {
  346. err = errors.New("cannot update with null properly")
  347. t.Error(err)
  348. panic(err)
  349. }
  350. cnt, err = testEngine.Id(userID).Delete(&Userinfo{})
  351. if err != nil {
  352. t.Error(err)
  353. panic(err)
  354. }
  355. if cnt != 1 {
  356. err = errors.New("delete not returned 1")
  357. t.Error(err)
  358. panic(err)
  359. }
  360. }
  361. err = testEngine.StoreEngine("Innodb").Sync2(&Article{})
  362. if err != nil {
  363. t.Error(err)
  364. panic(err)
  365. }
  366. defer func() {
  367. err = testEngine.DropTables(&Article{})
  368. if err != nil {
  369. t.Error(err)
  370. panic(err)
  371. }
  372. }()
  373. a := &Article{0, "1", "2", "3", "4", "5", 2}
  374. cnt, err = testEngine.Insert(a)
  375. if err != nil {
  376. t.Error(err)
  377. panic(err)
  378. }
  379. if cnt != 1 {
  380. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  381. t.Error(err)
  382. panic(err)
  383. }
  384. if a.Id == 0 {
  385. err = errors.New("insert returned id is 0")
  386. t.Error(err)
  387. panic(err)
  388. }
  389. cnt, err = testEngine.Id(a.Id).Update(&Article{Name: "6"})
  390. if err != nil {
  391. t.Error(err)
  392. panic(err)
  393. }
  394. if cnt != 1 {
  395. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  396. t.Error(err)
  397. panic(err)
  398. return
  399. }
  400. var s = "test"
  401. col1 := &UpdateAllCols{Ptr: &s}
  402. err = testEngine.Sync(col1)
  403. if err != nil {
  404. t.Error(err)
  405. panic(err)
  406. }
  407. _, err = testEngine.Insert(col1)
  408. if err != nil {
  409. t.Error(err)
  410. panic(err)
  411. }
  412. col2 := &UpdateAllCols{col1.Id, true, "", nil}
  413. _, err = testEngine.Id(col2.Id).AllCols().Update(col2)
  414. if err != nil {
  415. t.Error(err)
  416. panic(err)
  417. }
  418. col3 := &UpdateAllCols{}
  419. has, err = testEngine.Id(col2.Id).Get(col3)
  420. if err != nil {
  421. t.Error(err)
  422. panic(err)
  423. }
  424. if !has {
  425. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  426. t.Error(err)
  427. panic(err)
  428. return
  429. }
  430. if *col2 != *col3 {
  431. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  432. t.Error(err)
  433. panic(err)
  434. return
  435. }
  436. {
  437. col1 := &UpdateMustCols{}
  438. err = testEngine.Sync(col1)
  439. if err != nil {
  440. t.Error(err)
  441. panic(err)
  442. }
  443. _, err = testEngine.Insert(col1)
  444. if err != nil {
  445. t.Error(err)
  446. panic(err)
  447. }
  448. col2 := &UpdateMustCols{col1.Id, true, ""}
  449. boolStr := testEngine.ColumnMapper.Obj2Table("Bool")
  450. stringStr := testEngine.ColumnMapper.Obj2Table("String")
  451. _, err = testEngine.Id(col2.Id).MustCols(boolStr, stringStr).Update(col2)
  452. if err != nil {
  453. t.Error(err)
  454. panic(err)
  455. }
  456. col3 := &UpdateMustCols{}
  457. has, err := testEngine.Id(col2.Id).Get(col3)
  458. if err != nil {
  459. t.Error(err)
  460. panic(err)
  461. }
  462. if !has {
  463. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  464. t.Error(err)
  465. panic(err)
  466. return
  467. }
  468. if *col2 != *col3 {
  469. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  470. t.Error(err)
  471. panic(err)
  472. return
  473. }
  474. }
  475. {
  476. col1 := &UpdateIncr{}
  477. err = testEngine.Sync(col1)
  478. if err != nil {
  479. t.Error(err)
  480. panic(err)
  481. }
  482. _, err = testEngine.Insert(col1)
  483. if err != nil {
  484. t.Error(err)
  485. panic(err)
  486. }
  487. cnt, err := testEngine.Id(col1.Id).Incr(testEngine.ColumnMapper.Obj2Table("Cnt")).Update(col1)
  488. if err != nil {
  489. t.Error(err)
  490. panic(err)
  491. }
  492. if cnt != 1 {
  493. err = errors.New("update incr failed")
  494. t.Error(err)
  495. panic(err)
  496. }
  497. newCol := new(UpdateIncr)
  498. has, err := testEngine.Id(col1.Id).Get(newCol)
  499. if err != nil {
  500. t.Error(err)
  501. panic(err)
  502. }
  503. if !has {
  504. err = errors.New("has incr failed")
  505. t.Error(err)
  506. panic(err)
  507. }
  508. if 1 != newCol.Cnt {
  509. err = fmt.Errorf("incr failed %v %v %v", newCol.Cnt, newCol, col1)
  510. t.Error(err)
  511. panic(err)
  512. }
  513. }
  514. }
  515. type UpdatedUpdate struct {
  516. Id int64
  517. Updated time.Time `xorm:"updated"`
  518. }
  519. type UpdatedUpdate2 struct {
  520. Id int64
  521. Updated int64 `xorm:"updated"`
  522. }
  523. type UpdatedUpdate3 struct {
  524. Id int64
  525. Updated int `xorm:"updated bigint"`
  526. }
  527. type UpdatedUpdate4 struct {
  528. Id int64
  529. Updated int `xorm:"updated"`
  530. }
  531. type UpdatedUpdate5 struct {
  532. Id int64
  533. Updated time.Time `xorm:"updated bigint"`
  534. }
  535. func TestUpdateUpdated(t *testing.T) {
  536. assert.NoError(t, prepareEngine())
  537. di := new(UpdatedUpdate)
  538. err := testEngine.Sync2(di)
  539. if err != nil {
  540. t.Fatal(err)
  541. }
  542. _, err = testEngine.Insert(&UpdatedUpdate{})
  543. if err != nil {
  544. t.Fatal(err)
  545. }
  546. ci := &UpdatedUpdate{}
  547. _, err = testEngine.Id(1).Update(ci)
  548. if err != nil {
  549. t.Fatal(err)
  550. }
  551. has, err := testEngine.Id(1).Get(di)
  552. if err != nil {
  553. t.Fatal(err)
  554. }
  555. if !has {
  556. t.Fatal(ErrNotExist)
  557. }
  558. if ci.Updated.Unix() != di.Updated.Unix() {
  559. t.Fatal("should equal:", ci, di)
  560. }
  561. fmt.Println("ci:", ci, "di:", di)
  562. di2 := new(UpdatedUpdate2)
  563. err = testEngine.Sync2(di2)
  564. if err != nil {
  565. t.Fatal(err)
  566. }
  567. _, err = testEngine.Insert(&UpdatedUpdate2{})
  568. if err != nil {
  569. t.Fatal(err)
  570. }
  571. ci2 := &UpdatedUpdate2{}
  572. _, err = testEngine.Id(1).Update(ci2)
  573. if err != nil {
  574. t.Fatal(err)
  575. }
  576. has, err = testEngine.Id(1).Get(di2)
  577. if err != nil {
  578. t.Fatal(err)
  579. }
  580. if !has {
  581. t.Fatal(ErrNotExist)
  582. }
  583. if ci2.Updated != di2.Updated {
  584. t.Fatal("should equal:", ci2, di2)
  585. }
  586. fmt.Println("ci2:", ci2, "di2:", di2)
  587. di3 := new(UpdatedUpdate3)
  588. err = testEngine.Sync2(di3)
  589. if err != nil {
  590. t.Fatal(err)
  591. }
  592. _, err = testEngine.Insert(&UpdatedUpdate3{})
  593. if err != nil {
  594. t.Fatal(err)
  595. }
  596. ci3 := &UpdatedUpdate3{}
  597. _, err = testEngine.Id(1).Update(ci3)
  598. if err != nil {
  599. t.Fatal(err)
  600. }
  601. has, err = testEngine.Id(1).Get(di3)
  602. if err != nil {
  603. t.Fatal(err)
  604. }
  605. if !has {
  606. t.Fatal(ErrNotExist)
  607. }
  608. if ci3.Updated != di3.Updated {
  609. t.Fatal("should equal:", ci3, di3)
  610. }
  611. fmt.Println("ci3:", ci3, "di3:", di3)
  612. di4 := new(UpdatedUpdate4)
  613. err = testEngine.Sync2(di4)
  614. if err != nil {
  615. t.Fatal(err)
  616. }
  617. _, err = testEngine.Insert(&UpdatedUpdate4{})
  618. if err != nil {
  619. t.Fatal(err)
  620. }
  621. ci4 := &UpdatedUpdate4{}
  622. _, err = testEngine.Id(1).Update(ci4)
  623. if err != nil {
  624. t.Fatal(err)
  625. }
  626. has, err = testEngine.Id(1).Get(di4)
  627. if err != nil {
  628. t.Fatal(err)
  629. }
  630. if !has {
  631. t.Fatal(ErrNotExist)
  632. }
  633. if ci4.Updated != di4.Updated {
  634. t.Fatal("should equal:", ci4, di4)
  635. }
  636. fmt.Println("ci4:", ci4, "di4:", di4)
  637. di5 := new(UpdatedUpdate5)
  638. err = testEngine.Sync2(di5)
  639. if err != nil {
  640. t.Fatal(err)
  641. }
  642. _, err = testEngine.Insert(&UpdatedUpdate5{})
  643. if err != nil {
  644. t.Fatal(err)
  645. }
  646. ci5 := &UpdatedUpdate5{}
  647. _, err = testEngine.Id(1).Update(ci5)
  648. if err != nil {
  649. t.Fatal(err)
  650. }
  651. has, err = testEngine.Id(1).Get(di5)
  652. if err != nil {
  653. t.Fatal(err)
  654. }
  655. if !has {
  656. t.Fatal(ErrNotExist)
  657. }
  658. if ci5.Updated.Unix() != di5.Updated.Unix() {
  659. t.Fatal("should equal:", ci5, di5)
  660. }
  661. fmt.Println("ci5:", ci5, "di5:", di5)
  662. }
  663. func TestUpdateSameMapper(t *testing.T) {
  664. assert.NoError(t, prepareEngine())
  665. oldMapper := testEngine.ColumnMapper
  666. testEngine.unMapType(rValue(new(Userinfo)).Type())
  667. testEngine.unMapType(rValue(new(Condi)).Type())
  668. testEngine.unMapType(rValue(new(Article)).Type())
  669. testEngine.unMapType(rValue(new(UpdateAllCols)).Type())
  670. testEngine.unMapType(rValue(new(UpdateMustCols)).Type())
  671. testEngine.unMapType(rValue(new(UpdateIncr)).Type())
  672. testEngine.SetMapper(core.SameMapper{})
  673. defer func() {
  674. testEngine.unMapType(rValue(new(Userinfo)).Type())
  675. testEngine.unMapType(rValue(new(Condi)).Type())
  676. testEngine.unMapType(rValue(new(Article)).Type())
  677. testEngine.unMapType(rValue(new(UpdateAllCols)).Type())
  678. testEngine.unMapType(rValue(new(UpdateMustCols)).Type())
  679. testEngine.unMapType(rValue(new(UpdateIncr)).Type())
  680. testEngine.SetMapper(oldMapper)
  681. }()
  682. assertSync(t, new(Userinfo))
  683. _, err := testEngine.Insert(&Userinfo{
  684. Username: "user1",
  685. })
  686. assert.NoError(t, err)
  687. var ori Userinfo
  688. has, err := testEngine.Get(&ori)
  689. if err != nil {
  690. t.Error(err)
  691. panic(err)
  692. }
  693. if !has {
  694. t.Error(errors.New("not exist"))
  695. panic(errors.New("not exist"))
  696. }
  697. // update by id
  698. user := Userinfo{Username: "xxx", Height: 1.2}
  699. cnt, err := testEngine.Id(ori.Uid).Update(&user)
  700. if err != nil {
  701. t.Error(err)
  702. panic(err)
  703. }
  704. if cnt != 1 {
  705. err = errors.New("update not returned 1")
  706. t.Error(err)
  707. panic(err)
  708. return
  709. }
  710. condi := Condi{"Username": "zzz", "Departname": ""}
  711. cnt, err = testEngine.Table(&user).Id(ori.Uid).Update(&condi)
  712. if err != nil {
  713. t.Error(err)
  714. panic(err)
  715. }
  716. if cnt != 1 {
  717. err = errors.New("update not returned 1")
  718. t.Error(err)
  719. panic(err)
  720. return
  721. }
  722. cnt, err = testEngine.Update(&Userinfo{Username: "yyy"}, &user)
  723. if err != nil {
  724. t.Error(err)
  725. panic(err)
  726. }
  727. total, err := testEngine.Count(&user)
  728. if err != nil {
  729. t.Error(err)
  730. panic(err)
  731. }
  732. if cnt != total {
  733. err = errors.New("insert not returned 1")
  734. t.Error(err)
  735. panic(err)
  736. return
  737. }
  738. err = testEngine.Sync(&Article{})
  739. if err != nil {
  740. t.Error(err)
  741. panic(err)
  742. }
  743. defer func() {
  744. err = testEngine.DropTables(&Article{})
  745. if err != nil {
  746. t.Error(err)
  747. panic(err)
  748. }
  749. }()
  750. a := &Article{0, "1", "2", "3", "4", "5", 2}
  751. cnt, err = testEngine.Insert(a)
  752. if err != nil {
  753. t.Error(err)
  754. panic(err)
  755. }
  756. if cnt != 1 {
  757. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  758. t.Error(err)
  759. panic(err)
  760. }
  761. if a.Id == 0 {
  762. err = errors.New("insert returned id is 0")
  763. t.Error(err)
  764. panic(err)
  765. }
  766. cnt, err = testEngine.Id(a.Id).Update(&Article{Name: "6"})
  767. if err != nil {
  768. t.Error(err)
  769. panic(err)
  770. }
  771. if cnt != 1 {
  772. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  773. t.Error(err)
  774. panic(err)
  775. return
  776. }
  777. col1 := &UpdateAllCols{}
  778. err = testEngine.Sync(col1)
  779. if err != nil {
  780. t.Error(err)
  781. panic(err)
  782. }
  783. _, err = testEngine.Insert(col1)
  784. if err != nil {
  785. t.Error(err)
  786. panic(err)
  787. }
  788. col2 := &UpdateAllCols{col1.Id, true, "", nil}
  789. _, err = testEngine.Id(col2.Id).AllCols().Update(col2)
  790. if err != nil {
  791. t.Error(err)
  792. panic(err)
  793. }
  794. col3 := &UpdateAllCols{}
  795. has, err = testEngine.Id(col2.Id).Get(col3)
  796. if err != nil {
  797. t.Error(err)
  798. panic(err)
  799. }
  800. if !has {
  801. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  802. t.Error(err)
  803. panic(err)
  804. return
  805. }
  806. if *col2 != *col3 {
  807. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  808. t.Error(err)
  809. panic(err)
  810. return
  811. }
  812. {
  813. col1 := &UpdateMustCols{}
  814. err = testEngine.Sync(col1)
  815. if err != nil {
  816. t.Error(err)
  817. panic(err)
  818. }
  819. _, err = testEngine.Insert(col1)
  820. if err != nil {
  821. t.Error(err)
  822. panic(err)
  823. }
  824. col2 := &UpdateMustCols{col1.Id, true, ""}
  825. boolStr := testEngine.ColumnMapper.Obj2Table("Bool")
  826. stringStr := testEngine.ColumnMapper.Obj2Table("String")
  827. _, err = testEngine.Id(col2.Id).MustCols(boolStr, stringStr).Update(col2)
  828. if err != nil {
  829. t.Error(err)
  830. panic(err)
  831. }
  832. col3 := &UpdateMustCols{}
  833. has, err := testEngine.Id(col2.Id).Get(col3)
  834. if err != nil {
  835. t.Error(err)
  836. panic(err)
  837. }
  838. if !has {
  839. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  840. t.Error(err)
  841. panic(err)
  842. return
  843. }
  844. if *col2 != *col3 {
  845. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  846. t.Error(err)
  847. panic(err)
  848. return
  849. }
  850. }
  851. {
  852. col1 := &UpdateIncr{}
  853. err = testEngine.Sync(col1)
  854. if err != nil {
  855. t.Error(err)
  856. panic(err)
  857. }
  858. _, err = testEngine.Insert(col1)
  859. if err != nil {
  860. t.Error(err)
  861. panic(err)
  862. }
  863. cnt, err := testEngine.Id(col1.Id).Incr("`Cnt`").Update(col1)
  864. if err != nil {
  865. t.Error(err)
  866. panic(err)
  867. }
  868. if cnt != 1 {
  869. err = errors.New("update incr failed")
  870. t.Error(err)
  871. panic(err)
  872. }
  873. newCol := new(UpdateIncr)
  874. has, err := testEngine.Id(col1.Id).Get(newCol)
  875. if err != nil {
  876. t.Error(err)
  877. panic(err)
  878. }
  879. if !has {
  880. err = errors.New("has incr failed")
  881. t.Error(err)
  882. panic(err)
  883. }
  884. if 1 != newCol.Cnt {
  885. err = errors.New("incr failed")
  886. t.Error(err)
  887. panic(err)
  888. }
  889. }
  890. }
  891. func TestUseBool(t *testing.T) {
  892. assert.NoError(t, prepareEngine())
  893. assertSync(t, new(Userinfo))
  894. cnt1, err := testEngine.Count(&Userinfo{})
  895. if err != nil {
  896. t.Error(err)
  897. panic(err)
  898. }
  899. users := make([]Userinfo, 0)
  900. err = testEngine.Find(&users)
  901. if err != nil {
  902. t.Error(err)
  903. panic(err)
  904. }
  905. var fNumber int64
  906. for _, u := range users {
  907. if u.IsMan == false {
  908. fNumber += 1
  909. }
  910. }
  911. cnt2, err := testEngine.UseBool().Update(&Userinfo{IsMan: true})
  912. if err != nil {
  913. t.Error(err)
  914. panic(err)
  915. }
  916. if fNumber != cnt2 {
  917. fmt.Println("cnt1", cnt1, "fNumber", fNumber, "cnt2", cnt2)
  918. /*err = errors.New("Updated number is not corrected.")
  919. t.Error(err)
  920. panic(err)*/
  921. }
  922. _, err = testEngine.Update(&Userinfo{IsMan: true})
  923. if err == nil {
  924. err = errors.New("error condition")
  925. t.Error(err)
  926. panic(err)
  927. }
  928. }
  929. func TestBool(t *testing.T) {
  930. assert.NoError(t, prepareEngine())
  931. assertSync(t, new(Userinfo))
  932. _, err := testEngine.UseBool().Update(&Userinfo{IsMan: true})
  933. if err != nil {
  934. t.Error(err)
  935. panic(err)
  936. }
  937. users := make([]Userinfo, 0)
  938. err = testEngine.Find(&users)
  939. if err != nil {
  940. t.Error(err)
  941. panic(err)
  942. }
  943. for _, user := range users {
  944. if !user.IsMan {
  945. err = errors.New("update bool or find bool error")
  946. t.Error(err)
  947. panic(err)
  948. }
  949. }
  950. _, err = testEngine.UseBool().Update(&Userinfo{IsMan: false})
  951. if err != nil {
  952. t.Error(err)
  953. panic(err)
  954. }
  955. users = make([]Userinfo, 0)
  956. err = testEngine.Find(&users)
  957. if err != nil {
  958. t.Error(err)
  959. panic(err)
  960. }
  961. for _, user := range users {
  962. if user.IsMan {
  963. err = errors.New("update bool or find bool error")
  964. t.Error(err)
  965. panic(err)
  966. }
  967. }
  968. }