file_test.go 5.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. package hercules
  2. import (
  3. "testing"
  4. "github.com/stretchr/testify/assert"
  5. )
  6. func fixture() (*File, map[int]int64) {
  7. status := map[int]int64{}
  8. file := NewFile(0, 100, status)
  9. return file, status
  10. }
  11. func TestInitialize(t *testing.T) {
  12. file, status := fixture()
  13. dump := file.Dump()
  14. // Output:
  15. // 0 0
  16. // 100 -1
  17. assert.Equal(t, "0 0\n100 -1\n", dump)
  18. assert.Equal(t, int64(100), status[0])
  19. }
  20. func testPanic(t *testing.T, method func(*File), msg string) {
  21. defer func() {
  22. r := recover()
  23. assert.NotNil(t, r, "not panic()-ed")
  24. assert.IsType(t, "", r)
  25. assert.Contains(t, r.(string), msg)
  26. }()
  27. file, _ := fixture()
  28. method(file)
  29. }
  30. func TestBullshit(t *testing.T) {
  31. testPanic(t, func(file *File) { file.Update(1, -10, 10, 0) }, "insert")
  32. testPanic(t, func(file *File) { file.Update(1, 110, 10, 0) }, "insert")
  33. testPanic(t, func(file *File) { file.Update(1, -10, 0, 10) }, "delete")
  34. testPanic(t, func(file *File) { file.Update(1, 100, 0, 10) }, "delete")
  35. testPanic(t, func(file *File) { file.Update(1, 0, -10, 0) }, "length")
  36. testPanic(t, func(file *File) { file.Update(1, 0, 0, -10) }, "length")
  37. testPanic(t, func(file *File) { file.Update(1, 0, -10, -10) }, "length")
  38. testPanic(t, func(file *File) { file.Update(-1, 0, 10, 10) }, "time")
  39. file, status := fixture()
  40. file.Update(1, 10, 0, 0)
  41. assert.Equal(t, int64(100), status[0])
  42. assert.Equal(t, int64(0), status[1])
  43. }
  44. func TestInsert(t *testing.T) {
  45. file, status := fixture()
  46. file.Update(1, 10, 10, 0)
  47. dump := file.Dump()
  48. // Output:
  49. // 0 0
  50. // 10 1
  51. // 20 0
  52. // 110 -1
  53. assert.Equal(t, "0 0\n10 1\n20 0\n110 -1\n", dump)
  54. assert.Equal(t, int64(100), status[0])
  55. assert.Equal(t, int64(10), status[1])
  56. }
  57. func TestZeroInitialize(t *testing.T) {
  58. status := map[int]int64{}
  59. file := NewFile(0, 0, status)
  60. assert.NotContains(t, status, 0)
  61. dump := file.Dump()
  62. // Output:
  63. // 0 -1
  64. assert.Equal(t, "0 -1\n", dump)
  65. file.Update(1, 0, 10, 0)
  66. dump = file.Dump()
  67. // Output:
  68. // 0 1
  69. // 10 -1
  70. assert.Equal(t, "0 1\n10 -1\n", dump)
  71. assert.Equal(t, int64(10), status[1])
  72. }
  73. func TestDelete(t *testing.T) {
  74. file, status := fixture()
  75. file.Update(1, 10, 0, 10)
  76. dump := file.Dump()
  77. // Output:
  78. // 0 0
  79. // 90 -1
  80. assert.Equal(t, "0 0\n90 -1\n", dump)
  81. assert.Equal(t, int64(90), status[0])
  82. assert.Equal(t, int64(0), status[1])
  83. }
  84. func TestFused(t *testing.T) {
  85. file, status := fixture()
  86. file.Update(1, 10, 6, 7)
  87. dump := file.Dump()
  88. // Output:
  89. // 0 0
  90. // 10 1
  91. // 16 0
  92. // 99 -1
  93. assert.Equal(t, "0 0\n10 1\n16 0\n99 -1\n", dump)
  94. assert.Equal(t, int64(93), status[0])
  95. assert.Equal(t, int64(6), status[1])
  96. }
  97. func TestInsertSameStart(t *testing.T) {
  98. file, status := fixture()
  99. file.Update(1, 10, 10, 0)
  100. file.Update(2, 10, 10, 0)
  101. dump := file.Dump()
  102. // Output:
  103. // 0 0
  104. // 10 2
  105. // 20 1
  106. // 30 0
  107. // 120 -1
  108. assert.Equal(t, "0 0\n10 2\n20 1\n30 0\n120 -1\n", dump)
  109. assert.Equal(t, int64(100), status[0])
  110. assert.Equal(t, int64(10), status[1])
  111. assert.Equal(t, int64(10), status[2])
  112. }
  113. func TestInsertEnd(t *testing.T) {
  114. file, status := fixture()
  115. file.Update(1, 100, 10, 0)
  116. dump := file.Dump()
  117. // Output:
  118. // 0 0
  119. // 100 1
  120. // 110 -1
  121. assert.Equal(t, "0 0\n100 1\n110 -1\n", dump)
  122. assert.Equal(t, int64(100), status[0])
  123. assert.Equal(t, int64(10), status[1])
  124. }
  125. func TestDeleteSameStart0(t *testing.T) {
  126. file, status := fixture()
  127. file.Update(1, 0, 0, 10)
  128. dump := file.Dump()
  129. // Output:
  130. // 0 0
  131. // 90 -1
  132. assert.Equal(t, "0 0\n90 -1\n", dump)
  133. assert.Equal(t, int64(90), status[0])
  134. assert.Equal(t, int64(0), status[1])
  135. }
  136. func TestDeleteSameStartMiddle(t *testing.T) {
  137. file, status := fixture()
  138. file.Update(1, 10, 10, 0)
  139. file.Update(2, 10, 0, 5)
  140. dump := file.Dump()
  141. // Output:
  142. // 0 0
  143. // 10 1
  144. // 15 0
  145. // 105 -1
  146. assert.Equal(t, "0 0\n10 1\n15 0\n105 -1\n", dump)
  147. assert.Equal(t, int64(100), status[0])
  148. assert.Equal(t, int64(5), status[1])
  149. }
  150. func TestDeleteIntersection(t *testing.T) {
  151. file, status := fixture()
  152. file.Update(1, 10, 10, 0)
  153. file.Update(2, 15, 0, 10)
  154. dump := file.Dump()
  155. // Output:
  156. // 0 0
  157. // 10 1
  158. // 15 0
  159. // 100 -1
  160. assert.Equal(t, "0 0\n10 1\n15 0\n100 -1\n", dump)
  161. assert.Equal(t, int64(95), status[0])
  162. assert.Equal(t, int64(5), status[1])
  163. }
  164. func TestDeleteAll(t *testing.T) {
  165. file, status := fixture()
  166. file.Update(1, 0, 0, 100)
  167. // Output:
  168. // 0 -1
  169. dump := file.Dump()
  170. assert.Equal(t, "0 -1\n", dump)
  171. assert.Equal(t, int64(0), status[0])
  172. assert.Equal(t, int64(0), status[1])
  173. }
  174. func TestFusedIntersection(t *testing.T) {
  175. file, status := fixture()
  176. file.Update(1, 10, 10, 0)
  177. file.Update(2, 15, 3, 10)
  178. dump := file.Dump()
  179. // Output:
  180. // 0 0
  181. // 10 1
  182. // 15 2
  183. // 18 0
  184. // 103 -1
  185. assert.Equal(t, "0 0\n10 1\n15 2\n18 0\n103 -1\n", dump)
  186. assert.Equal(t, int64(95), status[0])
  187. assert.Equal(t, int64(5), status[1])
  188. assert.Equal(t, int64(3), status[2])
  189. }
  190. func TestTorture(t *testing.T) {
  191. file, status := fixture()
  192. // 0 0 | 100 -1 [0]: 100
  193. file.Update(1, 20, 30, 0)
  194. // 0 0 | 20 1 | 50 0 | 130 -1 [0]: 100, [1]: 30
  195. file.Update(2, 20, 0, 5)
  196. // 0 0 | 20 1 | 45 0 | 125 -1 [0]: 100, [1]: 25
  197. file.Update(3, 20, 0, 5)
  198. // 0 0 | 20 1 | 40 0 | 120 -1 [0]: 100, [1]: 20
  199. file.Update(4, 20, 10, 0)
  200. // 0 0 | 20 4 | 30 1 | 50 0 | 130 -1 [0]: 100, [1]: 20, [4]: 10
  201. file.Update(5, 45, 0, 10)
  202. // 0 0 | 20 4 | 30 1 | 45 0 | 120 -1 [0]: 95, [1]: 15, [4]: 10
  203. file.Update(6, 45, 5, 0)
  204. // 0 0 | 20 4 | 30 1 | 45 6 | 50 0 | 125 -1 [0]: 95, [1]: 15, [4]: 10, [6]: 5
  205. file.Update(7, 10, 0, 50)
  206. // 0 0 | 75 -1 [0]: 75
  207. file.Update(8, 0, 10, 10)
  208. // 0 8 | 10 0 | 75 -1 [0]: 65, [8]: 10
  209. dump := file.Dump()
  210. assert.Equal(t, "0 8\n10 0\n75 -1\n", dump)
  211. assert.Equal(t, int64(65), status[0])
  212. assert.Equal(t, int64(0), status[1])
  213. assert.Equal(t, int64(0), status[2])
  214. assert.Equal(t, int64(0), status[3])
  215. assert.Equal(t, int64(0), status[4])
  216. assert.Equal(t, int64(0), status[5])
  217. assert.Equal(t, int64(0), status[6])
  218. assert.Equal(t, int64(0), status[7])
  219. assert.Equal(t, int64(10), status[8])
  220. }