session_update_test.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248
  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 UpdateTable2 struct {
  38. Id int64
  39. Name string
  40. Age int
  41. }
  42. assert.NoError(t, testEngine.Sync2(new(UpdateTable2)))
  43. var tb = UpdateTable2{
  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(&UpdateTable2{
  56. Age: 30,
  57. })
  58. assert.NoError(t, err)
  59. assert.EqualValues(t, 1, cnt)
  60. var uts []UpdateTable2
  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 EngineInterface) 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.Dialect().DriverName() != "mysql" && testEngine.Dialect().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. Name string
  218. }
  219. type Article struct {
  220. Id int32 `xorm:"pk INT autoincr"`
  221. Name string `xorm:"VARCHAR(45)"`
  222. Img string `xorm:"VARCHAR(100)"`
  223. Aside string `xorm:"VARCHAR(200)"`
  224. Desc string `xorm:"VARCHAR(200)"`
  225. Content string `xorm:"TEXT"`
  226. Status int8 `xorm:"TINYINT(4)"`
  227. }
  228. func TestUpdateMap2(t *testing.T) {
  229. assert.NoError(t, prepareEngine())
  230. assertSync(t, new(UpdateMustCols))
  231. _, err := testEngine.Table("update_must_cols").Where("id =?", 1).Update(map[string]interface{}{
  232. "bool": true,
  233. })
  234. if err != nil {
  235. t.Error(err)
  236. panic(err)
  237. }
  238. }
  239. func TestUpdate1(t *testing.T) {
  240. assert.NoError(t, prepareEngine())
  241. assertSync(t, new(Userinfo))
  242. _, err := testEngine.Insert(&Userinfo{
  243. Username: "user1",
  244. })
  245. var ori Userinfo
  246. has, err := testEngine.Get(&ori)
  247. if err != nil {
  248. t.Error(err)
  249. panic(err)
  250. }
  251. if !has {
  252. t.Error(errors.New("not exist"))
  253. panic(errors.New("not exist"))
  254. }
  255. // update by id
  256. user := Userinfo{Username: "xxx", Height: 1.2}
  257. cnt, err := testEngine.ID(ori.Uid).Update(&user)
  258. if err != nil {
  259. t.Error(err)
  260. panic(err)
  261. }
  262. if cnt != 1 {
  263. err = errors.New("update not returned 1")
  264. t.Error(err)
  265. panic(err)
  266. return
  267. }
  268. condi := Condi{"username": "zzz", "departname": ""}
  269. cnt, err = testEngine.Table(&user).ID(ori.Uid).Update(&condi)
  270. if err != nil {
  271. t.Error(err)
  272. panic(err)
  273. }
  274. if cnt != 1 {
  275. err = errors.New("update not returned 1")
  276. t.Error(err)
  277. panic(err)
  278. return
  279. }
  280. cnt, err = testEngine.Update(&Userinfo{Username: "yyy"}, &user)
  281. if err != nil {
  282. t.Error(err)
  283. panic(err)
  284. }
  285. total, err := testEngine.Count(&user)
  286. if err != nil {
  287. t.Error(err)
  288. panic(err)
  289. }
  290. if cnt != total {
  291. err = errors.New("insert not returned 1")
  292. t.Error(err)
  293. panic(err)
  294. return
  295. }
  296. // nullable update
  297. {
  298. user := &Userinfo{Username: "not null data", Height: 180.5}
  299. _, err := testEngine.Insert(user)
  300. if err != nil {
  301. t.Error(err)
  302. panic(err)
  303. }
  304. userID := user.Uid
  305. has, err := testEngine.ID(userID).
  306. And("username = ?", user.Username).
  307. And("height = ?", user.Height).
  308. And("departname = ?", "").
  309. And("detail_id = ?", 0).
  310. And("is_man = ?", 0).
  311. Get(&Userinfo{})
  312. if err != nil {
  313. t.Error(err)
  314. panic(err)
  315. }
  316. if !has {
  317. err = errors.New("cannot insert properly")
  318. t.Error(err)
  319. panic(err)
  320. }
  321. updatedUser := &Userinfo{Username: "null data"}
  322. cnt, err = testEngine.ID(userID).
  323. Nullable("height", "departname", "is_man", "created").
  324. Update(updatedUser)
  325. if err != nil {
  326. t.Error(err)
  327. panic(err)
  328. }
  329. if cnt != 1 {
  330. err = errors.New("update not returned 1")
  331. t.Error(err)
  332. panic(err)
  333. }
  334. has, err = testEngine.ID(userID).
  335. And("username = ?", updatedUser.Username).
  336. And("height IS NULL").
  337. And("departname IS NULL").
  338. And("is_man IS NULL").
  339. And("created IS NULL").
  340. And("detail_id = ?", 0).
  341. Get(&Userinfo{})
  342. if err != nil {
  343. t.Error(err)
  344. panic(err)
  345. }
  346. if !has {
  347. err = errors.New("cannot update with null properly")
  348. t.Error(err)
  349. panic(err)
  350. }
  351. cnt, err = testEngine.ID(userID).Delete(&Userinfo{})
  352. if err != nil {
  353. t.Error(err)
  354. panic(err)
  355. }
  356. if cnt != 1 {
  357. err = errors.New("delete not returned 1")
  358. t.Error(err)
  359. panic(err)
  360. }
  361. }
  362. err = testEngine.StoreEngine("Innodb").Sync2(&Article{})
  363. if err != nil {
  364. t.Error(err)
  365. panic(err)
  366. }
  367. defer func() {
  368. err = testEngine.DropTables(&Article{})
  369. if err != nil {
  370. t.Error(err)
  371. panic(err)
  372. }
  373. }()
  374. a := &Article{0, "1", "2", "3", "4", "5", 2}
  375. cnt, err = testEngine.Insert(a)
  376. if err != nil {
  377. t.Error(err)
  378. panic(err)
  379. }
  380. if cnt != 1 {
  381. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  382. t.Error(err)
  383. panic(err)
  384. }
  385. if a.Id == 0 {
  386. err = errors.New("insert returned id is 0")
  387. t.Error(err)
  388. panic(err)
  389. }
  390. cnt, err = testEngine.ID(a.Id).Update(&Article{Name: "6"})
  391. if err != nil {
  392. t.Error(err)
  393. panic(err)
  394. }
  395. if cnt != 1 {
  396. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  397. t.Error(err)
  398. panic(err)
  399. return
  400. }
  401. var s = "test"
  402. col1 := &UpdateAllCols{Ptr: &s}
  403. err = testEngine.Sync(col1)
  404. if err != nil {
  405. t.Error(err)
  406. panic(err)
  407. }
  408. _, err = testEngine.Insert(col1)
  409. if err != nil {
  410. t.Error(err)
  411. panic(err)
  412. }
  413. col2 := &UpdateAllCols{col1.Id, true, "", nil}
  414. _, err = testEngine.ID(col2.Id).AllCols().Update(col2)
  415. if err != nil {
  416. t.Error(err)
  417. panic(err)
  418. }
  419. col3 := &UpdateAllCols{}
  420. has, err = testEngine.ID(col2.Id).Get(col3)
  421. if err != nil {
  422. t.Error(err)
  423. panic(err)
  424. }
  425. if !has {
  426. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  427. t.Error(err)
  428. panic(err)
  429. return
  430. }
  431. if *col2 != *col3 {
  432. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  433. t.Error(err)
  434. panic(err)
  435. return
  436. }
  437. {
  438. col1 := &UpdateMustCols{}
  439. err = testEngine.Sync(col1)
  440. if err != nil {
  441. t.Error(err)
  442. panic(err)
  443. }
  444. _, err = testEngine.Insert(col1)
  445. if err != nil {
  446. t.Error(err)
  447. panic(err)
  448. }
  449. col2 := &UpdateMustCols{col1.Id, true, ""}
  450. boolStr := testEngine.GetColumnMapper().Obj2Table("Bool")
  451. stringStr := testEngine.GetColumnMapper().Obj2Table("String")
  452. _, err = testEngine.ID(col2.Id).MustCols(boolStr, stringStr).Update(col2)
  453. if err != nil {
  454. t.Error(err)
  455. panic(err)
  456. }
  457. col3 := &UpdateMustCols{}
  458. has, err := testEngine.ID(col2.Id).Get(col3)
  459. if err != nil {
  460. t.Error(err)
  461. panic(err)
  462. }
  463. if !has {
  464. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  465. t.Error(err)
  466. panic(err)
  467. return
  468. }
  469. if *col2 != *col3 {
  470. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  471. t.Error(err)
  472. panic(err)
  473. return
  474. }
  475. }
  476. }
  477. func TestUpdateIncrDecr(t *testing.T) {
  478. assert.NoError(t, prepareEngine())
  479. col1 := &UpdateIncr{
  480. Name: "test",
  481. }
  482. assert.NoError(t, testEngine.Sync(col1))
  483. _, err := testEngine.Insert(col1)
  484. assert.NoError(t, err)
  485. colName := testEngine.GetColumnMapper().Obj2Table("Cnt")
  486. cnt, err := testEngine.ID(col1.Id).Incr(colName).Update(col1)
  487. assert.NoError(t, err)
  488. assert.EqualValues(t, 1, cnt)
  489. newCol := new(UpdateIncr)
  490. has, err := testEngine.ID(col1.Id).Get(newCol)
  491. assert.NoError(t, err)
  492. assert.True(t, has)
  493. assert.EqualValues(t, 1, newCol.Cnt)
  494. cnt, err = testEngine.ID(col1.Id).Decr(colName).Update(col1)
  495. assert.NoError(t, err)
  496. assert.EqualValues(t, 1, cnt)
  497. newCol = new(UpdateIncr)
  498. has, err = testEngine.ID(col1.Id).Get(newCol)
  499. assert.NoError(t, err)
  500. assert.True(t, has)
  501. assert.EqualValues(t, 0, newCol.Cnt)
  502. cnt, err = testEngine.ID(col1.Id).Cols(colName).Incr(colName).Update(col1)
  503. assert.NoError(t, err)
  504. assert.EqualValues(t, 1, cnt)
  505. }
  506. type UpdatedUpdate struct {
  507. Id int64
  508. Updated time.Time `xorm:"updated"`
  509. }
  510. type UpdatedUpdate2 struct {
  511. Id int64
  512. Updated int64 `xorm:"updated"`
  513. }
  514. type UpdatedUpdate3 struct {
  515. Id int64
  516. Updated int `xorm:"updated bigint"`
  517. }
  518. type UpdatedUpdate4 struct {
  519. Id int64
  520. Updated int `xorm:"updated"`
  521. }
  522. type UpdatedUpdate5 struct {
  523. Id int64
  524. Updated time.Time `xorm:"updated bigint"`
  525. }
  526. func TestUpdateUpdated(t *testing.T) {
  527. assert.NoError(t, prepareEngine())
  528. di := new(UpdatedUpdate)
  529. err := testEngine.Sync2(di)
  530. if err != nil {
  531. t.Fatal(err)
  532. }
  533. _, err = testEngine.Insert(&UpdatedUpdate{})
  534. if err != nil {
  535. t.Fatal(err)
  536. }
  537. ci := &UpdatedUpdate{}
  538. _, err = testEngine.ID(1).Update(ci)
  539. if err != nil {
  540. t.Fatal(err)
  541. }
  542. has, err := testEngine.ID(1).Get(di)
  543. if err != nil {
  544. t.Fatal(err)
  545. }
  546. if !has {
  547. t.Fatal(ErrNotExist)
  548. }
  549. if ci.Updated.Unix() != di.Updated.Unix() {
  550. t.Fatal("should equal:", ci, di)
  551. }
  552. fmt.Println("ci:", ci, "di:", di)
  553. di2 := new(UpdatedUpdate2)
  554. err = testEngine.Sync2(di2)
  555. assert.NoError(t, err)
  556. now := time.Now()
  557. var di20 UpdatedUpdate2
  558. cnt, err := testEngine.Insert(&di20)
  559. assert.NoError(t, err)
  560. assert.EqualValues(t, 1, cnt)
  561. assert.True(t, now.Unix() <= di20.Updated)
  562. var di21 UpdatedUpdate2
  563. has, err = testEngine.ID(di20.Id).Get(&di21)
  564. assert.NoError(t, err)
  565. assert.True(t, has)
  566. assert.EqualValues(t, di20.Updated, di21.Updated)
  567. ci2 := &UpdatedUpdate2{}
  568. _, err = testEngine.ID(1).Update(ci2)
  569. assert.NoError(t, err)
  570. has, err = testEngine.ID(1).Get(di2)
  571. assert.NoError(t, err)
  572. assert.True(t, has)
  573. assert.EqualValues(t, ci2.Updated, di2.Updated)
  574. assert.True(t, ci2.Updated >= di21.Updated)
  575. di3 := new(UpdatedUpdate3)
  576. err = testEngine.Sync2(di3)
  577. if err != nil {
  578. t.Fatal(err)
  579. }
  580. _, err = testEngine.Insert(&UpdatedUpdate3{})
  581. if err != nil {
  582. t.Fatal(err)
  583. }
  584. ci3 := &UpdatedUpdate3{}
  585. _, err = testEngine.ID(1).Update(ci3)
  586. if err != nil {
  587. t.Fatal(err)
  588. }
  589. has, err = testEngine.ID(1).Get(di3)
  590. if err != nil {
  591. t.Fatal(err)
  592. }
  593. if !has {
  594. t.Fatal(ErrNotExist)
  595. }
  596. if ci3.Updated != di3.Updated {
  597. t.Fatal("should equal:", ci3, di3)
  598. }
  599. fmt.Println("ci3:", ci3, "di3:", di3)
  600. di4 := new(UpdatedUpdate4)
  601. err = testEngine.Sync2(di4)
  602. if err != nil {
  603. t.Fatal(err)
  604. }
  605. _, err = testEngine.Insert(&UpdatedUpdate4{})
  606. if err != nil {
  607. t.Fatal(err)
  608. }
  609. ci4 := &UpdatedUpdate4{}
  610. _, err = testEngine.ID(1).Update(ci4)
  611. if err != nil {
  612. t.Fatal(err)
  613. }
  614. has, err = testEngine.ID(1).Get(di4)
  615. if err != nil {
  616. t.Fatal(err)
  617. }
  618. if !has {
  619. t.Fatal(ErrNotExist)
  620. }
  621. if ci4.Updated != di4.Updated {
  622. t.Fatal("should equal:", ci4, di4)
  623. }
  624. fmt.Println("ci4:", ci4, "di4:", di4)
  625. di5 := new(UpdatedUpdate5)
  626. err = testEngine.Sync2(di5)
  627. if err != nil {
  628. t.Fatal(err)
  629. }
  630. _, err = testEngine.Insert(&UpdatedUpdate5{})
  631. if err != nil {
  632. t.Fatal(err)
  633. }
  634. ci5 := &UpdatedUpdate5{}
  635. _, err = testEngine.ID(1).Update(ci5)
  636. if err != nil {
  637. t.Fatal(err)
  638. }
  639. has, err = testEngine.ID(1).Get(di5)
  640. if err != nil {
  641. t.Fatal(err)
  642. }
  643. if !has {
  644. t.Fatal(ErrNotExist)
  645. }
  646. if ci5.Updated.Unix() != di5.Updated.Unix() {
  647. t.Fatal("should equal:", ci5, di5)
  648. }
  649. fmt.Println("ci5:", ci5, "di5:", di5)
  650. }
  651. func TestUpdateSameMapper(t *testing.T) {
  652. assert.NoError(t, prepareEngine())
  653. oldMapper := testEngine.GetColumnMapper()
  654. testEngine.UnMapType(rValue(new(Userinfo)).Type())
  655. testEngine.UnMapType(rValue(new(Condi)).Type())
  656. testEngine.UnMapType(rValue(new(Article)).Type())
  657. testEngine.UnMapType(rValue(new(UpdateAllCols)).Type())
  658. testEngine.UnMapType(rValue(new(UpdateMustCols)).Type())
  659. testEngine.UnMapType(rValue(new(UpdateIncr)).Type())
  660. testEngine.SetMapper(core.SameMapper{})
  661. defer func() {
  662. testEngine.UnMapType(rValue(new(Userinfo)).Type())
  663. testEngine.UnMapType(rValue(new(Condi)).Type())
  664. testEngine.UnMapType(rValue(new(Article)).Type())
  665. testEngine.UnMapType(rValue(new(UpdateAllCols)).Type())
  666. testEngine.UnMapType(rValue(new(UpdateMustCols)).Type())
  667. testEngine.UnMapType(rValue(new(UpdateIncr)).Type())
  668. testEngine.SetMapper(oldMapper)
  669. }()
  670. assertSync(t, new(Userinfo))
  671. _, err := testEngine.Insert(&Userinfo{
  672. Username: "user1",
  673. })
  674. assert.NoError(t, err)
  675. var ori Userinfo
  676. has, err := testEngine.Get(&ori)
  677. if err != nil {
  678. t.Error(err)
  679. panic(err)
  680. }
  681. if !has {
  682. t.Error(errors.New("not exist"))
  683. panic(errors.New("not exist"))
  684. }
  685. // update by id
  686. user := Userinfo{Username: "xxx", Height: 1.2}
  687. cnt, err := testEngine.ID(ori.Uid).Update(&user)
  688. if err != nil {
  689. t.Error(err)
  690. panic(err)
  691. }
  692. if cnt != 1 {
  693. err = errors.New("update not returned 1")
  694. t.Error(err)
  695. panic(err)
  696. return
  697. }
  698. condi := Condi{"Username": "zzz", "Departname": ""}
  699. cnt, err = testEngine.Table(&user).ID(ori.Uid).Update(&condi)
  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. cnt, err = testEngine.Update(&Userinfo{Username: "yyy"}, &user)
  711. if err != nil {
  712. t.Error(err)
  713. panic(err)
  714. }
  715. total, err := testEngine.Count(&user)
  716. if err != nil {
  717. t.Error(err)
  718. panic(err)
  719. }
  720. if cnt != total {
  721. err = errors.New("insert not returned 1")
  722. t.Error(err)
  723. panic(err)
  724. return
  725. }
  726. err = testEngine.Sync(&Article{})
  727. if err != nil {
  728. t.Error(err)
  729. panic(err)
  730. }
  731. defer func() {
  732. err = testEngine.DropTables(&Article{})
  733. if err != nil {
  734. t.Error(err)
  735. panic(err)
  736. }
  737. }()
  738. a := &Article{0, "1", "2", "3", "4", "5", 2}
  739. cnt, err = testEngine.Insert(a)
  740. if err != nil {
  741. t.Error(err)
  742. panic(err)
  743. }
  744. if cnt != 1 {
  745. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  746. t.Error(err)
  747. panic(err)
  748. }
  749. if a.Id == 0 {
  750. err = errors.New("insert returned id is 0")
  751. t.Error(err)
  752. panic(err)
  753. }
  754. cnt, err = testEngine.ID(a.Id).Update(&Article{Name: "6"})
  755. if err != nil {
  756. t.Error(err)
  757. panic(err)
  758. }
  759. if cnt != 1 {
  760. err = errors.New(fmt.Sprintf("insert not returned 1 but %d", cnt))
  761. t.Error(err)
  762. panic(err)
  763. return
  764. }
  765. col1 := &UpdateAllCols{}
  766. err = testEngine.Sync(col1)
  767. if err != nil {
  768. t.Error(err)
  769. panic(err)
  770. }
  771. _, err = testEngine.Insert(col1)
  772. if err != nil {
  773. t.Error(err)
  774. panic(err)
  775. }
  776. col2 := &UpdateAllCols{col1.Id, true, "", nil}
  777. _, err = testEngine.ID(col2.Id).AllCols().Update(col2)
  778. if err != nil {
  779. t.Error(err)
  780. panic(err)
  781. }
  782. col3 := &UpdateAllCols{}
  783. has, err = testEngine.ID(col2.Id).Get(col3)
  784. if err != nil {
  785. t.Error(err)
  786. panic(err)
  787. }
  788. if !has {
  789. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  790. t.Error(err)
  791. panic(err)
  792. return
  793. }
  794. if *col2 != *col3 {
  795. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  796. t.Error(err)
  797. panic(err)
  798. return
  799. }
  800. {
  801. col1 := &UpdateMustCols{}
  802. err = testEngine.Sync(col1)
  803. if err != nil {
  804. t.Error(err)
  805. panic(err)
  806. }
  807. _, err = testEngine.Insert(col1)
  808. if err != nil {
  809. t.Error(err)
  810. panic(err)
  811. }
  812. col2 := &UpdateMustCols{col1.Id, true, ""}
  813. boolStr := testEngine.GetColumnMapper().Obj2Table("Bool")
  814. stringStr := testEngine.GetColumnMapper().Obj2Table("String")
  815. _, err = testEngine.ID(col2.Id).MustCols(boolStr, stringStr).Update(col2)
  816. if err != nil {
  817. t.Error(err)
  818. panic(err)
  819. }
  820. col3 := &UpdateMustCols{}
  821. has, err := testEngine.ID(col2.Id).Get(col3)
  822. if err != nil {
  823. t.Error(err)
  824. panic(err)
  825. }
  826. if !has {
  827. err = errors.New(fmt.Sprintf("cannot get id %d", col2.Id))
  828. t.Error(err)
  829. panic(err)
  830. return
  831. }
  832. if *col2 != *col3 {
  833. err = errors.New(fmt.Sprintf("col2 should eq col3"))
  834. t.Error(err)
  835. panic(err)
  836. return
  837. }
  838. }
  839. {
  840. col1 := &UpdateIncr{}
  841. err = testEngine.Sync(col1)
  842. if err != nil {
  843. t.Error(err)
  844. panic(err)
  845. }
  846. _, err = testEngine.Insert(col1)
  847. if err != nil {
  848. t.Error(err)
  849. panic(err)
  850. }
  851. cnt, err := testEngine.ID(col1.Id).Incr("`Cnt`").Update(col1)
  852. if err != nil {
  853. t.Error(err)
  854. panic(err)
  855. }
  856. if cnt != 1 {
  857. err = errors.New("update incr failed")
  858. t.Error(err)
  859. panic(err)
  860. }
  861. newCol := new(UpdateIncr)
  862. has, err := testEngine.ID(col1.Id).Get(newCol)
  863. if err != nil {
  864. t.Error(err)
  865. panic(err)
  866. }
  867. if !has {
  868. err = errors.New("has incr failed")
  869. t.Error(err)
  870. panic(err)
  871. }
  872. if 1 != newCol.Cnt {
  873. err = errors.New("incr failed")
  874. t.Error(err)
  875. panic(err)
  876. }
  877. }
  878. }
  879. func TestUseBool(t *testing.T) {
  880. assert.NoError(t, prepareEngine())
  881. assertSync(t, new(Userinfo))
  882. cnt1, err := testEngine.Count(&Userinfo{})
  883. if err != nil {
  884. t.Error(err)
  885. panic(err)
  886. }
  887. users := make([]Userinfo, 0)
  888. err = testEngine.Find(&users)
  889. if err != nil {
  890. t.Error(err)
  891. panic(err)
  892. }
  893. var fNumber int64
  894. for _, u := range users {
  895. if u.IsMan == false {
  896. fNumber += 1
  897. }
  898. }
  899. cnt2, err := testEngine.UseBool().Update(&Userinfo{IsMan: true})
  900. if err != nil {
  901. t.Error(err)
  902. panic(err)
  903. }
  904. if fNumber != cnt2 {
  905. fmt.Println("cnt1", cnt1, "fNumber", fNumber, "cnt2", cnt2)
  906. /*err = errors.New("Updated number is not corrected.")
  907. t.Error(err)
  908. panic(err)*/
  909. }
  910. _, err = testEngine.Update(&Userinfo{IsMan: true})
  911. if err == nil {
  912. err = errors.New("error condition")
  913. t.Error(err)
  914. panic(err)
  915. }
  916. }
  917. func TestBool(t *testing.T) {
  918. assert.NoError(t, prepareEngine())
  919. assertSync(t, new(Userinfo))
  920. _, err := testEngine.UseBool().Update(&Userinfo{IsMan: true})
  921. if err != nil {
  922. t.Error(err)
  923. panic(err)
  924. }
  925. users := make([]Userinfo, 0)
  926. err = testEngine.Find(&users)
  927. if err != nil {
  928. t.Error(err)
  929. panic(err)
  930. }
  931. for _, user := range users {
  932. if !user.IsMan {
  933. err = errors.New("update bool or find bool error")
  934. t.Error(err)
  935. panic(err)
  936. }
  937. }
  938. _, err = testEngine.UseBool().Update(&Userinfo{IsMan: false})
  939. if err != nil {
  940. t.Error(err)
  941. panic(err)
  942. }
  943. users = make([]Userinfo, 0)
  944. err = testEngine.Find(&users)
  945. if err != nil {
  946. t.Error(err)
  947. panic(err)
  948. }
  949. for _, user := range users {
  950. if user.IsMan {
  951. err = errors.New("update bool or find bool error")
  952. t.Error(err)
  953. panic(err)
  954. }
  955. }
  956. }
  957. func TestNoUpdate(t *testing.T) {
  958. assert.NoError(t, prepareEngine())
  959. type NoUpdate struct {
  960. Id int64
  961. Content string
  962. }
  963. assertSync(t, new(NoUpdate))
  964. cnt, err := testEngine.Insert(&NoUpdate{
  965. Content: "test",
  966. })
  967. assert.NoError(t, err)
  968. assert.EqualValues(t, 1, cnt)
  969. _, err = testEngine.ID(1).Update(&NoUpdate{})
  970. assert.Error(t, err)
  971. assert.EqualValues(t, "No content found to be updated", err.Error())
  972. }
  973. func TestNewUpdate(t *testing.T) {
  974. assert.NoError(t, prepareEngine())
  975. type TbUserInfo struct {
  976. Id int64 `xorm:"pk autoincr unique BIGINT" json:"id"`
  977. Phone string `xorm:"not null unique VARCHAR(20)" json:"phone"`
  978. UserName string `xorm:"VARCHAR(20)" json:"user_name"`
  979. Gender int `xorm:"default 0 INTEGER" json:"gender"`
  980. Pw string `xorm:"VARCHAR(100)" json:"pw"`
  981. Token string `xorm:"TEXT" json:"token"`
  982. Avatar string `xorm:"TEXT" json:"avatar"`
  983. Extras interface{} `xorm:"JSON" json:"extras"`
  984. Created time.Time `xorm:"DATETIME created"`
  985. Updated time.Time `xorm:"DATETIME updated"`
  986. Deleted time.Time `xorm:"DATETIME deleted"`
  987. }
  988. assertSync(t, new(TbUserInfo))
  989. targetUsr := TbUserInfo{Phone: "13126564922"}
  990. changeUsr := TbUserInfo{Token: "ABCDEFG"}
  991. af, err := testEngine.Update(&changeUsr, &targetUsr)
  992. assert.NoError(t, err)
  993. assert.EqualValues(t, 0, af)
  994. af, err = testEngine.Table(new(TbUserInfo)).Where("phone=?", 13126564922).Update(&changeUsr)
  995. assert.NoError(t, err)
  996. assert.EqualValues(t, 0, af)
  997. }
  998. func TestUpdateUpdate(t *testing.T) {
  999. assert.NoError(t, prepareEngine())
  1000. type PublicKeyUpdate struct {
  1001. Id int64
  1002. UpdatedUnix int64 `xorm:"updated"`
  1003. }
  1004. assertSync(t, new(PublicKeyUpdate))
  1005. cnt, err := testEngine.ID(1).Cols("updated_unix").Update(&PublicKeyUpdate{
  1006. UpdatedUnix: time.Now().Unix(),
  1007. })
  1008. assert.NoError(t, err)
  1009. assert.EqualValues(t, 0, cnt)
  1010. }
  1011. func TestCreatedUpdated2(t *testing.T) {
  1012. assert.NoError(t, prepareEngine())
  1013. type CreatedUpdatedStruct struct {
  1014. Id int64
  1015. Name string
  1016. CreateAt time.Time `xorm:"created" json:"create_at"`
  1017. UpdateAt time.Time `xorm:"updated" json:"update_at"`
  1018. }
  1019. assertSync(t, new(CreatedUpdatedStruct))
  1020. var s = CreatedUpdatedStruct{
  1021. Name: "test",
  1022. }
  1023. cnt, err := testEngine.Insert(&s)
  1024. assert.NoError(t, err)
  1025. assert.EqualValues(t, 1, cnt)
  1026. assert.EqualValues(t, s.UpdateAt.Unix(), s.CreateAt.Unix())
  1027. time.Sleep(time.Second)
  1028. var s1 = CreatedUpdatedStruct{
  1029. Name: "test1",
  1030. CreateAt: s.CreateAt,
  1031. UpdateAt: s.UpdateAt,
  1032. }
  1033. cnt, err = testEngine.ID(1).Update(&s1)
  1034. assert.NoError(t, err)
  1035. assert.EqualValues(t, 1, cnt)
  1036. assert.EqualValues(t, s.CreateAt.Unix(), s1.CreateAt.Unix())
  1037. assert.True(t, s1.UpdateAt.Unix() > s.UpdateAt.Unix())
  1038. var s2 CreatedUpdatedStruct
  1039. has, err := testEngine.ID(1).Get(&s2)
  1040. assert.NoError(t, err)
  1041. assert.True(t, has)
  1042. assert.EqualValues(t, s.CreateAt.Unix(), s2.CreateAt.Unix())
  1043. assert.True(t, s2.UpdateAt.Unix() > s.UpdateAt.Unix())
  1044. assert.True(t, s2.UpdateAt.Unix() > s2.CreateAt.Unix())
  1045. }
  1046. func TestUpdateMapCondition(t *testing.T) {
  1047. assert.NoError(t, prepareEngine())
  1048. type UpdateMapCondition struct {
  1049. Id int64
  1050. String string
  1051. }
  1052. assertSync(t, new(UpdateMapCondition))
  1053. var c = UpdateMapCondition{
  1054. String: "string",
  1055. }
  1056. _, err := testEngine.Insert(&c)
  1057. assert.NoError(t, err)
  1058. cnt, err := testEngine.Update(&UpdateMapCondition{
  1059. String: "string1",
  1060. }, map[string]interface{}{
  1061. "id": c.Id,
  1062. })
  1063. assert.NoError(t, err)
  1064. assert.EqualValues(t, 1, cnt)
  1065. var c2 UpdateMapCondition
  1066. has, err := testEngine.ID(c.Id).Get(&c2)
  1067. assert.NoError(t, err)
  1068. assert.True(t, has)
  1069. assert.EqualValues(t, "string1", c2.String)
  1070. }