session_pk_test.go 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119
  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. "testing"
  8. "github.com/go-xorm/core"
  9. "github.com/stretchr/testify/assert"
  10. )
  11. type IntId struct {
  12. Id int `xorm:"pk autoincr"`
  13. Name string
  14. }
  15. type Int16Id struct {
  16. Id int16 `xorm:"pk autoincr"`
  17. Name string
  18. }
  19. type Int32Id struct {
  20. Id int32 `xorm:"pk autoincr"`
  21. Name string
  22. }
  23. type UintId struct {
  24. Id uint `xorm:"pk autoincr"`
  25. Name string
  26. }
  27. type Uint16Id struct {
  28. Id uint16 `xorm:"pk autoincr"`
  29. Name string
  30. }
  31. type Uint32Id struct {
  32. Id uint32 `xorm:"pk autoincr"`
  33. Name string
  34. }
  35. type Uint64Id struct {
  36. Id uint64 `xorm:"pk autoincr"`
  37. Name string
  38. }
  39. type StringPK struct {
  40. Id string `xorm:"pk notnull"`
  41. Name string
  42. }
  43. type ID int64
  44. type MyIntPK struct {
  45. ID ID `xorm:"pk autoincr"`
  46. Name string
  47. }
  48. type StrID string
  49. type MyStringPK struct {
  50. ID StrID `xorm:"pk notnull"`
  51. Name string
  52. }
  53. func TestIntId(t *testing.T) {
  54. assert.NoError(t, prepareEngine())
  55. err := testEngine.DropTables(&IntId{})
  56. if err != nil {
  57. t.Error(err)
  58. panic(err)
  59. }
  60. err = testEngine.CreateTables(&IntId{})
  61. if err != nil {
  62. t.Error(err)
  63. panic(err)
  64. }
  65. cnt, err := testEngine.Insert(&IntId{Name: "test"})
  66. if err != nil {
  67. t.Error(err)
  68. panic(err)
  69. }
  70. if cnt != 1 {
  71. err = errors.New("insert count should be one")
  72. t.Error(err)
  73. panic(err)
  74. }
  75. bean := new(IntId)
  76. has, err := testEngine.Get(bean)
  77. if err != nil {
  78. t.Error(err)
  79. panic(err)
  80. }
  81. if !has {
  82. err = errors.New("get count should be one")
  83. t.Error(err)
  84. panic(err)
  85. }
  86. beans := make([]IntId, 0)
  87. err = testEngine.Find(&beans)
  88. if err != nil {
  89. t.Error(err)
  90. panic(err)
  91. }
  92. if len(beans) != 1 {
  93. err = errors.New("get count should be one")
  94. t.Error(err)
  95. panic(err)
  96. }
  97. beans2 := make(map[int]IntId)
  98. err = testEngine.Find(&beans2)
  99. if err != nil {
  100. t.Error(err)
  101. panic(err)
  102. }
  103. if len(beans2) != 1 {
  104. err = errors.New("get count should be one")
  105. t.Error(err)
  106. panic(err)
  107. }
  108. cnt, err = testEngine.Id(bean.Id).Delete(&IntId{})
  109. if err != nil {
  110. t.Error(err)
  111. panic(err)
  112. }
  113. if cnt != 1 {
  114. err = errors.New("insert count should be one")
  115. t.Error(err)
  116. panic(err)
  117. }
  118. }
  119. func TestInt16Id(t *testing.T) {
  120. assert.NoError(t, prepareEngine())
  121. err := testEngine.DropTables(&Int16Id{})
  122. if err != nil {
  123. t.Error(err)
  124. panic(err)
  125. }
  126. err = testEngine.CreateTables(&Int16Id{})
  127. if err != nil {
  128. t.Error(err)
  129. panic(err)
  130. }
  131. cnt, err := testEngine.Insert(&Int16Id{Name: "test"})
  132. if err != nil {
  133. t.Error(err)
  134. panic(err)
  135. }
  136. if cnt != 1 {
  137. err = errors.New("insert count should be one")
  138. t.Error(err)
  139. panic(err)
  140. }
  141. bean := new(Int16Id)
  142. has, err := testEngine.Get(bean)
  143. if err != nil {
  144. t.Error(err)
  145. panic(err)
  146. }
  147. if !has {
  148. err = errors.New("get count should be one")
  149. t.Error(err)
  150. panic(err)
  151. }
  152. beans := make([]Int16Id, 0)
  153. err = testEngine.Find(&beans)
  154. if err != nil {
  155. t.Error(err)
  156. panic(err)
  157. }
  158. if len(beans) != 1 {
  159. err = errors.New("get count should be one")
  160. t.Error(err)
  161. panic(err)
  162. }
  163. beans2 := make(map[int16]Int16Id, 0)
  164. err = testEngine.Find(&beans2)
  165. if err != nil {
  166. t.Error(err)
  167. panic(err)
  168. }
  169. if len(beans2) != 1 {
  170. err = errors.New("get count should be one")
  171. t.Error(err)
  172. panic(err)
  173. }
  174. cnt, err = testEngine.Id(bean.Id).Delete(&Int16Id{})
  175. if err != nil {
  176. t.Error(err)
  177. panic(err)
  178. }
  179. if cnt != 1 {
  180. err = errors.New("insert count should be one")
  181. t.Error(err)
  182. panic(err)
  183. }
  184. }
  185. func TestInt32Id(t *testing.T) {
  186. assert.NoError(t, prepareEngine())
  187. err := testEngine.DropTables(&Int32Id{})
  188. if err != nil {
  189. t.Error(err)
  190. panic(err)
  191. }
  192. err = testEngine.CreateTables(&Int32Id{})
  193. if err != nil {
  194. t.Error(err)
  195. panic(err)
  196. }
  197. cnt, err := testEngine.Insert(&Int32Id{Name: "test"})
  198. if err != nil {
  199. t.Error(err)
  200. panic(err)
  201. }
  202. if cnt != 1 {
  203. err = errors.New("insert count should be one")
  204. t.Error(err)
  205. panic(err)
  206. }
  207. bean := new(Int32Id)
  208. has, err := testEngine.Get(bean)
  209. if err != nil {
  210. t.Error(err)
  211. panic(err)
  212. }
  213. if !has {
  214. err = errors.New("get count should be one")
  215. t.Error(err)
  216. panic(err)
  217. }
  218. beans := make([]Int32Id, 0)
  219. err = testEngine.Find(&beans)
  220. if err != nil {
  221. t.Error(err)
  222. panic(err)
  223. }
  224. if len(beans) != 1 {
  225. err = errors.New("get count should be one")
  226. t.Error(err)
  227. panic(err)
  228. }
  229. beans2 := make(map[int32]Int32Id, 0)
  230. err = testEngine.Find(&beans2)
  231. if err != nil {
  232. t.Error(err)
  233. panic(err)
  234. }
  235. if len(beans2) != 1 {
  236. err = errors.New("get count should be one")
  237. t.Error(err)
  238. panic(err)
  239. }
  240. cnt, err = testEngine.Id(bean.Id).Delete(&Int32Id{})
  241. if err != nil {
  242. t.Error(err)
  243. panic(err)
  244. }
  245. if cnt != 1 {
  246. err = errors.New("insert count should be one")
  247. t.Error(err)
  248. panic(err)
  249. }
  250. }
  251. func TestUintId(t *testing.T) {
  252. assert.NoError(t, prepareEngine())
  253. err := testEngine.DropTables(&UintId{})
  254. if err != nil {
  255. t.Error(err)
  256. panic(err)
  257. }
  258. err = testEngine.CreateTables(&UintId{})
  259. if err != nil {
  260. t.Error(err)
  261. panic(err)
  262. }
  263. cnt, err := testEngine.Insert(&UintId{Name: "test"})
  264. if err != nil {
  265. t.Error(err)
  266. panic(err)
  267. }
  268. if cnt != 1 {
  269. err = errors.New("insert count should be one")
  270. t.Error(err)
  271. panic(err)
  272. }
  273. var inserts = []UintId{
  274. {Name: "test1"},
  275. {Name: "test2"},
  276. }
  277. cnt, err = testEngine.Insert(&inserts)
  278. if err != nil {
  279. t.Error(err)
  280. panic(err)
  281. }
  282. if cnt != 2 {
  283. err = errors.New("insert count should be two")
  284. t.Error(err)
  285. panic(err)
  286. }
  287. bean := new(UintId)
  288. has, err := testEngine.Get(bean)
  289. if err != nil {
  290. t.Error(err)
  291. panic(err)
  292. }
  293. if !has {
  294. err = errors.New("get count should be one")
  295. t.Error(err)
  296. panic(err)
  297. }
  298. beans := make([]UintId, 0)
  299. err = testEngine.Find(&beans)
  300. if err != nil {
  301. t.Error(err)
  302. panic(err)
  303. }
  304. if len(beans) != 3 {
  305. err = errors.New("get count should be three")
  306. t.Error(err)
  307. panic(err)
  308. }
  309. beans2 := make(map[uint]UintId, 0)
  310. err = testEngine.Find(&beans2)
  311. if err != nil {
  312. t.Error(err)
  313. panic(err)
  314. }
  315. if len(beans2) != 3 {
  316. err = errors.New("get count should be three")
  317. t.Error(err)
  318. panic(err)
  319. }
  320. cnt, err = testEngine.Id(bean.Id).Delete(&UintId{})
  321. if err != nil {
  322. t.Error(err)
  323. panic(err)
  324. }
  325. if cnt != 1 {
  326. err = errors.New("insert count should be one")
  327. t.Error(err)
  328. panic(err)
  329. }
  330. }
  331. func TestUint16Id(t *testing.T) {
  332. assert.NoError(t, prepareEngine())
  333. err := testEngine.DropTables(&Uint16Id{})
  334. if err != nil {
  335. t.Error(err)
  336. panic(err)
  337. }
  338. err = testEngine.CreateTables(&Uint16Id{})
  339. if err != nil {
  340. t.Error(err)
  341. panic(err)
  342. }
  343. cnt, err := testEngine.Insert(&Uint16Id{Name: "test"})
  344. if err != nil {
  345. t.Error(err)
  346. panic(err)
  347. }
  348. if cnt != 1 {
  349. err = errors.New("insert count should be one")
  350. t.Error(err)
  351. panic(err)
  352. }
  353. bean := new(Uint16Id)
  354. has, err := testEngine.Get(bean)
  355. if err != nil {
  356. t.Error(err)
  357. panic(err)
  358. }
  359. if !has {
  360. err = errors.New("get count should be one")
  361. t.Error(err)
  362. panic(err)
  363. }
  364. beans := make([]Uint16Id, 0)
  365. err = testEngine.Find(&beans)
  366. if err != nil {
  367. t.Error(err)
  368. panic(err)
  369. }
  370. if len(beans) != 1 {
  371. err = errors.New("get count should be one")
  372. t.Error(err)
  373. panic(err)
  374. }
  375. beans2 := make(map[uint16]Uint16Id, 0)
  376. err = testEngine.Find(&beans2)
  377. if err != nil {
  378. t.Error(err)
  379. panic(err)
  380. }
  381. if len(beans2) != 1 {
  382. err = errors.New("get count should be one")
  383. t.Error(err)
  384. panic(err)
  385. }
  386. cnt, err = testEngine.Id(bean.Id).Delete(&Uint16Id{})
  387. if err != nil {
  388. t.Error(err)
  389. panic(err)
  390. }
  391. if cnt != 1 {
  392. err = errors.New("insert count should be one")
  393. t.Error(err)
  394. panic(err)
  395. }
  396. }
  397. func TestUint32Id(t *testing.T) {
  398. assert.NoError(t, prepareEngine())
  399. err := testEngine.DropTables(&Uint32Id{})
  400. if err != nil {
  401. t.Error(err)
  402. panic(err)
  403. }
  404. err = testEngine.CreateTables(&Uint32Id{})
  405. if err != nil {
  406. t.Error(err)
  407. panic(err)
  408. }
  409. cnt, err := testEngine.Insert(&Uint32Id{Name: "test"})
  410. if err != nil {
  411. t.Error(err)
  412. panic(err)
  413. }
  414. if cnt != 1 {
  415. err = errors.New("insert count should be one")
  416. t.Error(err)
  417. panic(err)
  418. }
  419. bean := new(Uint32Id)
  420. has, err := testEngine.Get(bean)
  421. if err != nil {
  422. t.Error(err)
  423. panic(err)
  424. }
  425. if !has {
  426. err = errors.New("get count should be one")
  427. t.Error(err)
  428. panic(err)
  429. }
  430. beans := make([]Uint32Id, 0)
  431. err = testEngine.Find(&beans)
  432. if err != nil {
  433. t.Error(err)
  434. panic(err)
  435. }
  436. if len(beans) != 1 {
  437. err = errors.New("get count should be one")
  438. t.Error(err)
  439. panic(err)
  440. }
  441. beans2 := make(map[uint32]Uint32Id, 0)
  442. err = testEngine.Find(&beans2)
  443. if err != nil {
  444. t.Error(err)
  445. panic(err)
  446. }
  447. if len(beans2) != 1 {
  448. err = errors.New("get count should be one")
  449. t.Error(err)
  450. panic(err)
  451. }
  452. cnt, err = testEngine.Id(bean.Id).Delete(&Uint32Id{})
  453. if err != nil {
  454. t.Error(err)
  455. panic(err)
  456. }
  457. if cnt != 1 {
  458. err = errors.New("insert count should be one")
  459. t.Error(err)
  460. panic(err)
  461. }
  462. }
  463. func TestUint64Id(t *testing.T) {
  464. assert.NoError(t, prepareEngine())
  465. err := testEngine.DropTables(&Uint64Id{})
  466. if err != nil {
  467. t.Error(err)
  468. panic(err)
  469. }
  470. err = testEngine.CreateTables(&Uint64Id{})
  471. if err != nil {
  472. t.Error(err)
  473. panic(err)
  474. }
  475. idbean := &Uint64Id{Name: "test"}
  476. cnt, err := testEngine.Insert(idbean)
  477. if err != nil {
  478. t.Error(err)
  479. panic(err)
  480. }
  481. if cnt != 1 {
  482. err = errors.New("insert count should be one")
  483. t.Error(err)
  484. panic(err)
  485. }
  486. bean := new(Uint64Id)
  487. has, err := testEngine.Get(bean)
  488. if err != nil {
  489. t.Error(err)
  490. panic(err)
  491. }
  492. if !has {
  493. err = errors.New("get count should be one")
  494. t.Error(err)
  495. panic(err)
  496. }
  497. if bean.Id != idbean.Id {
  498. panic(errors.New("should be equal"))
  499. }
  500. beans := make([]Uint64Id, 0)
  501. err = testEngine.Find(&beans)
  502. if err != nil {
  503. t.Error(err)
  504. panic(err)
  505. }
  506. if len(beans) != 1 {
  507. err = errors.New("get count should be one")
  508. t.Error(err)
  509. panic(err)
  510. }
  511. if *bean != beans[0] {
  512. panic(errors.New("should be equal"))
  513. }
  514. beans2 := make(map[uint64]Uint64Id, 0)
  515. err = testEngine.Find(&beans2)
  516. if err != nil {
  517. t.Error(err)
  518. panic(err)
  519. }
  520. if len(beans2) != 1 {
  521. err = errors.New("get count should be one")
  522. t.Error(err)
  523. panic(err)
  524. }
  525. if *bean != beans2[bean.Id] {
  526. panic(errors.New("should be equal"))
  527. }
  528. cnt, err = testEngine.Id(bean.Id).Delete(&Uint64Id{})
  529. if err != nil {
  530. t.Error(err)
  531. panic(err)
  532. }
  533. if cnt != 1 {
  534. err = errors.New("insert count should be one")
  535. t.Error(err)
  536. panic(err)
  537. }
  538. }
  539. func TestStringPK(t *testing.T) {
  540. assert.NoError(t, prepareEngine())
  541. err := testEngine.DropTables(&StringPK{})
  542. if err != nil {
  543. t.Error(err)
  544. panic(err)
  545. }
  546. err = testEngine.CreateTables(&StringPK{})
  547. if err != nil {
  548. t.Error(err)
  549. panic(err)
  550. }
  551. cnt, err := testEngine.Insert(&StringPK{Id: "1-1-2", Name: "test"})
  552. if err != nil {
  553. t.Error(err)
  554. panic(err)
  555. }
  556. if cnt != 1 {
  557. err = errors.New("insert count should be one")
  558. t.Error(err)
  559. panic(err)
  560. }
  561. bean := new(StringPK)
  562. has, err := testEngine.Get(bean)
  563. if err != nil {
  564. t.Error(err)
  565. panic(err)
  566. }
  567. if !has {
  568. err = errors.New("get count should be one")
  569. t.Error(err)
  570. panic(err)
  571. }
  572. beans := make([]StringPK, 0)
  573. err = testEngine.Find(&beans)
  574. if err != nil {
  575. t.Error(err)
  576. panic(err)
  577. }
  578. if len(beans) != 1 {
  579. err = errors.New("get count should be one")
  580. t.Error(err)
  581. panic(err)
  582. }
  583. beans2 := make(map[string]StringPK)
  584. err = testEngine.Find(&beans2)
  585. if err != nil {
  586. t.Error(err)
  587. panic(err)
  588. }
  589. if len(beans2) != 1 {
  590. err = errors.New("get count should be one")
  591. t.Error(err)
  592. panic(err)
  593. }
  594. cnt, err = testEngine.Id(bean.Id).Delete(&StringPK{})
  595. if err != nil {
  596. t.Error(err)
  597. panic(err)
  598. }
  599. if cnt != 1 {
  600. err = errors.New("insert count should be one")
  601. t.Error(err)
  602. panic(err)
  603. }
  604. }
  605. type CompositeKey struct {
  606. Id1 int64 `xorm:"id1 pk"`
  607. Id2 int64 `xorm:"id2 pk"`
  608. UpdateStr string
  609. }
  610. func TestCompositeKey(t *testing.T) {
  611. assert.NoError(t, prepareEngine())
  612. err := testEngine.DropTables(&CompositeKey{})
  613. if err != nil {
  614. t.Error(err)
  615. panic(err)
  616. }
  617. err = testEngine.CreateTables(&CompositeKey{})
  618. if err != nil {
  619. t.Error(err)
  620. panic(err)
  621. }
  622. cnt, err := testEngine.Insert(&CompositeKey{11, 22, ""})
  623. if err != nil {
  624. t.Error(err)
  625. } else if cnt != 1 {
  626. t.Error(errors.New("failed to insert CompositeKey{11, 22}"))
  627. }
  628. cnt, err = testEngine.Insert(&CompositeKey{11, 22, ""})
  629. if err == nil || cnt == 1 {
  630. t.Error(errors.New("inserted CompositeKey{11, 22}"))
  631. }
  632. var compositeKeyVal CompositeKey
  633. has, err := testEngine.Id(core.PK{11, 22}).Get(&compositeKeyVal)
  634. if err != nil {
  635. t.Error(err)
  636. } else if !has {
  637. t.Error(errors.New("can't get CompositeKey{11, 22}"))
  638. }
  639. var compositeKeyVal2 CompositeKey
  640. // test passing PK ptr, this test seem failed withCache
  641. has, err = testEngine.Id(&core.PK{11, 22}).Get(&compositeKeyVal2)
  642. if err != nil {
  643. t.Error(err)
  644. } else if !has {
  645. t.Error(errors.New("can't get CompositeKey{11, 22}"))
  646. }
  647. if compositeKeyVal != compositeKeyVal2 {
  648. t.Error(errors.New("should be equal"))
  649. }
  650. var cps = make([]CompositeKey, 0)
  651. err = testEngine.Find(&cps)
  652. if err != nil {
  653. t.Error(err)
  654. }
  655. if len(cps) != 1 {
  656. t.Error(errors.New("should has one record"))
  657. }
  658. if cps[0] != compositeKeyVal {
  659. t.Error(errors.New("should be equal"))
  660. }
  661. cnt, err = testEngine.Insert(&CompositeKey{22, 22, ""})
  662. if err != nil {
  663. t.Error(err)
  664. } else if cnt != 1 {
  665. t.Error(errors.New("failed to insert CompositeKey{22, 22}"))
  666. }
  667. if testEngine.Cacher != nil {
  668. testEngine.Cacher.ClearBeans(testEngine.TableInfo(compositeKeyVal).Name)
  669. }
  670. cps = make([]CompositeKey, 0)
  671. err = testEngine.Find(&cps)
  672. if err != nil {
  673. t.Error(err)
  674. }
  675. if len(cps) != 2 {
  676. t.Error(errors.New("should has two record"))
  677. }
  678. if cps[0] != compositeKeyVal {
  679. t.Error(errors.New("should be equeal"))
  680. }
  681. compositeKeyVal = CompositeKey{UpdateStr: "test1"}
  682. cnt, err = testEngine.Id(core.PK{11, 22}).Update(&compositeKeyVal)
  683. if err != nil {
  684. t.Error(err)
  685. } else if cnt != 1 {
  686. t.Error(errors.New("can't update CompositeKey{11, 22}"))
  687. }
  688. cnt, err = testEngine.Id(core.PK{11, 22}).Delete(&CompositeKey{})
  689. if err != nil {
  690. t.Error(err)
  691. } else if cnt != 1 {
  692. t.Error(errors.New("can't delete CompositeKey{11, 22}"))
  693. }
  694. }
  695. type User struct {
  696. UserId string `xorm:"varchar(19) not null pk"`
  697. NickName string `xorm:"varchar(19) not null"`
  698. GameId uint32 `xorm:"integer pk"`
  699. Score int32 `xorm:"integer"`
  700. }
  701. func TestCompositeKey2(t *testing.T) {
  702. assert.NoError(t, prepareEngine())
  703. err := testEngine.DropTables(&User{})
  704. if err != nil {
  705. t.Error(err)
  706. panic(err)
  707. }
  708. err = testEngine.CreateTables(&User{})
  709. if err != nil {
  710. t.Error(err)
  711. panic(err)
  712. }
  713. cnt, err := testEngine.Insert(&User{"11", "nick", 22, 5})
  714. if err != nil {
  715. t.Error(err)
  716. } else if cnt != 1 {
  717. t.Error(errors.New("failed to insert User{11, 22}"))
  718. }
  719. cnt, err = testEngine.Insert(&User{"11", "nick", 22, 6})
  720. if err == nil || cnt == 1 {
  721. t.Error(errors.New("inserted User{11, 22}"))
  722. }
  723. var user User
  724. has, err := testEngine.Id(core.PK{"11", 22}).Get(&user)
  725. if err != nil {
  726. t.Error(err)
  727. } else if !has {
  728. t.Error(errors.New("can't get User{11, 22}"))
  729. }
  730. // test passing PK ptr, this test seem failed withCache
  731. has, err = testEngine.Id(&core.PK{"11", 22}).Get(&user)
  732. if err != nil {
  733. t.Error(err)
  734. } else if !has {
  735. t.Error(errors.New("can't get User{11, 22}"))
  736. }
  737. user = User{NickName: "test1"}
  738. cnt, err = testEngine.Id(core.PK{"11", 22}).Update(&user)
  739. if err != nil {
  740. t.Error(err)
  741. } else if cnt != 1 {
  742. t.Error(errors.New("can't update User{11, 22}"))
  743. }
  744. cnt, err = testEngine.Id(core.PK{"11", 22}).Delete(&User{})
  745. if err != nil {
  746. t.Error(err)
  747. } else if cnt != 1 {
  748. t.Error(errors.New("can't delete CompositeKey{11, 22}"))
  749. }
  750. }
  751. type MyString string
  752. type UserPK2 struct {
  753. UserId MyString `xorm:"varchar(19) not null pk"`
  754. NickName string `xorm:"varchar(19) not null"`
  755. GameId uint32 `xorm:"integer pk"`
  756. Score int32 `xorm:"integer"`
  757. }
  758. func TestCompositeKey3(t *testing.T) {
  759. assert.NoError(t, prepareEngine())
  760. err := testEngine.DropTables(&UserPK2{})
  761. if err != nil {
  762. t.Error(err)
  763. panic(err)
  764. }
  765. err = testEngine.CreateTables(&UserPK2{})
  766. if err != nil {
  767. t.Error(err)
  768. panic(err)
  769. }
  770. cnt, err := testEngine.Insert(&UserPK2{"11", "nick", 22, 5})
  771. if err != nil {
  772. t.Error(err)
  773. } else if cnt != 1 {
  774. t.Error(errors.New("failed to insert User{11, 22}"))
  775. }
  776. cnt, err = testEngine.Insert(&UserPK2{"11", "nick", 22, 6})
  777. if err == nil || cnt == 1 {
  778. t.Error(errors.New("inserted User{11, 22}"))
  779. }
  780. var user UserPK2
  781. has, err := testEngine.Id(core.PK{"11", 22}).Get(&user)
  782. if err != nil {
  783. t.Error(err)
  784. } else if !has {
  785. t.Error(errors.New("can't get User{11, 22}"))
  786. }
  787. // test passing PK ptr, this test seem failed withCache
  788. has, err = testEngine.Id(&core.PK{"11", 22}).Get(&user)
  789. if err != nil {
  790. t.Error(err)
  791. } else if !has {
  792. t.Error(errors.New("can't get User{11, 22}"))
  793. }
  794. user = UserPK2{NickName: "test1"}
  795. cnt, err = testEngine.Id(core.PK{"11", 22}).Update(&user)
  796. if err != nil {
  797. t.Error(err)
  798. } else if cnt != 1 {
  799. t.Error(errors.New("can't update User{11, 22}"))
  800. }
  801. cnt, err = testEngine.Id(core.PK{"11", 22}).Delete(&UserPK2{})
  802. if err != nil {
  803. t.Error(err)
  804. } else if cnt != 1 {
  805. t.Error(errors.New("can't delete CompositeKey{11, 22}"))
  806. }
  807. }
  808. func TestMyIntId(t *testing.T) {
  809. assert.NoError(t, prepareEngine())
  810. err := testEngine.DropTables(&MyIntPK{})
  811. if err != nil {
  812. t.Error(err)
  813. panic(err)
  814. }
  815. err = testEngine.CreateTables(&MyIntPK{})
  816. if err != nil {
  817. t.Error(err)
  818. panic(err)
  819. }
  820. idbean := &MyIntPK{Name: "test"}
  821. cnt, err := testEngine.Insert(idbean)
  822. if err != nil {
  823. t.Error(err)
  824. panic(err)
  825. }
  826. if cnt != 1 {
  827. err = errors.New("insert count should be one")
  828. t.Error(err)
  829. panic(err)
  830. }
  831. bean := new(MyIntPK)
  832. has, err := testEngine.Get(bean)
  833. if err != nil {
  834. t.Error(err)
  835. panic(err)
  836. }
  837. if !has {
  838. err = errors.New("get count should be one")
  839. t.Error(err)
  840. panic(err)
  841. }
  842. if bean.ID != idbean.ID {
  843. panic(errors.New("should be equal"))
  844. }
  845. var beans []MyIntPK
  846. err = testEngine.Find(&beans)
  847. if err != nil {
  848. t.Error(err)
  849. panic(err)
  850. }
  851. if len(beans) != 1 {
  852. err = errors.New("get count should be one")
  853. t.Error(err)
  854. panic(err)
  855. }
  856. if *bean != beans[0] {
  857. panic(errors.New("should be equal"))
  858. }
  859. beans2 := make(map[ID]MyIntPK, 0)
  860. err = testEngine.Find(&beans2)
  861. if err != nil {
  862. t.Error(err)
  863. panic(err)
  864. }
  865. if len(beans2) != 1 {
  866. err = errors.New("get count should be one")
  867. t.Error(err)
  868. panic(err)
  869. }
  870. if *bean != beans2[bean.ID] {
  871. panic(errors.New("should be equal"))
  872. }
  873. cnt, err = testEngine.Id(bean.ID).Delete(&MyIntPK{})
  874. if err != nil {
  875. t.Error(err)
  876. panic(err)
  877. }
  878. if cnt != 1 {
  879. err = errors.New("insert count should be one")
  880. t.Error(err)
  881. panic(err)
  882. }
  883. }
  884. func TestMyStringId(t *testing.T) {
  885. assert.NoError(t, prepareEngine())
  886. err := testEngine.DropTables(&MyStringPK{})
  887. if err != nil {
  888. t.Error(err)
  889. panic(err)
  890. }
  891. err = testEngine.CreateTables(&MyStringPK{})
  892. if err != nil {
  893. t.Error(err)
  894. panic(err)
  895. }
  896. idbean := &MyStringPK{ID: "1111", Name: "test"}
  897. cnt, err := testEngine.Insert(idbean)
  898. if err != nil {
  899. t.Error(err)
  900. panic(err)
  901. }
  902. if cnt != 1 {
  903. err = errors.New("insert count should be one")
  904. t.Error(err)
  905. panic(err)
  906. }
  907. bean := new(MyStringPK)
  908. has, err := testEngine.Get(bean)
  909. if err != nil {
  910. t.Error(err)
  911. panic(err)
  912. }
  913. if !has {
  914. err = errors.New("get count should be one")
  915. t.Error(err)
  916. panic(err)
  917. }
  918. if bean.ID != idbean.ID {
  919. panic(errors.New("should be equal"))
  920. }
  921. var beans []MyStringPK
  922. err = testEngine.Find(&beans)
  923. if err != nil {
  924. t.Error(err)
  925. panic(err)
  926. }
  927. if len(beans) != 1 {
  928. err = errors.New("get count should be one")
  929. t.Error(err)
  930. panic(err)
  931. }
  932. if *bean != beans[0] {
  933. panic(errors.New("should be equal"))
  934. }
  935. beans2 := make(map[StrID]MyStringPK, 0)
  936. err = testEngine.Find(&beans2)
  937. if err != nil {
  938. t.Error(err)
  939. panic(err)
  940. }
  941. if len(beans2) != 1 {
  942. err = errors.New("get count should be one")
  943. t.Error(err)
  944. panic(err)
  945. }
  946. if *bean != beans2[bean.ID] {
  947. panic(errors.New("should be equal"))
  948. }
  949. cnt, err = testEngine.Id(bean.ID).Delete(&MyStringPK{})
  950. if err != nil {
  951. t.Error(err)
  952. panic(err)
  953. }
  954. if cnt != 1 {
  955. err = errors.New("insert count should be one")
  956. t.Error(err)
  957. panic(err)
  958. }
  959. }
  960. func TestSingleAutoIncrColumn(t *testing.T) {
  961. type Account struct {
  962. Id int64 `xorm:"pk autoincr"`
  963. }
  964. assert.NoError(t, prepareEngine())
  965. assertSync(t, new(Account))
  966. _, err := testEngine.Insert(&Account{})
  967. assert.NoError(t, err)
  968. }