burndown_test.go 33 KB

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