file_test.go 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604
  1. package burndown
  2. import (
  3. "testing"
  4. "github.com/stretchr/testify/assert"
  5. "gopkg.in/src-d/hercules.v4/internal/rbtree"
  6. "fmt"
  7. )
  8. func updateStatusFile(
  9. status interface{}, _ int, previousTime int, delta int) {
  10. status.(map[int]int64)[previousTime] += int64(delta)
  11. }
  12. func fixtureFile() (*File, map[int]int64) {
  13. status := map[int]int64{}
  14. file := NewFile(0, 100, NewStatus(status, updateStatusFile))
  15. return file, status
  16. }
  17. func TestInitializeFile(t *testing.T) {
  18. file, status := fixtureFile()
  19. dump := file.Dump()
  20. // Output:
  21. // 0 0
  22. // 100 -1
  23. assert.Equal(t, "0 0\n100 -1\n", dump)
  24. assert.Equal(t, int64(100), status[0])
  25. }
  26. func testPanicFile(t *testing.T, method func(*File), msg string) {
  27. defer func() {
  28. r := recover()
  29. assert.NotNil(t, r, "not panic()-ed")
  30. assert.IsType(t, "", r)
  31. assert.Contains(t, r.(string), msg)
  32. }()
  33. file, _ := fixtureFile()
  34. method(file)
  35. }
  36. func TestBullshitFile(t *testing.T) {
  37. testPanicFile(t, func(file *File) { file.Update(1, -10, 10, 0) }, "insert")
  38. testPanicFile(t, func(file *File) { file.Update(1, 110, 10, 0) }, "insert")
  39. testPanicFile(t, func(file *File) { file.Update(1, -10, 0, 10) }, "delete")
  40. testPanicFile(t, func(file *File) { file.Update(1, 100, 0, 10) }, "delete")
  41. testPanicFile(t, func(file *File) { file.Update(1, 0, -10, 0) }, "Length")
  42. testPanicFile(t, func(file *File) { file.Update(1, 0, 0, -10) }, "Length")
  43. testPanicFile(t, func(file *File) { file.Update(1, 0, -10, -10) }, "Length")
  44. testPanicFile(t, func(file *File) { file.Update(-1, 0, 10, 10) }, "time")
  45. file, status := fixtureFile()
  46. file.Update(1, 10, 0, 0)
  47. assert.Equal(t, int64(100), status[0])
  48. assert.Equal(t, int64(0), status[1])
  49. }
  50. func TestCloneFile(t *testing.T) {
  51. file, status := fixtureFile()
  52. // 0 0 | 100 -1 [0]: 100
  53. file.Update(1, 20, 30, 0)
  54. // 0 0 | 20 1 | 50 0 | 130 -1 [0]: 100, [1]: 30
  55. file.Update(2, 20, 0, 5)
  56. // 0 0 | 20 1 | 45 0 | 125 -1 [0]: 100, [1]: 25
  57. file.Update(3, 20, 0, 5)
  58. // 0 0 | 20 1 | 40 0 | 120 -1 [0]: 100, [1]: 20
  59. file.Update(4, 20, 10, 0)
  60. // 0 0 | 20 4 | 30 1 | 50 0 | 130 -1 [0]: 100, [1]: 20, [4]: 10
  61. clone := file.Clone(false)
  62. clone.Update(5, 45, 0, 10)
  63. // 0 0 | 20 4 | 30 1 | 45 0 | 120 -1 [0]: 95, [1]: 15, [4]: 10
  64. clone.Update(6, 45, 5, 0)
  65. // 0 0 | 20 4 | 30 1 | 45 6 | 50 0 | 125 -1 [0]: 95, [1]: 15, [4]: 10, [6]: 5
  66. assert.Equal(t, int64(95), status[0])
  67. assert.Equal(t, int64(15), status[1])
  68. assert.Equal(t, int64(0), status[2])
  69. assert.Equal(t, int64(0), status[3])
  70. assert.Equal(t, int64(10), status[4])
  71. assert.Equal(t, int64(0), status[5])
  72. assert.Equal(t, int64(5), status[6])
  73. dump := file.Dump()
  74. // Output:
  75. // 0 0
  76. // 20 4
  77. // 30 1
  78. // 50 0
  79. // 130 -1
  80. assert.Equal(t, "0 0\n20 4\n30 1\n50 0\n130 -1\n", dump)
  81. dump = clone.Dump()
  82. // Output:
  83. // 0 0
  84. // 20 4
  85. // 30 1
  86. // 45 6
  87. // 50 0
  88. // 125 -1
  89. assert.Equal(t, "0 0\n20 4\n30 1\n45 6\n50 0\n125 -1\n", dump)
  90. }
  91. func TestCloneFileClearStatus(t *testing.T) {
  92. file, status := fixtureFile()
  93. // 0 0 | 100 -1 [0]: 100
  94. file.Update(1, 20, 30, 0)
  95. // 0 0 | 20 1 | 50 0 | 130 -1 [0]: 100, [1]: 30
  96. file.Update(2, 20, 0, 5)
  97. // 0 0 | 20 1 | 45 0 | 125 -1 [0]: 100, [1]: 25
  98. file.Update(3, 20, 0, 5)
  99. // 0 0 | 20 1 | 40 0 | 120 -1 [0]: 100, [1]: 20
  100. file.Update(4, 20, 10, 0)
  101. // 0 0 | 20 4 | 30 1 | 50 0 | 130 -1 [0]: 100, [1]: 20, [4]: 10
  102. newStatus := map[int]int64{}
  103. clone := file.Clone(true, NewStatus(newStatus, updateStatusFile))
  104. clone.Update(5, 45, 0, 10)
  105. // 0 0 | 20 4 | 30 1 | 45 0 | 120 -1 [0]: -5, [1]: -5
  106. clone.Update(6, 45, 5, 0)
  107. // 0 0 | 20 4 | 30 1 | 45 6 | 50 0 | 125 -1 [0]: -5, [1]: -5, [6]: 5
  108. assert.Equal(t, int64(100), status[0])
  109. assert.Equal(t, int64(20), status[1])
  110. assert.Equal(t, int64(0), status[2])
  111. assert.Equal(t, int64(0), status[3])
  112. assert.Equal(t, int64(10), status[4])
  113. assert.Equal(t, int64(-5), newStatus[0])
  114. assert.Equal(t, int64(-5), newStatus[1])
  115. assert.Equal(t, int64(0), newStatus[2])
  116. assert.Equal(t, int64(0), newStatus[3])
  117. assert.Equal(t, int64(0), newStatus[4])
  118. assert.Equal(t, int64(0), newStatus[5])
  119. assert.Equal(t, int64(5), newStatus[6])
  120. dump := file.Dump()
  121. // Output:
  122. // 0 0
  123. // 20 4
  124. // 30 1
  125. // 50 0
  126. // 130 -1
  127. assert.Equal(t, "0 0\n20 4\n30 1\n50 0\n130 -1\n", dump)
  128. dump = clone.Dump()
  129. // Output:
  130. // 0 0
  131. // 20 4
  132. // 30 1
  133. // 45 6
  134. // 50 0
  135. // 125 -1
  136. assert.Equal(t, "0 0\n20 4\n30 1\n45 6\n50 0\n125 -1\n", dump)
  137. }
  138. func TestLenFile(t *testing.T) {
  139. file, _ := fixtureFile()
  140. assert.Equal(t, 100, file.Len())
  141. }
  142. func TestInsertFile(t *testing.T) {
  143. file, status := fixtureFile()
  144. file.Update(1, 10, 10, 0)
  145. dump := file.Dump()
  146. // Output:
  147. // 0 0
  148. // 10 1
  149. // 20 0
  150. // 110 -1
  151. assert.Equal(t, "0 0\n10 1\n20 0\n110 -1\n", dump)
  152. assert.Equal(t, int64(100), status[0])
  153. assert.Equal(t, int64(10), status[1])
  154. }
  155. func TestZeroInitializeFile(t *testing.T) {
  156. status := map[int]int64{}
  157. file := NewFile(0, 0, NewStatus(status, updateStatusFile))
  158. assert.NotContains(t, status, 0)
  159. dump := file.Dump()
  160. // Output:
  161. // 0 -1
  162. assert.Equal(t, "0 -1\n", dump)
  163. file.Update(1, 0, 10, 0)
  164. dump = file.Dump()
  165. // Output:
  166. // 0 1
  167. // 10 -1
  168. assert.Equal(t, "0 1\n10 -1\n", dump)
  169. assert.Equal(t, int64(10), status[1])
  170. }
  171. func TestDeleteFile(t *testing.T) {
  172. file, status := fixtureFile()
  173. file.Update(1, 10, 0, 10)
  174. dump := file.Dump()
  175. // Output:
  176. // 0 0
  177. // 90 -1
  178. assert.Equal(t, "0 0\n90 -1\n", dump)
  179. assert.Equal(t, int64(90), status[0])
  180. assert.Equal(t, int64(0), status[1])
  181. }
  182. func TestFusedFile(t *testing.T) {
  183. file, status := fixtureFile()
  184. file.Update(1, 10, 6, 7)
  185. dump := file.Dump()
  186. // Output:
  187. // 0 0
  188. // 10 1
  189. // 16 0
  190. // 99 -1
  191. assert.Equal(t, "0 0\n10 1\n16 0\n99 -1\n", dump)
  192. assert.Equal(t, int64(93), status[0])
  193. assert.Equal(t, int64(6), status[1])
  194. }
  195. func TestInsertSameTimeFile(t *testing.T) {
  196. file, status := fixtureFile()
  197. file.Update(0, 5, 10, 0)
  198. dump := file.Dump()
  199. // Output:
  200. // 0 0
  201. // 110 -1
  202. assert.Equal(t, "0 0\n110 -1\n", dump)
  203. assert.Equal(t, int64(110), status[0])
  204. }
  205. func TestInsertSameStartFile(t *testing.T) {
  206. file, status := fixtureFile()
  207. file.Update(1, 10, 10, 0)
  208. file.Update(2, 10, 10, 0)
  209. dump := file.Dump()
  210. // Output:
  211. // 0 0
  212. // 10 2
  213. // 20 1
  214. // 30 0
  215. // 120 -1
  216. assert.Equal(t, "0 0\n10 2\n20 1\n30 0\n120 -1\n", dump)
  217. assert.Equal(t, int64(100), status[0])
  218. assert.Equal(t, int64(10), status[1])
  219. assert.Equal(t, int64(10), status[2])
  220. }
  221. func TestInsertEndFile(t *testing.T) {
  222. file, status := fixtureFile()
  223. file.Update(1, 100, 10, 0)
  224. dump := file.Dump()
  225. // Output:
  226. // 0 0
  227. // 100 1
  228. // 110 -1
  229. assert.Equal(t, "0 0\n100 1\n110 -1\n", dump)
  230. assert.Equal(t, int64(100), status[0])
  231. assert.Equal(t, int64(10), status[1])
  232. }
  233. func TestDeleteSameStart0File(t *testing.T) {
  234. file, status := fixtureFile()
  235. file.Update(1, 0, 0, 10)
  236. dump := file.Dump()
  237. // Output:
  238. // 0 0
  239. // 90 -1
  240. assert.Equal(t, "0 0\n90 -1\n", dump)
  241. assert.Equal(t, int64(90), status[0])
  242. assert.Equal(t, int64(0), status[1])
  243. }
  244. func TestDeleteSameStartMiddleFile(t *testing.T) {
  245. file, status := fixtureFile()
  246. file.Update(1, 10, 10, 0)
  247. file.Update(2, 10, 0, 5)
  248. dump := file.Dump()
  249. // Output:
  250. // 0 0
  251. // 10 1
  252. // 15 0
  253. // 105 -1
  254. assert.Equal(t, "0 0\n10 1\n15 0\n105 -1\n", dump)
  255. assert.Equal(t, int64(100), status[0])
  256. assert.Equal(t, int64(5), status[1])
  257. }
  258. func TestDeleteIntersectionFile(t *testing.T) {
  259. file, status := fixtureFile()
  260. file.Update(1, 10, 10, 0)
  261. file.Update(2, 15, 0, 10)
  262. dump := file.Dump()
  263. // Output:
  264. // 0 0
  265. // 10 1
  266. // 15 0
  267. // 100 -1
  268. assert.Equal(t, "0 0\n10 1\n15 0\n100 -1\n", dump)
  269. assert.Equal(t, int64(95), status[0])
  270. assert.Equal(t, int64(5), status[1])
  271. }
  272. func TestDeleteAllFile(t *testing.T) {
  273. file, status := fixtureFile()
  274. file.Update(1, 0, 0, 100)
  275. // Output:
  276. // 0 -1
  277. dump := file.Dump()
  278. assert.Equal(t, "0 -1\n", dump)
  279. assert.Equal(t, int64(0), status[0])
  280. assert.Equal(t, int64(0), status[1])
  281. }
  282. func TestFusedIntersectionFile(t *testing.T) {
  283. file, status := fixtureFile()
  284. file.Update(1, 10, 10, 0)
  285. file.Update(2, 15, 3, 10)
  286. dump := file.Dump()
  287. // Output:
  288. // 0 0
  289. // 10 1
  290. // 15 2
  291. // 18 0
  292. // 103 -1
  293. assert.Equal(t, "0 0\n10 1\n15 2\n18 0\n103 -1\n", dump)
  294. assert.Equal(t, int64(95), status[0])
  295. assert.Equal(t, int64(5), status[1])
  296. assert.Equal(t, int64(3), status[2])
  297. }
  298. func TestTortureFile(t *testing.T) {
  299. file, status := fixtureFile()
  300. // 0 0 | 100 -1 [0]: 100
  301. file.Update(1, 20, 30, 0)
  302. // 0 0 | 20 1 | 50 0 | 130 -1 [0]: 100, [1]: 30
  303. file.Update(2, 20, 0, 5)
  304. // 0 0 | 20 1 | 45 0 | 125 -1 [0]: 100, [1]: 25
  305. file.Update(3, 20, 0, 5)
  306. // 0 0 | 20 1 | 40 0 | 120 -1 [0]: 100, [1]: 20
  307. file.Update(4, 20, 10, 0)
  308. // 0 0 | 20 4 | 30 1 | 50 0 | 130 -1 [0]: 100, [1]: 20, [4]: 10
  309. file.Update(5, 45, 0, 10)
  310. // 0 0 | 20 4 | 30 1 | 45 0 | 120 -1 [0]: 95, [1]: 15, [4]: 10
  311. file.Update(6, 45, 5, 0)
  312. // 0 0 | 20 4 | 30 1 | 45 6 | 50 0 | 125 -1 [0]: 95, [1]: 15, [4]: 10, [6]: 5
  313. file.Update(7, 10, 0, 50)
  314. // 0 0 | 75 -1 [0]: 75
  315. file.Update(8, 0, 10, 10)
  316. // 0 8 | 10 0 | 75 -1 [0]: 65, [8]: 10
  317. dump := file.Dump()
  318. assert.Equal(t, "0 8\n10 0\n75 -1\n", dump)
  319. assert.Equal(t, int64(65), status[0])
  320. assert.Equal(t, int64(0), status[1])
  321. assert.Equal(t, int64(0), status[2])
  322. assert.Equal(t, int64(0), status[3])
  323. assert.Equal(t, int64(0), status[4])
  324. assert.Equal(t, int64(0), status[5])
  325. assert.Equal(t, int64(0), status[6])
  326. assert.Equal(t, int64(0), status[7])
  327. assert.Equal(t, int64(10), status[8])
  328. }
  329. func TestInsertDeleteSameTimeFile(t *testing.T) {
  330. file, status := fixtureFile()
  331. file.Update(0, 10, 10, 20)
  332. dump := file.Dump()
  333. assert.Equal(t, "0 0\n90 -1\n", dump)
  334. assert.Equal(t, int64(90), status[0])
  335. file.Update(0, 10, 20, 10)
  336. dump = file.Dump()
  337. assert.Equal(t, "0 0\n100 -1\n", dump)
  338. assert.Equal(t, int64(100), status[0])
  339. }
  340. func TestBug1File(t *testing.T) {
  341. file, status := fixtureFile()
  342. file.Update(316, 1, 86, 0)
  343. file.Update(316, 87, 0, 99)
  344. file.Update(251, 0, 1, 0)
  345. file.Update(251, 1, 0, 1)
  346. dump := file.Dump()
  347. assert.Equal(t, "0 251\n1 316\n87 -1\n", dump)
  348. assert.Equal(t, int64(1), status[251])
  349. assert.Equal(t, int64(86), status[316])
  350. file.Update(316, 0, 0, 1)
  351. file.Update(316, 0, 1, 0)
  352. dump = file.Dump()
  353. assert.Equal(t, "0 316\n87 -1\n", dump)
  354. assert.Equal(t, int64(0), status[251])
  355. assert.Equal(t, int64(87), status[316])
  356. }
  357. func TestBug2File(t *testing.T) {
  358. file, status := fixtureFile()
  359. file.Update(316, 1, 86, 0)
  360. file.Update(316, 87, 0, 99)
  361. file.Update(251, 0, 1, 0)
  362. file.Update(251, 1, 0, 1)
  363. dump := file.Dump()
  364. assert.Equal(t, "0 251\n1 316\n87 -1\n", dump)
  365. file.Update(316, 0, 1, 1)
  366. dump = file.Dump()
  367. assert.Equal(t, "0 316\n87 -1\n", dump)
  368. assert.Equal(t, int64(0), status[251])
  369. assert.Equal(t, int64(87), status[316])
  370. }
  371. func TestJoinFile(t *testing.T) {
  372. file, status := fixtureFile()
  373. file.Update(1, 10, 10, 0)
  374. file.Update(1, 30, 10, 0)
  375. file.Update(1, 20, 10, 10)
  376. dump := file.Dump()
  377. assert.Equal(t, "0 0\n10 1\n40 0\n120 -1\n", dump)
  378. assert.Equal(t, int64(90), status[0])
  379. assert.Equal(t, int64(30), status[1])
  380. }
  381. func TestBug3File(t *testing.T) {
  382. file, status := fixtureFile()
  383. file.Update(0, 1, 0, 99)
  384. file.Update(0, 0, 1, 1)
  385. dump := file.Dump()
  386. assert.Equal(t, "0 0\n1 -1\n", dump)
  387. assert.Equal(t, int64(1), status[0])
  388. }
  389. func TestBug4File(t *testing.T) {
  390. status := map[int]int64{}
  391. file := NewFile(0, 10, NewStatus(status, updateStatusFile))
  392. file.Update(125, 0, 20, 9)
  393. file.Update(125, 0, 20, 20)
  394. file.Update(166, 12, 1, 1)
  395. file.Update(214, 2, 1, 1)
  396. file.Update(214, 4, 9, 0)
  397. file.Update(214, 27, 1, 1)
  398. file.Update(215, 3, 1, 1)
  399. file.Update(215, 13, 1, 1)
  400. file.Update(215, 17, 1, 1)
  401. file.Update(215, 19, 5, 0)
  402. file.Update(215, 25, 0, 1)
  403. file.Update(215, 31, 6, 1)
  404. file.Update(215, 27, 15, 0)
  405. file.Update(215, 2, 25, 4)
  406. file.Update(215, 28, 1, 1)
  407. file.Update(215, 30, 7, 2)
  408. file.Update(215, 38, 1, 0)
  409. file.Update(215, 40, 4, 2)
  410. file.Update(215, 46, 1, 0)
  411. file.Update(215, 49, 1, 0)
  412. file.Update(215, 52, 2, 6)
  413. dump := file.Dump()
  414. assert.Equal(t, "0 125\n2 215\n48 125\n50 215\n69 125\n73 215\n79 125\n80 0\n81 -1\n", dump)
  415. }
  416. func TestBug5File(t *testing.T) {
  417. status := map[int]int64{}
  418. keys := []int{0, 2, 4, 7, 10}
  419. vals := []int{24, 28, 24, 28, -1}
  420. file := NewFileFromTree(keys, vals, NewStatus(status, updateStatusFile))
  421. file.Update(28, 0, 1, 3)
  422. dump := file.Dump()
  423. assert.Equal(t, "0 28\n2 24\n5 28\n8 -1\n", dump)
  424. keys = []int{0, 1, 16, 18}
  425. vals = []int{305, 0, 157, -1}
  426. file = NewFileFromTree(keys, vals, NewStatus(status, updateStatusFile))
  427. file.Update(310, 0, 0, 2)
  428. dump = file.Dump()
  429. assert.Equal(t, "0 0\n14 157\n16 -1\n", dump)
  430. }
  431. func TestNewFileFromTreeInvalidSize(t *testing.T) {
  432. keys := [...]int{1, 2, 3}
  433. vals := [...]int{4, 5}
  434. assert.Panics(t, func() { NewFileFromTree(keys[:], vals[:]) })
  435. }
  436. func TestUpdatePanic(t *testing.T) {
  437. keys := [...]int{0}
  438. vals := [...]int{-1}
  439. file := NewFileFromTree(keys[:], vals[:])
  440. file.tree.DeleteWithKey(0)
  441. file.tree.Insert(rbtree.Item{Key: -1, Value: -1})
  442. var paniced interface{}
  443. func() {
  444. defer func() {
  445. paniced = recover()
  446. }()
  447. file.Update(1, 0, 1, 0)
  448. }()
  449. assert.Contains(t, paniced, "invalid tree state")
  450. }
  451. func TestFileStatus(t *testing.T) {
  452. f, _ := fixtureFile()
  453. assert.Panics(t, func() { f.Status(1) })
  454. assert.NotNil(t, f.Status(0))
  455. }
  456. func TestFileValidate(t *testing.T) {
  457. keys := [...]int{0}
  458. vals := [...]int{-1}
  459. file := NewFileFromTree(keys[:], vals[:])
  460. file.tree.DeleteWithKey(0)
  461. file.tree.Insert(rbtree.Item{Key: -1, Value: -1})
  462. assert.Panics(t, func() { file.Validate() })
  463. file.tree.DeleteWithKey(-1)
  464. file.tree.Insert(rbtree.Item{Key: 0, Value: -1})
  465. file.Validate()
  466. file.tree.DeleteWithKey(0)
  467. file.tree.Insert(rbtree.Item{Key: 0, Value: 0})
  468. assert.Panics(t, func() { file.Validate() })
  469. file.tree.DeleteWithKey(0)
  470. file.tree.Insert(rbtree.Item{Key: 0, Value: 1})
  471. file.tree.Insert(rbtree.Item{Key: 1, Value: 1})
  472. file.tree.Insert(rbtree.Item{Key: 2, Value: -1})
  473. file.Validate()
  474. file.tree.FindGE(2).Item().Key = 1
  475. assert.Panics(t, func() { file.Validate() })
  476. }
  477. func TestFileFlatten(t *testing.T) {
  478. file, _ := fixtureFile()
  479. // 0 0 | 100 -1 [0]: 100
  480. file.Update(1, 20, 30, 0)
  481. // 0 0 | 20 1 | 50 0 | 130 -1 [0]: 100, [1]: 30
  482. file.Update(2, 20, 0, 5)
  483. // 0 0 | 20 1 | 45 0 | 125 -1 [0]: 100, [1]: 25
  484. file.Update(3, 20, 0, 5)
  485. // 0 0 | 20 1 | 40 0 | 120 -1 [0]: 100, [1]: 20
  486. file.Update(4, 20, 10, 0)
  487. // 0 0 | 20 4 | 30 1 | 50 0 | 130 -1 [0]: 100, [1]: 20, [4]: 10
  488. lines := file.flatten()
  489. for i := 0; i < 20; i++ {
  490. assert.Equal(t, 0, lines[i], fmt.Sprintf("line %d", i))
  491. }
  492. for i := 20; i < 30; i++ {
  493. assert.Equal(t, 4, lines[i], fmt.Sprintf("line %d", i))
  494. }
  495. for i := 30; i < 50; i++ {
  496. assert.Equal(t, 1, lines[i], fmt.Sprintf("line %d", i))
  497. }
  498. for i := 50; i < 130; i++ {
  499. assert.Equal(t, 0, lines[i], fmt.Sprintf("line %d", i))
  500. }
  501. assert.Len(t, lines, 130)
  502. }
  503. func TestFileMergeMark(t *testing.T) {
  504. file, status := fixtureFile()
  505. // 0 0 | 100 -1 [0]: 100
  506. file.Update(1, 20, 30, 0)
  507. // 0 0 | 20 1 | 50 0 | 130 -1 [0]: 100, [1]: 30
  508. file.Update(2, 20, 0, 5)
  509. // 0 0 | 20 1 | 45 0 | 125 -1 [0]: 100, [1]: 25
  510. file.Update(3, 20, 0, 5)
  511. // 0 0 | 20 1 | 40 0 | 120 -1 [0]: 100, [1]: 20
  512. file.Update(4, 20, 10, 0)
  513. // 0 0 | 20 4 | 30 1 | 50 0 | 130 -1 [0]: 100, [1]: 20, [4]: 10
  514. file.Update(TreeMergeMark, 60, 20, 20)
  515. // 0 0 | 20 4 | 30 1 | 50 0 | 60 M | 80 0 | 130 -1
  516. // [0]: 100, [1]: 20, [4]: 10
  517. dump := file.Dump()
  518. assert.Equal(t, "0 0\n20 4\n30 1\n50 0\n60 16383\n80 0\n130 -1\n", dump)
  519. assert.Contains(t, status, 0)
  520. assert.Equal(t, int64(100), status[0])
  521. assert.Equal(t, int64(20), status[1])
  522. assert.Equal(t, int64(0), status[2])
  523. assert.Equal(t, int64(0), status[3])
  524. assert.Equal(t, int64(10), status[4])
  525. assert.NotContains(t, status, TreeMergeMark)
  526. }
  527. func TestFileMerge(t *testing.T) {
  528. file1, status := fixtureFile()
  529. // 0 0 | 100 -1 [0]: 100
  530. file1.Update(1, 20, 30, 0)
  531. // 0 0 | 20 1 | 50 0 | 130 -1 [0]: 100, [1]: 30
  532. file1.Update(2, 20, 0, 5)
  533. // 0 0 | 20 1 | 45 0 | 125 -1 [0]: 100, [1]: 25
  534. file1.Update(3, 20, 0, 5)
  535. // 0 0 | 20 1 | 40 0 | 120 -1 [0]: 100, [1]: 20
  536. file1.Update(4, 20, 10, 0)
  537. // 0 0 | 20 4 | 30 1 | 50 0 | 130 -1 [0]: 100, [1]: 20, [4]: 10
  538. file2 := file1.Clone(false)
  539. file1.Update(TreeMergeMark, 60, 30, 30)
  540. // 0 0 | 20 4 | 30 1 | 50 0 | 60 M | 90 0 | 130 -1
  541. // [0]: 70, [1]: 20, [4]: 10
  542. file2.Update(5, 60, 20, 20)
  543. // 0 0 | 20 4 | 30 1 | 50 0 | 60 5 | 80 0 | 130 -1
  544. // [0]: 80, [1]: 20, [4]: 10, [5]: 20
  545. file2.Update(TreeMergeMark, 80, 10, 10)
  546. // 0 0 | 20 4 | 30 1 | 50 0 | 60 5 | 80 M | 90 0 | 130 -1
  547. // [0]: 70, [1]: 20, [4]: 10, [5]: 20
  548. file2.Update(6, 0, 10, 10)
  549. // 0 6 | 10 0 | 20 4 | 30 1 | 50 0 | 60 5 | 80 M | 90 0 | 130 -1
  550. // [0]: 60, [1]: 20, [4]: 10, [5]: 20, [6]: 10
  551. file1.Merge(7, file2)
  552. // 0 0 | 20 4 | 30 1 | 50 0 | 60 5 | 80 7 | 90 0 | 130 -1
  553. // [0]: 70, [1]: 20, [4]: 10, [5]: 20, [6]: 0, [7]: 10
  554. dump := file1.Dump()
  555. assert.Equal(t, "0 0\n20 4\n30 1\n50 0\n60 5\n80 7\n90 0\n130 -1\n", dump)
  556. assert.Equal(t, int64(70), status[0])
  557. assert.Equal(t, int64(20), status[1])
  558. assert.Equal(t, int64(0), status[2])
  559. assert.Equal(t, int64(0), status[3])
  560. assert.Equal(t, int64(10), status[4])
  561. assert.Equal(t, int64(20), status[5])
  562. assert.Equal(t, int64(0), status[6])
  563. assert.Equal(t, int64(10), status[7])
  564. }