burndown_test.go 39 KB

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