burndown_test.go 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086
  1. package leaves
  2. import (
  3. "bytes"
  4. "io"
  5. "io/ioutil"
  6. "path"
  7. "testing"
  8. "gopkg.in/src-d/hercules.v4/internal/core"
  9. "gopkg.in/src-d/hercules.v4/internal/test/fixtures"
  10. "github.com/gogo/protobuf/proto"
  11. "github.com/stretchr/testify/assert"
  12. "gopkg.in/src-d/go-git.v4/plumbing"
  13. "gopkg.in/src-d/go-git.v4/plumbing/object"
  14. "gopkg.in/src-d/hercules.v4/internal/pb"
  15. items "gopkg.in/src-d/hercules.v4/internal/plumbing"
  16. "gopkg.in/src-d/hercules.v4/internal/plumbing/identity"
  17. "gopkg.in/src-d/hercules.v4/internal/test"
  18. )
  19. func TestBurndownMeta(t *testing.T) {
  20. burndown := BurndownAnalysis{}
  21. assert.Equal(t, burndown.Name(), "Burndown")
  22. assert.Equal(t, len(burndown.Provides()), 0)
  23. required := [...]string{
  24. items.DependencyFileDiff, items.DependencyTreeChanges, items.DependencyBlobCache,
  25. items.DependencyDay, identity.DependencyAuthor}
  26. for _, name := range required {
  27. assert.Contains(t, burndown.Requires(), name)
  28. }
  29. opts := burndown.ListConfigurationOptions()
  30. matches := 0
  31. for _, opt := range opts {
  32. switch opt.Name {
  33. case ConfigBurndownGranularity, ConfigBurndownSampling, ConfigBurndownTrackFiles,
  34. ConfigBurndownTrackPeople, ConfigBurndownDebug:
  35. matches++
  36. }
  37. }
  38. assert.Len(t, opts, matches)
  39. assert.Equal(t, burndown.Flag(), "burndown")
  40. }
  41. func TestBurndownConfigure(t *testing.T) {
  42. burndown := BurndownAnalysis{}
  43. facts := map[string]interface{}{}
  44. facts[ConfigBurndownGranularity] = 100
  45. facts[ConfigBurndownSampling] = 200
  46. facts[ConfigBurndownTrackFiles] = true
  47. facts[ConfigBurndownTrackPeople] = true
  48. facts[ConfigBurndownDebug] = true
  49. facts[identity.FactIdentityDetectorPeopleCount] = 5
  50. facts[identity.FactIdentityDetectorReversedPeopleDict] = burndown.Requires()
  51. burndown.Configure(facts)
  52. assert.Equal(t, burndown.Granularity, 100)
  53. assert.Equal(t, burndown.Sampling, 200)
  54. assert.Equal(t, burndown.TrackFiles, true)
  55. assert.Equal(t, burndown.PeopleNumber, 5)
  56. assert.Equal(t, burndown.Debug, true)
  57. assert.Equal(t, burndown.reversedPeopleDict, burndown.Requires())
  58. facts[ConfigBurndownTrackPeople] = false
  59. facts[identity.FactIdentityDetectorPeopleCount] = 50
  60. burndown.Configure(facts)
  61. assert.Equal(t, burndown.PeopleNumber, 0)
  62. facts = map[string]interface{}{}
  63. burndown.Configure(facts)
  64. assert.Equal(t, burndown.Granularity, 100)
  65. assert.Equal(t, burndown.Sampling, 200)
  66. assert.Equal(t, burndown.TrackFiles, true)
  67. assert.Equal(t, burndown.PeopleNumber, 0)
  68. assert.Equal(t, burndown.Debug, true)
  69. assert.Equal(t, burndown.reversedPeopleDict, burndown.Requires())
  70. }
  71. func TestBurndownRegistration(t *testing.T) {
  72. summoned := core.Registry.Summon((&BurndownAnalysis{}).Name())
  73. assert.Len(t, summoned, 1)
  74. assert.Equal(t, summoned[0].Name(), "Burndown")
  75. leaves := core.Registry.GetLeaves()
  76. matched := false
  77. for _, tp := range leaves {
  78. if tp.Flag() == (&BurndownAnalysis{}).Flag() {
  79. matched = true
  80. break
  81. }
  82. }
  83. assert.True(t, matched)
  84. }
  85. func TestBurndownInitialize(t *testing.T) {
  86. burndown := BurndownAnalysis{}
  87. burndown.Sampling = -10
  88. burndown.Granularity = DefaultBurndownGranularity
  89. burndown.Initialize(test.Repository)
  90. assert.Equal(t, burndown.Sampling, DefaultBurndownGranularity)
  91. assert.Equal(t, burndown.Granularity, DefaultBurndownGranularity)
  92. burndown.Sampling = 0
  93. burndown.Granularity = DefaultBurndownGranularity - 1
  94. burndown.Initialize(test.Repository)
  95. assert.Equal(t, burndown.Sampling, DefaultBurndownGranularity-1)
  96. assert.Equal(t, burndown.Granularity, DefaultBurndownGranularity-1)
  97. burndown.Sampling = DefaultBurndownGranularity - 1
  98. burndown.Granularity = -10
  99. burndown.Initialize(test.Repository)
  100. assert.Equal(t, burndown.Sampling, DefaultBurndownGranularity-1)
  101. assert.Equal(t, burndown.Granularity, DefaultBurndownGranularity)
  102. }
  103. func TestBurndownConsumeFinalize(t *testing.T) {
  104. burndown := BurndownAnalysis{
  105. Granularity: 30,
  106. Sampling: 30,
  107. PeopleNumber: 2,
  108. TrackFiles: true,
  109. }
  110. burndown.Initialize(test.Repository)
  111. deps := map[string]interface{}{}
  112. // stage 1
  113. deps[identity.DependencyAuthor] = 0
  114. deps[items.DependencyDay] = 0
  115. cache := map[plumbing.Hash]*object.Blob{}
  116. hash := plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe")
  117. cache[hash], _ = test.Repository.BlobObject(hash)
  118. hash = plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9")
  119. cache[hash], _ = test.Repository.BlobObject(hash)
  120. hash = plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1")
  121. cache[hash], _ = test.Repository.BlobObject(hash)
  122. hash = plumbing.NewHash("dc248ba2b22048cc730c571a748e8ffcf7085ab9")
  123. cache[hash], _ = test.Repository.BlobObject(hash)
  124. deps[items.DependencyBlobCache] = cache
  125. changes := make(object.Changes, 3)
  126. treeFrom, _ := test.Repository.TreeObject(plumbing.NewHash(
  127. "a1eb2ea76eb7f9bfbde9b243861474421000eb96"))
  128. treeTo, _ := test.Repository.TreeObject(plumbing.NewHash(
  129. "994eac1cd07235bb9815e547a75c84265dea00f5"))
  130. changes[0] = &object.Change{From: object.ChangeEntry{
  131. Name: "analyser.go",
  132. Tree: treeFrom,
  133. TreeEntry: object.TreeEntry{
  134. Name: "analyser.go",
  135. Mode: 0100644,
  136. Hash: plumbing.NewHash("dc248ba2b22048cc730c571a748e8ffcf7085ab9"),
  137. },
  138. }, To: object.ChangeEntry{
  139. Name: "analyser.go",
  140. Tree: treeTo,
  141. TreeEntry: object.TreeEntry{
  142. Name: "analyser.go",
  143. Mode: 0100644,
  144. Hash: plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1"),
  145. },
  146. }}
  147. changes[1] = &object.Change{From: object.ChangeEntry{}, To: object.ChangeEntry{
  148. Name: "cmd/hercules/main.go",
  149. Tree: treeTo,
  150. TreeEntry: object.TreeEntry{
  151. Name: "cmd/hercules/main.go",
  152. Mode: 0100644,
  153. Hash: plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9"),
  154. },
  155. },
  156. }
  157. changes[2] = &object.Change{From: object.ChangeEntry{}, To: object.ChangeEntry{
  158. Name: ".travis.yml",
  159. Tree: treeTo,
  160. TreeEntry: object.TreeEntry{
  161. Name: ".travis.yml",
  162. Mode: 0100644,
  163. Hash: plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe"),
  164. },
  165. },
  166. }
  167. deps[items.DependencyTreeChanges] = changes
  168. fd := fixtures.FileDiff()
  169. result, err := fd.Consume(deps)
  170. assert.Nil(t, err)
  171. deps[items.DependencyFileDiff] = result[items.DependencyFileDiff]
  172. deps[core.DependencyCommit], _ = test.Repository.CommitObject(plumbing.NewHash(
  173. "cce947b98a050c6d356bc6ba95030254914027b1"))
  174. result, err = burndown.Consume(deps)
  175. assert.Nil(t, result)
  176. assert.Nil(t, err)
  177. assert.Equal(t, burndown.previousDay, 0)
  178. assert.Equal(t, len(burndown.files), 3)
  179. assert.Equal(t, burndown.files["cmd/hercules/main.go"].Len(), 207)
  180. assert.Equal(t, burndown.files["analyser.go"].Len(), 926)
  181. assert.Equal(t, burndown.files[".travis.yml"].Len(), 12)
  182. assert.Equal(t, len(burndown.people), 2)
  183. assert.Equal(t, burndown.people[0][0], int64(12+207+926))
  184. assert.Equal(t, len(burndown.globalStatus), 1)
  185. assert.Equal(t, burndown.globalStatus[0], int64(12+207+926))
  186. assert.Equal(t, len(burndown.globalHistory), 0)
  187. assert.Equal(t, len(burndown.fileHistories), 0)
  188. burndown2 := BurndownAnalysis{
  189. Granularity: 30,
  190. Sampling: 0,
  191. }
  192. burndown2.Initialize(test.Repository)
  193. _, err = burndown2.Consume(deps)
  194. assert.Nil(t, err)
  195. assert.Equal(t, len(burndown2.people), 0)
  196. assert.Equal(t, len(burndown2.peopleHistories), 0)
  197. assert.Equal(t, len(burndown2.fileHistories), 0)
  198. // stage 2
  199. // 2b1ed978194a94edeabbca6de7ff3b5771d4d665
  200. deps[identity.DependencyAuthor] = 1
  201. deps[items.DependencyDay] = 30
  202. cache = map[plumbing.Hash]*object.Blob{}
  203. hash = plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe")
  204. cache[hash], _ = test.Repository.BlobObject(hash)
  205. hash = plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1")
  206. cache[hash], _ = test.Repository.BlobObject(hash)
  207. hash = plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2")
  208. cache[hash], _ = test.Repository.BlobObject(hash)
  209. hash = plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9")
  210. cache[hash], _ = test.Repository.BlobObject(hash)
  211. hash = plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72")
  212. cache[hash], _ = test.Repository.BlobObject(hash)
  213. deps[items.DependencyBlobCache] = cache
  214. changes = make(object.Changes, 3)
  215. treeFrom, _ = test.Repository.TreeObject(plumbing.NewHash(
  216. "96c6ece9b2f3c7c51b83516400d278dea5605100"))
  217. treeTo, _ = test.Repository.TreeObject(plumbing.NewHash(
  218. "251f2094d7b523d5bcc60e663b6cf38151bf8844"))
  219. changes[0] = &object.Change{From: object.ChangeEntry{
  220. Name: "analyser.go",
  221. Tree: treeFrom,
  222. TreeEntry: object.TreeEntry{
  223. Name: "analyser.go",
  224. Mode: 0100644,
  225. Hash: plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1"),
  226. },
  227. }, To: object.ChangeEntry{
  228. Name: "burndown.go",
  229. Tree: treeTo,
  230. TreeEntry: object.TreeEntry{
  231. Name: "burndown.go",
  232. Mode: 0100644,
  233. Hash: plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2"),
  234. },
  235. },
  236. }
  237. changes[1] = &object.Change{From: object.ChangeEntry{
  238. Name: "cmd/hercules/main.go",
  239. Tree: treeFrom,
  240. TreeEntry: object.TreeEntry{
  241. Name: "cmd/hercules/main.go",
  242. Mode: 0100644,
  243. Hash: plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9"),
  244. },
  245. }, To: object.ChangeEntry{
  246. Name: "cmd/hercules/main.go",
  247. Tree: treeTo,
  248. TreeEntry: object.TreeEntry{
  249. Name: "cmd/hercules/main.go",
  250. Mode: 0100644,
  251. Hash: plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72"),
  252. },
  253. },
  254. }
  255. changes[2] = &object.Change{From: object.ChangeEntry{
  256. Name: ".travis.yml",
  257. Tree: treeTo,
  258. TreeEntry: object.TreeEntry{
  259. Name: ".travis.yml",
  260. Mode: 0100644,
  261. Hash: plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe"),
  262. },
  263. }, To: object.ChangeEntry{},
  264. }
  265. deps[items.DependencyTreeChanges] = changes
  266. fd = fixtures.FileDiff()
  267. result, err = fd.Consume(deps)
  268. assert.Nil(t, err)
  269. deps[items.DependencyFileDiff] = result[items.DependencyFileDiff]
  270. result, err = burndown.Consume(deps)
  271. assert.Nil(t, result)
  272. assert.Nil(t, err)
  273. assert.Equal(t, burndown.previousDay, 30)
  274. assert.Equal(t, len(burndown.files), 2)
  275. assert.Equal(t, burndown.files["cmd/hercules/main.go"].Len(), 290)
  276. assert.Equal(t, burndown.files["burndown.go"].Len(), 543)
  277. assert.Equal(t, len(burndown.people), 2)
  278. assert.Equal(t, len(burndown.globalStatus), 2)
  279. assert.Equal(t, burndown.globalStatus[0], int64(464))
  280. assert.Equal(t, burndown.globalStatus[1], int64(0))
  281. assert.Equal(t, len(burndown.globalHistory), 1)
  282. assert.Equal(t, len(burndown.globalHistory[0]), 2)
  283. assert.Equal(t, len(burndown.fileHistories), 3)
  284. out := burndown.Finalize().(BurndownResult)
  285. /*
  286. GlobalHistory [][]int64
  287. FileHistories map[string][][]int64
  288. PeopleHistories [][][]int64
  289. PeopleMatrix [][]int64
  290. */
  291. assert.Equal(t, len(out.GlobalHistory), 2)
  292. for i := 0; i < 2; i++ {
  293. assert.Equal(t, len(out.GlobalHistory[i]), 2)
  294. }
  295. assert.Equal(t, len(out.GlobalHistory), 2)
  296. assert.Equal(t, out.GlobalHistory[0][0], int64(1145))
  297. assert.Equal(t, out.GlobalHistory[0][1], int64(0))
  298. assert.Equal(t, out.GlobalHistory[1][0], int64(464))
  299. assert.Equal(t, out.GlobalHistory[1][1], int64(369))
  300. assert.Equal(t, len(out.FileHistories), 2)
  301. assert.Equal(t, len(out.FileHistories["cmd/hercules/main.go"]), 2)
  302. assert.Equal(t, len(out.FileHistories["burndown.go"]), 2)
  303. assert.Equal(t, len(out.FileHistories["cmd/hercules/main.go"][0]), 2)
  304. assert.Equal(t, len(out.FileHistories["burndown.go"][0]), 2)
  305. assert.Equal(t, len(out.PeopleMatrix), 2)
  306. assert.Equal(t, len(out.PeopleMatrix[0]), 4)
  307. assert.Equal(t, len(out.PeopleMatrix[1]), 4)
  308. assert.Equal(t, out.PeopleMatrix[0][0], int64(1145))
  309. assert.Equal(t, out.PeopleMatrix[0][1], int64(0))
  310. assert.Equal(t, out.PeopleMatrix[0][2], int64(0))
  311. assert.Equal(t, out.PeopleMatrix[0][3], int64(-681))
  312. assert.Equal(t, out.PeopleMatrix[1][0], int64(369))
  313. assert.Equal(t, out.PeopleMatrix[1][1], int64(0))
  314. assert.Equal(t, out.PeopleMatrix[1][2], int64(0))
  315. assert.Equal(t, out.PeopleMatrix[1][3], int64(0))
  316. assert.Equal(t, len(out.PeopleHistories), 2)
  317. for i := 0; i < 2; i++ {
  318. assert.Equal(t, len(out.PeopleHistories[i]), 2)
  319. assert.Equal(t, len(out.PeopleHistories[i][0]), 2)
  320. assert.Equal(t, len(out.PeopleHistories[i][1]), 2)
  321. }
  322. }
  323. func TestBurndownSerialize(t *testing.T) {
  324. burndown := BurndownAnalysis{
  325. Granularity: 30,
  326. Sampling: 30,
  327. PeopleNumber: 2,
  328. TrackFiles: true,
  329. }
  330. burndown.Initialize(test.Repository)
  331. deps := map[string]interface{}{}
  332. // stage 1
  333. deps[identity.DependencyAuthor] = 0
  334. deps[items.DependencyDay] = 0
  335. cache := map[plumbing.Hash]*object.Blob{}
  336. hash := plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe")
  337. cache[hash], _ = test.Repository.BlobObject(hash)
  338. hash = plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9")
  339. cache[hash], _ = test.Repository.BlobObject(hash)
  340. hash = plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1")
  341. cache[hash], _ = test.Repository.BlobObject(hash)
  342. hash = plumbing.NewHash("dc248ba2b22048cc730c571a748e8ffcf7085ab9")
  343. cache[hash], _ = test.Repository.BlobObject(hash)
  344. deps[items.DependencyBlobCache] = cache
  345. changes := make(object.Changes, 3)
  346. treeFrom, _ := test.Repository.TreeObject(plumbing.NewHash(
  347. "a1eb2ea76eb7f9bfbde9b243861474421000eb96"))
  348. treeTo, _ := test.Repository.TreeObject(plumbing.NewHash(
  349. "994eac1cd07235bb9815e547a75c84265dea00f5"))
  350. changes[0] = &object.Change{From: object.ChangeEntry{
  351. Name: "analyser.go",
  352. Tree: treeFrom,
  353. TreeEntry: object.TreeEntry{
  354. Name: "analyser.go",
  355. Mode: 0100644,
  356. Hash: plumbing.NewHash("dc248ba2b22048cc730c571a748e8ffcf7085ab9"),
  357. },
  358. }, To: object.ChangeEntry{
  359. Name: "analyser.go",
  360. Tree: treeTo,
  361. TreeEntry: object.TreeEntry{
  362. Name: "analyser.go",
  363. Mode: 0100644,
  364. Hash: plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1"),
  365. },
  366. }}
  367. changes[1] = &object.Change{From: object.ChangeEntry{}, To: object.ChangeEntry{
  368. Name: "cmd/hercules/main.go",
  369. Tree: treeTo,
  370. TreeEntry: object.TreeEntry{
  371. Name: "cmd/hercules/main.go",
  372. Mode: 0100644,
  373. Hash: plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9"),
  374. },
  375. },
  376. }
  377. changes[2] = &object.Change{From: object.ChangeEntry{}, To: object.ChangeEntry{
  378. Name: ".travis.yml",
  379. Tree: treeTo,
  380. TreeEntry: object.TreeEntry{
  381. Name: ".travis.yml",
  382. Mode: 0100644,
  383. Hash: plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe"),
  384. },
  385. },
  386. }
  387. deps[items.DependencyTreeChanges] = changes
  388. deps[core.DependencyCommit], _ = test.Repository.CommitObject(plumbing.NewHash(
  389. "cce947b98a050c6d356bc6ba95030254914027b1"))
  390. fd := fixtures.FileDiff()
  391. result, _ := fd.Consume(deps)
  392. deps[items.DependencyFileDiff] = result[items.DependencyFileDiff]
  393. burndown.Consume(deps)
  394. // stage 2
  395. // 2b1ed978194a94edeabbca6de7ff3b5771d4d665
  396. deps[identity.DependencyAuthor] = 1
  397. deps[items.DependencyDay] = 30
  398. cache = map[plumbing.Hash]*object.Blob{}
  399. hash = plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe")
  400. cache[hash], _ = test.Repository.BlobObject(hash)
  401. hash = plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1")
  402. cache[hash], _ = test.Repository.BlobObject(hash)
  403. hash = plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2")
  404. cache[hash], _ = test.Repository.BlobObject(hash)
  405. hash = plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9")
  406. cache[hash], _ = test.Repository.BlobObject(hash)
  407. hash = plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72")
  408. cache[hash], _ = test.Repository.BlobObject(hash)
  409. deps[items.DependencyBlobCache] = cache
  410. changes = make(object.Changes, 3)
  411. treeFrom, _ = test.Repository.TreeObject(plumbing.NewHash(
  412. "96c6ece9b2f3c7c51b83516400d278dea5605100"))
  413. treeTo, _ = test.Repository.TreeObject(plumbing.NewHash(
  414. "251f2094d7b523d5bcc60e663b6cf38151bf8844"))
  415. changes[0] = &object.Change{From: object.ChangeEntry{
  416. Name: "analyser.go",
  417. Tree: treeFrom,
  418. TreeEntry: object.TreeEntry{
  419. Name: "analyser.go",
  420. Mode: 0100644,
  421. Hash: plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1"),
  422. },
  423. }, To: object.ChangeEntry{
  424. Name: "burndown.go",
  425. Tree: treeTo,
  426. TreeEntry: object.TreeEntry{
  427. Name: "burndown.go",
  428. Mode: 0100644,
  429. Hash: plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2"),
  430. },
  431. },
  432. }
  433. changes[1] = &object.Change{From: object.ChangeEntry{
  434. Name: "cmd/hercules/main.go",
  435. Tree: treeFrom,
  436. TreeEntry: object.TreeEntry{
  437. Name: "cmd/hercules/main.go",
  438. Mode: 0100644,
  439. Hash: plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9"),
  440. },
  441. }, To: object.ChangeEntry{
  442. Name: "cmd/hercules/main.go",
  443. Tree: treeTo,
  444. TreeEntry: object.TreeEntry{
  445. Name: "cmd/hercules/main.go",
  446. Mode: 0100644,
  447. Hash: plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72"),
  448. },
  449. },
  450. }
  451. changes[2] = &object.Change{From: object.ChangeEntry{
  452. Name: ".travis.yml",
  453. Tree: treeTo,
  454. TreeEntry: object.TreeEntry{
  455. Name: ".travis.yml",
  456. Mode: 0100644,
  457. Hash: plumbing.NewHash("291286b4ac41952cbd1389fda66420ec03c1a9fe"),
  458. },
  459. }, To: object.ChangeEntry{},
  460. }
  461. deps[items.DependencyTreeChanges] = changes
  462. fd = fixtures.FileDiff()
  463. result, _ = fd.Consume(deps)
  464. deps[items.DependencyFileDiff] = result[items.DependencyFileDiff]
  465. people := [...]string{"one@srcd", "two@srcd"}
  466. burndown.reversedPeopleDict = people[:]
  467. burndown.Consume(deps)
  468. out := burndown.Finalize().(BurndownResult)
  469. buffer := &bytes.Buffer{}
  470. burndown.Serialize(out, false, buffer)
  471. assert.Equal(t, buffer.String(), ` granularity: 30
  472. sampling: 30
  473. "project": |-
  474. 1145 0
  475. 464 369
  476. files:
  477. "burndown.go": |-
  478. 0 0
  479. 293 250
  480. "cmd/hercules/main.go": |-
  481. 207 0
  482. 171 119
  483. people_sequence:
  484. - "one@srcd"
  485. - "two@srcd"
  486. people:
  487. "one@srcd": |-
  488. 1145 0
  489. 464 0
  490. "two@srcd": |-
  491. 0 0
  492. 0 369
  493. people_interaction: |-
  494. 1145 0 0 -681
  495. 369 0 0 0
  496. `)
  497. buffer = &bytes.Buffer{}
  498. burndown.Serialize(out, true, buffer)
  499. msg := pb.BurndownAnalysisResults{}
  500. proto.Unmarshal(buffer.Bytes(), &msg)
  501. assert.Equal(t, msg.Granularity, int32(30))
  502. assert.Equal(t, msg.Sampling, int32(30))
  503. assert.Equal(t, msg.Project.Name, "project")
  504. assert.Equal(t, msg.Project.NumberOfRows, int32(2))
  505. assert.Equal(t, msg.Project.NumberOfColumns, int32(2))
  506. assert.Len(t, msg.Project.Rows, 2)
  507. assert.Len(t, msg.Project.Rows[0].Columns, 1)
  508. assert.Equal(t, msg.Project.Rows[0].Columns[0], uint32(1145))
  509. assert.Len(t, msg.Project.Rows[1].Columns, 2)
  510. assert.Equal(t, msg.Project.Rows[1].Columns[0], uint32(464))
  511. assert.Equal(t, msg.Project.Rows[1].Columns[1], uint32(369))
  512. assert.Len(t, msg.Files, 2)
  513. assert.Equal(t, msg.Files[0].Name, "burndown.go")
  514. assert.Equal(t, msg.Files[1].Name, "cmd/hercules/main.go")
  515. assert.Len(t, msg.Files[0].Rows, 2)
  516. assert.Len(t, msg.Files[0].Rows[0].Columns, 0)
  517. assert.Len(t, msg.Files[0].Rows[1].Columns, 2)
  518. assert.Equal(t, msg.Files[0].Rows[1].Columns[0], uint32(293))
  519. assert.Equal(t, msg.Files[0].Rows[1].Columns[1], uint32(250))
  520. assert.Len(t, msg.People, 2)
  521. assert.Equal(t, msg.People[0].Name, "one@srcd")
  522. assert.Equal(t, msg.People[1].Name, "two@srcd")
  523. assert.Len(t, msg.People[0].Rows, 2)
  524. assert.Len(t, msg.People[0].Rows[0].Columns, 1)
  525. assert.Len(t, msg.People[0].Rows[1].Columns, 1)
  526. assert.Equal(t, msg.People[0].Rows[0].Columns[0], uint32(1145))
  527. assert.Equal(t, msg.People[0].Rows[1].Columns[0], uint32(464))
  528. assert.Len(t, msg.People[1].Rows, 2)
  529. assert.Len(t, msg.People[1].Rows[0].Columns, 0)
  530. assert.Len(t, msg.People[1].Rows[1].Columns, 2)
  531. assert.Equal(t, msg.People[1].Rows[1].Columns[0], uint32(0))
  532. assert.Equal(t, msg.People[1].Rows[1].Columns[1], uint32(369))
  533. assert.Equal(t, msg.PeopleInteraction.NumberOfRows, int32(2))
  534. assert.Equal(t, msg.PeopleInteraction.NumberOfColumns, int32(4))
  535. data := [...]int64{1145, -681, 369}
  536. assert.Equal(t, msg.PeopleInteraction.Data, data[:])
  537. indices := [...]int32{0, 3, 0}
  538. assert.Equal(t, msg.PeopleInteraction.Indices, indices[:])
  539. indptr := [...]int64{0, 2, 3}
  540. assert.Equal(t, msg.PeopleInteraction.Indptr, indptr[:])
  541. }
  542. type panickingCloser struct {
  543. }
  544. func (c panickingCloser) Close() error {
  545. return io.EOF
  546. }
  547. func TestCheckClose(t *testing.T) {
  548. closer := panickingCloser{}
  549. assert.Panics(t, func() { checkClose(closer) })
  550. }
  551. func TestBurndownAddMatrix(t *testing.T) {
  552. size := 5*3 + 1
  553. daily := make([][]float32, size)
  554. for i := range daily {
  555. daily[i] = make([]float32, size)
  556. }
  557. added := make([][]int64, 5)
  558. for i := range added {
  559. added[i] = make([]int64, 3)
  560. switch i {
  561. case 0:
  562. added[i][0] = 10
  563. case 1:
  564. added[i][0] = 18
  565. added[i][1] = 2
  566. case 2:
  567. added[i][0] = 12
  568. added[i][1] = 14
  569. case 3:
  570. added[i][0] = 10
  571. added[i][1] = 12
  572. added[i][2] = 6
  573. case 4:
  574. added[i][0] = 8
  575. added[i][1] = 9
  576. added[i][2] = 13
  577. }
  578. }
  579. assert.Panics(t, func() {
  580. daily2 := make([][]float32, 16)
  581. for i := range daily2 {
  582. daily2[i] = make([]float32, 15)
  583. }
  584. addBurndownMatrix(added, 5, 3, daily2, 1)
  585. })
  586. assert.Panics(t, func() {
  587. daily2 := make([][]float32, 15)
  588. for i := range daily2 {
  589. daily2[i] = make([]float32, 16)
  590. }
  591. addBurndownMatrix(added, 5, 3, daily2, 1)
  592. })
  593. // yaml.PrintMatrix(os.Stdout, added, 0, "test", true)
  594. /*
  595. "test": |-
  596. 10 0 0
  597. 18 2 0
  598. 12 14 0
  599. 10 12 6
  600. 8 9 13
  601. */
  602. addBurndownMatrix(added, 5, 3, daily, 1)
  603. for i := range daily[0] {
  604. assert.Equal(t, daily[0][i], float32(0))
  605. }
  606. for i := range daily {
  607. assert.Equal(t, daily[i][0], float32(0))
  608. }
  609. /*for _, row := range daily {
  610. fmt.Println(row)
  611. }*/
  612. // check pinned points
  613. for y := 0; y < 5; y++ {
  614. for x := 0; x < 3; x++ {
  615. var sum float32
  616. for i := x * 5; i < (x+1)*5; i++ {
  617. sum += daily[(y+1)*3][i+1]
  618. }
  619. assert.InDelta(t, sum, added[y][x], 0.00001)
  620. }
  621. }
  622. // check overall trend: 0 -> const -> peak -> decay
  623. for x := 0; x < 15; x++ {
  624. for y := 0; y < x; y++ {
  625. assert.Zero(t, daily[y+1][x+1])
  626. }
  627. var prev float32
  628. for y := x; y < ((x+3)/5)*5; y++ {
  629. if prev == 0 {
  630. prev = daily[y+1][x+1]
  631. }
  632. assert.Equal(t, daily[y+1][x+1], prev)
  633. }
  634. for y := ((x + 3) / 5) * 5; y < 15; y++ {
  635. if prev == 0 {
  636. prev = daily[y+1][x+1]
  637. }
  638. assert.True(t, daily[y+1][x+1] <= prev)
  639. prev = daily[y+1][x+1]
  640. }
  641. }
  642. }
  643. func TestBurndownAddMatrixCrazy(t *testing.T) {
  644. size := 5 * 3
  645. daily := make([][]float32, size)
  646. for i := range daily {
  647. daily[i] = make([]float32, size)
  648. }
  649. added := make([][]int64, 5)
  650. for i := range added {
  651. added[i] = make([]int64, 3)
  652. switch i {
  653. case 0:
  654. added[i][0] = 10
  655. case 1:
  656. added[i][0] = 9
  657. added[i][1] = 2
  658. case 2:
  659. added[i][0] = 8
  660. added[i][1] = 16
  661. case 3:
  662. added[i][0] = 7
  663. added[i][1] = 12
  664. added[i][2] = 6
  665. case 4:
  666. added[i][0] = 6
  667. added[i][1] = 9
  668. added[i][2] = 13
  669. }
  670. }
  671. // yaml.PrintMatrix(os.Stdout, added, 0, "test", true)
  672. /*
  673. "test": |-
  674. 10 0 0
  675. 9 2 0
  676. 8 16 0
  677. 7 12 6
  678. 6 9 13
  679. */
  680. addBurndownMatrix(added, 5, 3, daily, 0)
  681. /*
  682. for _, row := range daily {
  683. for _, v := range row {
  684. fmt.Print(v, " ")
  685. }
  686. fmt.Println()
  687. }
  688. */
  689. // check pinned points
  690. for y := 0; y < 5; y++ {
  691. for x := 0; x < 3; x++ {
  692. var sum float32
  693. for i := x * 5; i < (x+1)*5; i++ {
  694. sum += daily[(y+1)*3-1][i]
  695. }
  696. assert.InDelta(t, sum, added[y][x], 0.00001)
  697. }
  698. }
  699. // check overall trend: 0 -> const -> peak -> decay
  700. for x := 0; x < 15; x++ {
  701. for y := 0; y < x; y++ {
  702. assert.Zero(t, daily[y][x])
  703. }
  704. var prev float32
  705. for y := x; y < ((x+3)/5)*5; y++ {
  706. if prev == 0 {
  707. prev = daily[y][x]
  708. }
  709. assert.Equal(t, daily[y][x], prev)
  710. }
  711. for y := ((x + 3) / 5) * 5; y < 15; y++ {
  712. if prev == 0 {
  713. prev = daily[y][x]
  714. }
  715. assert.True(t, daily[y][x] <= prev)
  716. prev = daily[y][x]
  717. }
  718. }
  719. }
  720. func TestBurndownAddMatrixNaNs(t *testing.T) {
  721. size := 4 * 4
  722. daily := make([][]float32, size)
  723. for i := range daily {
  724. daily[i] = make([]float32, size)
  725. }
  726. added := make([][]int64, 4)
  727. for i := range added {
  728. added[i] = make([]int64, 4)
  729. switch i {
  730. case 0:
  731. added[i][0] = 20
  732. case 1:
  733. added[i][0] = 18
  734. added[i][1] = 30
  735. case 2:
  736. added[i][0] = 15
  737. added[i][1] = 25
  738. added[i][2] = 28
  739. case 3:
  740. added[i][0] = 12
  741. added[i][1] = 20
  742. added[i][2] = 25
  743. added[i][3] = 40
  744. }
  745. }
  746. // yaml.PrintMatrix(os.Stdout, added, 0, "test", true)
  747. /*
  748. "test": |-
  749. 20 0 0 0
  750. 18 30 0 0
  751. 15 25 28 0
  752. 12 20 25 40
  753. */
  754. addBurndownMatrix(added, 4, 4, daily, 0)
  755. /*
  756. for _, row := range daily {
  757. for _, v := range row {
  758. fmt.Print(v, " ")
  759. }
  760. fmt.Println()
  761. }
  762. */
  763. // check pinned points
  764. for y := 0; y < 4; y++ {
  765. for x := 0; x < 4; x++ {
  766. var sum float32
  767. for i := x * 4; i < (x+1)*4; i++ {
  768. sum += daily[(y+1)*4-1][i]
  769. }
  770. assert.InDelta(t, sum, added[y][x], 0.00001)
  771. }
  772. }
  773. // check overall trend: 0 -> const -> peak -> decay
  774. for x := 0; x < 16; x++ {
  775. for y := 0; y < x; y++ {
  776. assert.Zero(t, daily[y][x])
  777. }
  778. var prev float32
  779. for y := x - 4; y < x; y++ {
  780. if y < 0 {
  781. continue
  782. }
  783. if prev == 0 {
  784. prev = daily[y][x]
  785. }
  786. assert.Equal(t, daily[y][x], prev)
  787. }
  788. for y := x; y < 16; y++ {
  789. if prev == 0 {
  790. prev = daily[y][x]
  791. }
  792. assert.True(t, daily[y][x] <= prev)
  793. prev = daily[y][x]
  794. }
  795. }
  796. }
  797. func TestBurndownMergeGlobalHistory(t *testing.T) {
  798. people1 := [...]string{"one", "two"}
  799. res1 := BurndownResult{
  800. GlobalHistory: [][]int64{},
  801. FileHistories: map[string][][]int64{},
  802. PeopleHistories: [][][]int64{},
  803. PeopleMatrix: [][]int64{},
  804. reversedPeopleDict: people1[:],
  805. sampling: 15,
  806. granularity: 20,
  807. }
  808. c1 := core.CommonAnalysisResult{
  809. BeginTime: 600566400, // 1989 Jan 12
  810. EndTime: 604713600, // 1989 March 1
  811. CommitsNumber: 10,
  812. RunTime: 100000,
  813. }
  814. // 48 days
  815. res1.GlobalHistory = make([][]int64, 48/15+1 /* 4 samples */)
  816. for i := range res1.GlobalHistory {
  817. res1.GlobalHistory[i] = make([]int64, 48/20+1 /* 3 bands */)
  818. switch i {
  819. case 0:
  820. res1.GlobalHistory[i][0] = 1000
  821. case 1:
  822. res1.GlobalHistory[i][0] = 1100
  823. res1.GlobalHistory[i][1] = 400
  824. case 2:
  825. res1.GlobalHistory[i][0] = 900
  826. res1.GlobalHistory[i][1] = 750
  827. res1.GlobalHistory[i][2] = 100
  828. case 3:
  829. res1.GlobalHistory[i][0] = 850
  830. res1.GlobalHistory[i][1] = 700
  831. res1.GlobalHistory[i][2] = 150
  832. }
  833. }
  834. res1.FileHistories["file1"] = res1.GlobalHistory
  835. res1.FileHistories["file2"] = res1.GlobalHistory
  836. res1.PeopleHistories = append(res1.PeopleHistories, res1.GlobalHistory)
  837. res1.PeopleHistories = append(res1.PeopleHistories, res1.GlobalHistory)
  838. res1.PeopleMatrix = append(res1.PeopleMatrix, make([]int64, 4))
  839. res1.PeopleMatrix = append(res1.PeopleMatrix, make([]int64, 4))
  840. res1.PeopleMatrix[0][0] = 10
  841. res1.PeopleMatrix[0][1] = 20
  842. res1.PeopleMatrix[0][2] = 30
  843. res1.PeopleMatrix[0][3] = 40
  844. res1.PeopleMatrix[1][0] = 50
  845. res1.PeopleMatrix[1][1] = 60
  846. res1.PeopleMatrix[1][2] = 70
  847. res1.PeopleMatrix[1][3] = 80
  848. people2 := [...]string{"two", "three"}
  849. res2 := BurndownResult{
  850. GlobalHistory: [][]int64{},
  851. FileHistories: map[string][][]int64{},
  852. PeopleHistories: [][][]int64{},
  853. PeopleMatrix: [][]int64{},
  854. reversedPeopleDict: people2[:],
  855. sampling: 14,
  856. granularity: 19,
  857. }
  858. c2 := core.CommonAnalysisResult{
  859. BeginTime: 601084800, // 1989 Jan 18
  860. EndTime: 605923200, // 1989 March 15
  861. CommitsNumber: 10,
  862. RunTime: 100000,
  863. }
  864. // 56 days
  865. res2.GlobalHistory = make([][]int64, 56/14 /* 4 samples */)
  866. for i := range res2.GlobalHistory {
  867. res2.GlobalHistory[i] = make([]int64, 56/19+1 /* 3 bands */)
  868. switch i {
  869. case 0:
  870. res2.GlobalHistory[i][0] = 900
  871. case 1:
  872. res2.GlobalHistory[i][0] = 1100
  873. res2.GlobalHistory[i][1] = 400
  874. case 2:
  875. res2.GlobalHistory[i][0] = 900
  876. res2.GlobalHistory[i][1] = 750
  877. res2.GlobalHistory[i][2] = 100
  878. case 3:
  879. res2.GlobalHistory[i][0] = 800
  880. res2.GlobalHistory[i][1] = 600
  881. res2.GlobalHistory[i][2] = 600
  882. }
  883. }
  884. res2.FileHistories["file2"] = res2.GlobalHistory
  885. res2.FileHistories["file3"] = res2.GlobalHistory
  886. res2.PeopleHistories = append(res2.PeopleHistories, res2.GlobalHistory)
  887. res2.PeopleHistories = append(res2.PeopleHistories, res2.GlobalHistory)
  888. res2.PeopleMatrix = append(res2.PeopleMatrix, make([]int64, 4))
  889. res2.PeopleMatrix = append(res2.PeopleMatrix, make([]int64, 4))
  890. res2.PeopleMatrix[0][0] = 100
  891. res2.PeopleMatrix[0][1] = 200
  892. res2.PeopleMatrix[0][2] = 300
  893. res2.PeopleMatrix[0][3] = 400
  894. res2.PeopleMatrix[1][0] = 500
  895. res2.PeopleMatrix[1][1] = 600
  896. res2.PeopleMatrix[1][2] = 700
  897. res2.PeopleMatrix[1][3] = 800
  898. burndown := BurndownAnalysis{}
  899. merged := burndown.MergeResults(res1, res2, &c1, &c2).(BurndownResult)
  900. assert.Equal(t, merged.granularity, 19)
  901. assert.Equal(t, merged.sampling, 14)
  902. assert.Len(t, merged.GlobalHistory, 5)
  903. for _, row := range merged.GlobalHistory {
  904. assert.Len(t, row, 4)
  905. }
  906. assert.Equal(t, merged.FileHistories["file1"], res1.GlobalHistory)
  907. assert.Equal(t, merged.FileHistories["file2"], merged.GlobalHistory)
  908. assert.Equal(t, merged.FileHistories["file3"], res2.GlobalHistory)
  909. assert.Len(t, merged.reversedPeopleDict, 3)
  910. assert.Equal(t, merged.PeopleHistories[0], res1.GlobalHistory)
  911. assert.Equal(t, merged.PeopleHistories[1], merged.GlobalHistory)
  912. assert.Equal(t, merged.PeopleHistories[2], res2.GlobalHistory)
  913. assert.Len(t, merged.PeopleMatrix, 3)
  914. for _, row := range merged.PeopleMatrix {
  915. assert.Len(t, row, 5)
  916. }
  917. assert.Equal(t, merged.PeopleMatrix[0][0], int64(10))
  918. assert.Equal(t, merged.PeopleMatrix[0][1], int64(20))
  919. assert.Equal(t, merged.PeopleMatrix[0][2], int64(30))
  920. assert.Equal(t, merged.PeopleMatrix[0][3], int64(40))
  921. assert.Equal(t, merged.PeopleMatrix[0][4], int64(0))
  922. assert.Equal(t, merged.PeopleMatrix[1][0], int64(150))
  923. assert.Equal(t, merged.PeopleMatrix[1][1], int64(260))
  924. assert.Equal(t, merged.PeopleMatrix[1][2], int64(70))
  925. assert.Equal(t, merged.PeopleMatrix[1][3], int64(380))
  926. assert.Equal(t, merged.PeopleMatrix[1][4], int64(400))
  927. assert.Equal(t, merged.PeopleMatrix[2][0], int64(500))
  928. assert.Equal(t, merged.PeopleMatrix[2][1], int64(600))
  929. assert.Equal(t, merged.PeopleMatrix[2][2], int64(0))
  930. assert.Equal(t, merged.PeopleMatrix[2][3], int64(700))
  931. assert.Equal(t, merged.PeopleMatrix[2][4], int64(800))
  932. burndown.serializeBinary(&merged, ioutil.Discard)
  933. }
  934. func TestBurndownMergeNils(t *testing.T) {
  935. res1 := BurndownResult{
  936. GlobalHistory: [][]int64{},
  937. FileHistories: map[string][][]int64{},
  938. PeopleHistories: [][][]int64{},
  939. PeopleMatrix: [][]int64{},
  940. reversedPeopleDict: []string{},
  941. sampling: 15,
  942. granularity: 20,
  943. }
  944. c1 := core.CommonAnalysisResult{
  945. BeginTime: 600566400, // 1989 Jan 12
  946. EndTime: 604713600, // 1989 March 1
  947. CommitsNumber: 10,
  948. RunTime: 100000,
  949. }
  950. res2 := BurndownResult{
  951. GlobalHistory: nil,
  952. FileHistories: nil,
  953. PeopleHistories: nil,
  954. PeopleMatrix: nil,
  955. reversedPeopleDict: nil,
  956. sampling: 14,
  957. granularity: 19,
  958. }
  959. c2 := core.CommonAnalysisResult{
  960. BeginTime: 601084800, // 1989 Jan 18
  961. EndTime: 605923200, // 1989 March 15
  962. CommitsNumber: 10,
  963. RunTime: 100000,
  964. }
  965. burndown := BurndownAnalysis{}
  966. merged := burndown.MergeResults(res1, res2, &c1, &c2).(BurndownResult)
  967. assert.Equal(t, merged.granularity, 19)
  968. assert.Equal(t, merged.sampling, 14)
  969. assert.Nil(t, merged.GlobalHistory)
  970. assert.Nil(t, merged.FileHistories)
  971. assert.Nil(t, merged.PeopleHistories)
  972. assert.Nil(t, merged.PeopleMatrix)
  973. burndown.serializeBinary(&merged, ioutil.Discard)
  974. res2.GlobalHistory = make([][]int64, 56/14 /* 4 samples */)
  975. for i := range res2.GlobalHistory {
  976. res2.GlobalHistory[i] = make([]int64, 56/19+1 /* 3 bands */)
  977. switch i {
  978. case 0:
  979. res2.GlobalHistory[i][0] = 900
  980. case 1:
  981. res2.GlobalHistory[i][0] = 1100
  982. res2.GlobalHistory[i][1] = 400
  983. case 2:
  984. res2.GlobalHistory[i][0] = 900
  985. res2.GlobalHistory[i][1] = 750
  986. res2.GlobalHistory[i][2] = 100
  987. case 3:
  988. res2.GlobalHistory[i][0] = 800
  989. res2.GlobalHistory[i][1] = 600
  990. res2.GlobalHistory[i][2] = 600
  991. }
  992. }
  993. people1 := [...]string{"one", "two"}
  994. res1.reversedPeopleDict = people1[:]
  995. res1.PeopleMatrix = append(res1.PeopleMatrix, make([]int64, 4))
  996. res1.PeopleMatrix = append(res1.PeopleMatrix, make([]int64, 4))
  997. res1.PeopleMatrix[0][0] = 10
  998. res1.PeopleMatrix[0][1] = 20
  999. res1.PeopleMatrix[0][2] = 30
  1000. res1.PeopleMatrix[0][3] = 40
  1001. res1.PeopleMatrix[1][0] = 50
  1002. res1.PeopleMatrix[1][1] = 60
  1003. res1.PeopleMatrix[1][2] = 70
  1004. res1.PeopleMatrix[1][3] = 80
  1005. people2 := [...]string{"two", "three"}
  1006. res2.reversedPeopleDict = people2[:]
  1007. merged = burndown.MergeResults(res1, res2, &c1, &c2).(BurndownResult)
  1008. mgh := [5][4]int64{
  1009. {0, 0, 0, 0},
  1010. {578, 0, 0, 0},
  1011. {798, 546, 0, 0},
  1012. {664, 884, 222, 0},
  1013. {547, 663, 610, 178},
  1014. }
  1015. mgh2 := [...][]int64{
  1016. mgh[0][:], mgh[1][:], mgh[2][:], mgh[3][:], mgh[4][:],
  1017. }
  1018. mgh3 := mgh2[:]
  1019. assert.Equal(t, mgh3, merged.GlobalHistory)
  1020. assert.Len(t, merged.PeopleMatrix, 3)
  1021. for _, row := range merged.PeopleMatrix {
  1022. assert.Len(t, row, 5)
  1023. }
  1024. assert.Equal(t, merged.PeopleMatrix[0][0], int64(10))
  1025. assert.Equal(t, merged.PeopleMatrix[0][1], int64(20))
  1026. assert.Equal(t, merged.PeopleMatrix[0][2], int64(30))
  1027. assert.Equal(t, merged.PeopleMatrix[0][3], int64(40))
  1028. assert.Equal(t, merged.PeopleMatrix[0][4], int64(0))
  1029. assert.Equal(t, merged.PeopleMatrix[1][0], int64(50))
  1030. assert.Equal(t, merged.PeopleMatrix[1][1], int64(60))
  1031. assert.Equal(t, merged.PeopleMatrix[1][2], int64(70))
  1032. assert.Equal(t, merged.PeopleMatrix[1][3], int64(80))
  1033. assert.Equal(t, merged.PeopleMatrix[1][4], int64(0))
  1034. assert.Equal(t, merged.PeopleMatrix[2][0], int64(0))
  1035. assert.Equal(t, merged.PeopleMatrix[2][1], int64(0))
  1036. assert.Equal(t, merged.PeopleMatrix[2][2], int64(0))
  1037. assert.Equal(t, merged.PeopleMatrix[2][3], int64(0))
  1038. assert.Equal(t, merged.PeopleMatrix[2][4], int64(0))
  1039. burndown.serializeBinary(&merged, ioutil.Discard)
  1040. }
  1041. func TestBurndownDeserialize(t *testing.T) {
  1042. allBuffer, err := ioutil.ReadFile(path.Join("..", "internal", "test_data", "burndown.pb"))
  1043. assert.Nil(t, err)
  1044. message := pb.AnalysisResults{}
  1045. err = proto.Unmarshal(allBuffer, &message)
  1046. assert.Nil(t, err)
  1047. burndown := BurndownAnalysis{}
  1048. iresult, err := burndown.Deserialize(message.Contents[burndown.Name()])
  1049. assert.Nil(t, err)
  1050. result := iresult.(BurndownResult)
  1051. assert.True(t, len(result.GlobalHistory) > 0)
  1052. assert.True(t, len(result.FileHistories) > 0)
  1053. assert.True(t, len(result.reversedPeopleDict) > 0)
  1054. assert.True(t, len(result.PeopleHistories) > 0)
  1055. assert.True(t, len(result.PeopleMatrix) > 0)
  1056. assert.Equal(t, result.granularity, 30)
  1057. assert.Equal(t, result.sampling, 30)
  1058. }