couples_test.go 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523
  1. package leaves
  2. import (
  3. "bytes"
  4. "io/ioutil"
  5. "path"
  6. "strings"
  7. "testing"
  8. "github.com/gogo/protobuf/proto"
  9. "github.com/stretchr/testify/assert"
  10. gitplumbing "gopkg.in/src-d/go-git.v4/plumbing"
  11. "gopkg.in/src-d/go-git.v4/plumbing/object"
  12. "gopkg.in/src-d/hercules.v6/internal/core"
  13. "gopkg.in/src-d/hercules.v6/internal/pb"
  14. "gopkg.in/src-d/hercules.v6/internal/plumbing"
  15. "gopkg.in/src-d/hercules.v6/internal/plumbing/identity"
  16. "gopkg.in/src-d/hercules.v6/internal/test"
  17. )
  18. func fixtureCouples() *CouplesAnalysis {
  19. c := CouplesAnalysis{PeopleNumber: 3}
  20. c.Initialize(test.Repository)
  21. return &c
  22. }
  23. func TestCouplesMeta(t *testing.T) {
  24. c := fixtureCouples()
  25. assert.Equal(t, c.Name(), "Couples")
  26. assert.Equal(t, len(c.Provides()), 0)
  27. assert.Equal(t, len(c.Requires()), 2)
  28. assert.Equal(t, c.Requires()[0], identity.DependencyAuthor)
  29. assert.Equal(t, c.Requires()[1], plumbing.DependencyTreeChanges)
  30. assert.Equal(t, c.Flag(), "couples")
  31. assert.Len(t, c.ListConfigurationOptions(), 0)
  32. }
  33. func TestCouplesRegistration(t *testing.T) {
  34. summoned := core.Registry.Summon((&CouplesAnalysis{}).Name())
  35. assert.Len(t, summoned, 1)
  36. assert.Equal(t, summoned[0].Name(), "Couples")
  37. leaves := core.Registry.GetLeaves()
  38. matched := false
  39. for _, tp := range leaves {
  40. if tp.Flag() == (&CouplesAnalysis{}).Flag() {
  41. matched = true
  42. break
  43. }
  44. }
  45. assert.True(t, matched)
  46. }
  47. func generateChanges(names ...string) object.Changes {
  48. changes := make(object.Changes, 0, len(names))
  49. for _, name := range names {
  50. action := name[:1]
  51. name = name[1:]
  52. var change object.Change
  53. if action == "+" {
  54. change = object.Change{
  55. From: object.ChangeEntry{},
  56. To: object.ChangeEntry{Name: name},
  57. }
  58. } else if action == "-" {
  59. change = object.Change{
  60. From: object.ChangeEntry{Name: name},
  61. To: object.ChangeEntry{},
  62. }
  63. } else if action == "=" {
  64. change = object.Change{
  65. From: object.ChangeEntry{Name: name},
  66. To: object.ChangeEntry{Name: name},
  67. }
  68. } else {
  69. if action != ">" {
  70. panic("Invalid action.")
  71. }
  72. parts := strings.Split(name, ">")
  73. change = object.Change{
  74. From: object.ChangeEntry{Name: parts[0]},
  75. To: object.ChangeEntry{Name: parts[1]},
  76. }
  77. }
  78. changes = append(changes, &change)
  79. }
  80. return changes
  81. }
  82. func TestCouplesConsumeFinalize(t *testing.T) {
  83. c := fixtureCouples()
  84. deps := map[string]interface{}{}
  85. deps[identity.DependencyAuthor] = 0
  86. deps[core.DependencyCommit], _ = test.Repository.CommitObject(gitplumbing.NewHash(
  87. "a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
  88. deps[core.DependencyIsMerge] = false
  89. deps[plumbing.DependencyTreeChanges] = generateChanges("+LICENSE2", "+file2.go", "+rbtree2.go")
  90. c.Consume(deps)
  91. deps[plumbing.DependencyTreeChanges] = generateChanges("+README.md", "-LICENSE2", "=analyser.go", ">file2.go>file_test.go")
  92. c.Consume(deps)
  93. deps[identity.DependencyAuthor] = 1
  94. deps[plumbing.DependencyTreeChanges] = generateChanges("=README.md", "=analyser.go", "-rbtree2.go")
  95. c.Consume(deps)
  96. deps[identity.DependencyAuthor] = 2
  97. deps[plumbing.DependencyTreeChanges] = generateChanges("=file_test.go")
  98. c.Consume(deps)
  99. assert.Equal(t, len(c.people[0]), 6)
  100. assert.Equal(t, c.people[0]["README.md"], 1)
  101. assert.Equal(t, c.people[0]["LICENSE2"], 2)
  102. assert.Equal(t, c.people[0]["analyser.go"], 1)
  103. assert.Equal(t, c.people[0]["file2.go"], 1)
  104. assert.Equal(t, c.people[0]["file_test.go"], 1)
  105. assert.Equal(t, c.people[0]["rbtree2.go"], 1)
  106. assert.Equal(t, len(c.people[1]), 3)
  107. assert.Equal(t, c.people[1]["README.md"], 1)
  108. assert.Equal(t, c.people[1]["analyser.go"], 1)
  109. assert.Equal(t, c.people[1]["rbtree2.go"], 1)
  110. assert.Equal(t, len(c.people[2]), 1)
  111. assert.Equal(t, c.people[2]["file_test.go"], 1)
  112. assert.Equal(t, len(c.files["README.md"]), 3)
  113. assert.Equal(t, c.files["README.md"], map[string]int{
  114. "README.md": 2,
  115. "analyser.go": 2,
  116. "file_test.go": 1,
  117. })
  118. assert.Equal(t, c.files["LICENSE2"], map[string]int{
  119. "LICENSE2": 1,
  120. "file2.go": 1,
  121. "rbtree2.go": 1,
  122. })
  123. assert.Equal(t, c.files["file2.go"], map[string]int{
  124. "LICENSE2": 1,
  125. "file2.go": 1,
  126. "rbtree2.go": 1,
  127. })
  128. assert.Equal(t, c.files["rbtree2.go"], map[string]int{
  129. "LICENSE2": 1,
  130. "file2.go": 1,
  131. "rbtree2.go": 1,
  132. })
  133. assert.Equal(t, c.files["analyser.go"], map[string]int{
  134. "analyser.go": 2,
  135. "README.md": 2,
  136. "file_test.go": 1,
  137. })
  138. assert.Equal(t, c.files["file_test.go"], map[string]int{
  139. "file_test.go": 2,
  140. "README.md": 1,
  141. "analyser.go": 1,
  142. })
  143. assert.Equal(t, c.peopleCommits[0], 2)
  144. assert.Equal(t, c.peopleCommits[1], 1)
  145. assert.Equal(t, c.peopleCommits[2], 1)
  146. cr := c.Finalize().(CouplesResult)
  147. assert.Equal(t, len(cr.Files), 3)
  148. assert.Equal(t, cr.Files[0], "README.md")
  149. assert.Equal(t, cr.Files[1], "analyser.go")
  150. assert.Equal(t, cr.Files[2], "file_test.go")
  151. assert.Equal(t, len(cr.PeopleFiles[0]), 3)
  152. assert.Equal(t, cr.PeopleFiles[0][0], 0)
  153. assert.Equal(t, cr.PeopleFiles[0][1], 1)
  154. assert.Equal(t, cr.PeopleFiles[0][2], 2)
  155. assert.Equal(t, len(cr.PeopleFiles[1]), 2)
  156. assert.Equal(t, cr.PeopleFiles[1][0], 0)
  157. assert.Equal(t, cr.PeopleFiles[1][1], 1)
  158. assert.Equal(t, len(cr.PeopleFiles[2]), 1)
  159. assert.Equal(t, cr.PeopleFiles[2][0], 2)
  160. assert.Equal(t, len(cr.PeopleMatrix[0]), 3)
  161. assert.Equal(t, cr.PeopleMatrix[0][0], int64(7))
  162. assert.Equal(t, cr.PeopleMatrix[0][1], int64(3))
  163. assert.Equal(t, cr.PeopleMatrix[0][2], int64(1))
  164. assert.Equal(t, len(cr.PeopleMatrix[1]), 2)
  165. assert.Equal(t, cr.PeopleMatrix[1][0], int64(3))
  166. assert.Equal(t, cr.PeopleMatrix[1][1], int64(3))
  167. assert.Equal(t, len(cr.PeopleMatrix[2]), 2)
  168. assert.Equal(t, cr.PeopleMatrix[2][0], int64(1))
  169. assert.Equal(t, cr.PeopleMatrix[2][2], int64(1))
  170. assert.Equal(t, len(cr.FilesMatrix), 3)
  171. assert.Equal(t, len(cr.FilesMatrix[0]), 3)
  172. assert.Equal(t, cr.FilesMatrix[0][2], int64(1))
  173. assert.Equal(t, cr.FilesMatrix[0][0], int64(2))
  174. assert.Equal(t, cr.FilesMatrix[0][1], int64(2))
  175. assert.Equal(t, len(cr.FilesMatrix[1]), 3)
  176. assert.Equal(t, cr.FilesMatrix[1][2], int64(1))
  177. assert.Equal(t, cr.FilesMatrix[1][0], int64(2))
  178. assert.Equal(t, cr.FilesMatrix[1][1], int64(2))
  179. assert.Equal(t, len(cr.FilesMatrix[2]), 3)
  180. assert.Equal(t, cr.FilesMatrix[2][0], int64(1))
  181. assert.Equal(t, cr.FilesMatrix[2][1], int64(1))
  182. assert.Equal(t, cr.FilesMatrix[2][2], int64(3))
  183. }
  184. func TestCouplesConsumeFinalizeAuthorMissing(t *testing.T) {
  185. c := fixtureCouples()
  186. deps := map[string]interface{}{}
  187. deps[identity.DependencyAuthor] = 0
  188. deps[core.DependencyCommit], _ = test.Repository.CommitObject(gitplumbing.NewHash(
  189. "a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
  190. deps[core.DependencyIsMerge] = false
  191. deps[plumbing.DependencyTreeChanges] = generateChanges("+LICENSE2", "+file2.go", "+rbtree2.go")
  192. c.Consume(deps)
  193. deps[plumbing.DependencyTreeChanges] = generateChanges("+README.md", "-LICENSE2", "=analyser.go", ">file2.go>file_test.go")
  194. c.Consume(deps)
  195. deps[identity.DependencyAuthor] = 1
  196. deps[plumbing.DependencyTreeChanges] = generateChanges("=README.md", "=analyser.go", "-rbtree2.go")
  197. c.Consume(deps)
  198. deps[identity.DependencyAuthor] = identity.AuthorMissing
  199. deps[plumbing.DependencyTreeChanges] = generateChanges("=file_test.go")
  200. c.Consume(deps)
  201. assert.Equal(t, len(c.people[0]), 6)
  202. assert.Equal(t, c.people[0]["README.md"], 1)
  203. assert.Equal(t, c.people[0]["LICENSE2"], 2)
  204. assert.Equal(t, c.people[0]["analyser.go"], 1)
  205. assert.Equal(t, c.people[0]["file2.go"], 1)
  206. assert.Equal(t, c.people[0]["file_test.go"], 1)
  207. assert.Equal(t, c.people[0]["rbtree2.go"], 1)
  208. assert.Equal(t, len(c.people[1]), 3)
  209. assert.Equal(t, c.people[1]["README.md"], 1)
  210. assert.Equal(t, c.people[1]["analyser.go"], 1)
  211. assert.Equal(t, c.people[1]["rbtree2.go"], 1)
  212. assert.Equal(t, len(c.people[2]), 0)
  213. assert.Equal(t, len(c.files["README.md"]), 3)
  214. assert.Equal(t, c.files["README.md"], map[string]int{
  215. "README.md": 2,
  216. "analyser.go": 2,
  217. "file_test.go": 1,
  218. })
  219. assert.Equal(t, c.files["LICENSE2"], map[string]int{
  220. "LICENSE2": 1,
  221. "file2.go": 1,
  222. "rbtree2.go": 1,
  223. })
  224. assert.Equal(t, c.files["file2.go"], map[string]int{
  225. "LICENSE2": 1,
  226. "file2.go": 1,
  227. "rbtree2.go": 1,
  228. })
  229. assert.Equal(t, c.files["rbtree2.go"], map[string]int{
  230. "LICENSE2": 1,
  231. "file2.go": 1,
  232. "rbtree2.go": 1,
  233. })
  234. assert.Equal(t, c.files["analyser.go"], map[string]int{
  235. "analyser.go": 2,
  236. "README.md": 2,
  237. "file_test.go": 1,
  238. })
  239. assert.Equal(t, c.files["file_test.go"], map[string]int{
  240. "file_test.go": 2,
  241. "README.md": 1,
  242. "analyser.go": 1,
  243. })
  244. assert.Equal(t, c.peopleCommits[0], 2)
  245. assert.Equal(t, c.peopleCommits[1], 1)
  246. assert.Equal(t, c.peopleCommits[2], 0)
  247. cr := c.Finalize().(CouplesResult)
  248. assert.Equal(t, len(cr.Files), 3)
  249. assert.Equal(t, cr.Files[0], "README.md")
  250. assert.Equal(t, cr.Files[1], "analyser.go")
  251. assert.Equal(t, cr.Files[2], "file_test.go")
  252. assert.Equal(t, len(cr.PeopleFiles[0]), 3)
  253. assert.Equal(t, cr.PeopleFiles[0][0], 0)
  254. assert.Equal(t, cr.PeopleFiles[0][1], 1)
  255. assert.Equal(t, cr.PeopleFiles[0][2], 2)
  256. assert.Equal(t, len(cr.PeopleFiles[1]), 2)
  257. assert.Equal(t, cr.PeopleFiles[1][0], 0)
  258. assert.Equal(t, cr.PeopleFiles[1][1], 1)
  259. assert.Equal(t, len(cr.PeopleFiles[2]), 0)
  260. assert.Equal(t, len(cr.PeopleMatrix[0]), 3)
  261. assert.Equal(t, cr.PeopleMatrix[0][0], int64(7))
  262. assert.Equal(t, cr.PeopleMatrix[0][1], int64(3))
  263. assert.Equal(t, cr.PeopleMatrix[0][2], int64(0))
  264. assert.Equal(t, len(cr.PeopleMatrix[1]), 2)
  265. assert.Equal(t, cr.PeopleMatrix[1][0], int64(3))
  266. assert.Equal(t, cr.PeopleMatrix[1][1], int64(3))
  267. assert.Equal(t, len(cr.PeopleMatrix[2]), 0)
  268. assert.Equal(t, len(cr.FilesMatrix), 3)
  269. assert.Equal(t, len(cr.FilesMatrix[0]), 3)
  270. assert.Equal(t, cr.FilesMatrix[0][2], int64(1))
  271. assert.Equal(t, cr.FilesMatrix[0][0], int64(2))
  272. assert.Equal(t, cr.FilesMatrix[0][1], int64(2))
  273. assert.Equal(t, len(cr.FilesMatrix[1]), 3)
  274. assert.Equal(t, cr.FilesMatrix[1][2], int64(1))
  275. assert.Equal(t, cr.FilesMatrix[1][0], int64(2))
  276. assert.Equal(t, cr.FilesMatrix[1][1], int64(2))
  277. assert.Equal(t, len(cr.FilesMatrix[2]), 3)
  278. assert.Equal(t, cr.FilesMatrix[2][0], int64(1))
  279. assert.Equal(t, cr.FilesMatrix[2][1], int64(1))
  280. assert.Equal(t, cr.FilesMatrix[2][2], int64(3))
  281. }
  282. func TestCouplesFork(t *testing.T) {
  283. couples1 := fixtureCouples()
  284. clones := couples1.Fork(1)
  285. assert.Len(t, clones, 1)
  286. couples2 := clones[0].(*CouplesAnalysis)
  287. assert.True(t, couples1 != couples2)
  288. assert.Equal(t, *couples1, *couples2)
  289. couples1.Merge([]core.PipelineItem{couples2})
  290. }
  291. func TestCouplesSerialize(t *testing.T) {
  292. c := fixtureCouples()
  293. result := CouplesResult{
  294. PeopleMatrix: []map[int]int64{
  295. {0: 7, 1: 3, 2: 1}, {0: 3, 1: 3}, {0: 1, 2: 1}, {},
  296. },
  297. PeopleFiles: [][]int{
  298. {0, 1, 2}, {1, 2}, {0}, {},
  299. },
  300. FilesMatrix: []map[int]int64{
  301. {1: 1, 2: 1, 0: 3}, {1: 2, 2: 2, 0: 1}, {2: 2, 0: 1, 1: 2},
  302. },
  303. Files: []string{"five", "one", "three"},
  304. reversedPeopleDict: []string{"p1", "p2", "p3"},
  305. }
  306. buffer := &bytes.Buffer{}
  307. c.Serialize(result, false, buffer)
  308. assert.Equal(t, buffer.String(), ` files_coocc:
  309. index:
  310. - "five"
  311. - "one"
  312. - "three"
  313. matrix:
  314. - {0: 3, 1: 1, 2: 1}
  315. - {0: 1, 1: 2, 2: 2}
  316. - {0: 1, 1: 2, 2: 2}
  317. people_coocc:
  318. index:
  319. - "p1"
  320. - "p2"
  321. - "p3"
  322. matrix:
  323. - {0: 7, 1: 3, 2: 1}
  324. - {0: 3, 1: 3}
  325. - {0: 1, 2: 1}
  326. - {}
  327. author_files:
  328. - "p3":
  329. - "five"
  330. - "p2":
  331. - "one"
  332. - "three"
  333. - "p1":
  334. - "five"
  335. - "one"
  336. - "three"
  337. `)
  338. buffer = &bytes.Buffer{}
  339. c.Serialize(result, true, buffer)
  340. msg := pb.CouplesAnalysisResults{}
  341. proto.Unmarshal(buffer.Bytes(), &msg)
  342. assert.Len(t, msg.PeopleFiles, 3)
  343. tmp1 := [...]int32{0, 1, 2}
  344. assert.Equal(t, msg.PeopleFiles[0].Files, tmp1[:])
  345. tmp2 := [...]int32{1, 2}
  346. assert.Equal(t, msg.PeopleFiles[1].Files, tmp2[:])
  347. tmp3 := [...]int32{0}
  348. assert.Equal(t, msg.PeopleFiles[2].Files, tmp3[:])
  349. assert.Equal(t, msg.PeopleCouples.Index, result.reversedPeopleDict)
  350. assert.Equal(t, msg.PeopleCouples.Matrix.NumberOfRows, int32(4))
  351. assert.Equal(t, msg.PeopleCouples.Matrix.NumberOfColumns, int32(4))
  352. data := [...]int64{7, 3, 1, 3, 3, 1, 1}
  353. assert.Equal(t, msg.PeopleCouples.Matrix.Data, data[:])
  354. indices := [...]int32{0, 1, 2, 0, 1, 0, 2}
  355. assert.Equal(t, msg.PeopleCouples.Matrix.Indices, indices[:])
  356. indptr := [...]int64{0, 3, 5, 7, 7}
  357. assert.Equal(t, msg.PeopleCouples.Matrix.Indptr, indptr[:])
  358. files := [...]string{"five", "one", "three"}
  359. assert.Equal(t, msg.FileCouples.Index, files[:])
  360. assert.Equal(t, msg.FileCouples.Matrix.NumberOfRows, int32(3))
  361. assert.Equal(t, msg.FileCouples.Matrix.NumberOfColumns, int32(3))
  362. data2 := [...]int64{3, 1, 1, 1, 2, 2, 1, 2, 2}
  363. assert.Equal(t, msg.FileCouples.Matrix.Data, data2[:])
  364. indices2 := [...]int32{0, 1, 2, 0, 1, 2, 0, 1, 2}
  365. assert.Equal(t, msg.FileCouples.Matrix.Indices, indices2[:])
  366. indptr2 := [...]int64{0, 3, 6, 9}
  367. assert.Equal(t, msg.FileCouples.Matrix.Indptr, indptr2[:])
  368. }
  369. func TestCouplesDeserialize(t *testing.T) {
  370. allBuffer, err := ioutil.ReadFile(path.Join("..", "internal", "test_data", "couples.pb"))
  371. assert.Nil(t, err)
  372. message := pb.AnalysisResults{}
  373. err = proto.Unmarshal(allBuffer, &message)
  374. assert.Nil(t, err)
  375. couples := CouplesAnalysis{}
  376. iresult, err := couples.Deserialize(message.Contents[couples.Name()])
  377. assert.Nil(t, err)
  378. result := iresult.(CouplesResult)
  379. assert.Len(t, result.reversedPeopleDict, 2)
  380. assert.Len(t, result.PeopleFiles, 2)
  381. assert.Len(t, result.PeopleMatrix, 3)
  382. assert.Len(t, result.Files, 74)
  383. assert.Len(t, result.FilesMatrix, 74)
  384. }
  385. func TestCouplesMerge(t *testing.T) {
  386. r1, r2 := CouplesResult{}, CouplesResult{}
  387. people1 := [...]string{"one", "two"}
  388. people2 := [...]string{"two", "three"}
  389. r1.reversedPeopleDict = people1[:]
  390. r2.reversedPeopleDict = people2[:]
  391. r1.Files = people1[:]
  392. r2.Files = people2[:]
  393. r1.PeopleFiles = make([][]int, 2)
  394. r1.PeopleFiles[0] = make([]int, 2)
  395. r1.PeopleFiles[0][0] = 0
  396. r1.PeopleFiles[0][1] = 1
  397. r1.PeopleFiles[1] = make([]int, 1)
  398. r1.PeopleFiles[1][0] = 0
  399. r2.PeopleFiles = make([][]int, 2)
  400. r2.PeopleFiles[0] = make([]int, 1)
  401. r2.PeopleFiles[0][0] = 1
  402. r2.PeopleFiles[1] = make([]int, 2)
  403. r2.PeopleFiles[1][0] = 0
  404. r2.PeopleFiles[1][1] = 1
  405. r1.FilesMatrix = make([]map[int]int64, 2)
  406. r1.FilesMatrix[0] = map[int]int64{}
  407. r1.FilesMatrix[1] = map[int]int64{}
  408. r1.FilesMatrix[0][1] = 100
  409. r1.FilesMatrix[1][0] = 100
  410. r2.FilesMatrix = make([]map[int]int64, 2)
  411. r2.FilesMatrix[0] = map[int]int64{}
  412. r2.FilesMatrix[1] = map[int]int64{}
  413. r2.FilesMatrix[0][1] = 200
  414. r2.FilesMatrix[1][0] = 200
  415. r1.PeopleMatrix = make([]map[int]int64, 3)
  416. r1.PeopleMatrix[0] = map[int]int64{}
  417. r1.PeopleMatrix[1] = map[int]int64{}
  418. r1.PeopleMatrix[2] = map[int]int64{}
  419. r1.PeopleMatrix[0][1] = 100
  420. r1.PeopleMatrix[1][0] = 100
  421. r1.PeopleMatrix[2][0] = 300
  422. r1.PeopleMatrix[2][1] = 400
  423. r2.PeopleMatrix = make([]map[int]int64, 3)
  424. r2.PeopleMatrix[0] = map[int]int64{}
  425. r2.PeopleMatrix[1] = map[int]int64{}
  426. r2.PeopleMatrix[2] = map[int]int64{}
  427. r2.PeopleMatrix[0][1] = 10
  428. r2.PeopleMatrix[1][0] = 10
  429. r2.PeopleMatrix[2][0] = 30
  430. r2.PeopleMatrix[2][1] = 40
  431. couples := CouplesAnalysis{}
  432. merged := couples.MergeResults(r1, r2, nil, nil).(CouplesResult)
  433. mergedPeople := [...]string{"one", "two", "three"}
  434. assert.Equal(t, merged.reversedPeopleDict, mergedPeople[:])
  435. assert.Equal(t, merged.Files, mergedPeople[:])
  436. assert.Len(t, merged.PeopleFiles, 3)
  437. assert.Equal(t, merged.PeopleFiles[0], getSlice(0, 1))
  438. assert.Equal(t, merged.PeopleFiles[1], getSlice(0, 2))
  439. assert.Equal(t, merged.PeopleFiles[2], getSlice(1, 2))
  440. assert.Len(t, merged.PeopleMatrix, 4)
  441. assert.Equal(t, merged.PeopleMatrix[0], getCouplesMap(1, 100))
  442. assert.Equal(t, merged.PeopleMatrix[1], getCouplesMap(0, 100, 2, 10))
  443. assert.Equal(t, merged.PeopleMatrix[2], getCouplesMap(1, 10))
  444. assert.Equal(t, merged.PeopleMatrix[3], getCouplesMap(0, 300, 1, 430, 2, 40))
  445. assert.Len(t, merged.FilesMatrix, 3)
  446. assert.Equal(t, merged.FilesMatrix[0], getCouplesMap(1, 100))
  447. assert.Equal(t, merged.FilesMatrix[1], getCouplesMap(0, 100, 2, 200))
  448. assert.Equal(t, merged.FilesMatrix[2], getCouplesMap(1, 200))
  449. }
  450. func TestCouplesCurrentFiles(t *testing.T) {
  451. c := fixtureCouples()
  452. c.lastCommit, _ = test.Repository.CommitObject(gitplumbing.NewHash(
  453. "cce947b98a050c6d356bc6ba95030254914027b1"))
  454. files := c.currentFiles()
  455. assert.Equal(t, files, map[string]bool{".gitignore": true, "LICENSE": true})
  456. }
  457. func TestCouplesPropagateRenames(t *testing.T) {
  458. c := fixtureCouples()
  459. c.files["one"] = map[string]int{
  460. "one": 1,
  461. "two": 2,
  462. "three": 3,
  463. }
  464. c.files["two"] = map[string]int{
  465. "one": 2,
  466. "two": 10,
  467. "three": 1,
  468. "four": 7,
  469. }
  470. c.files["three"] = map[string]int{
  471. "one": 3,
  472. "two": 1,
  473. "three": 3,
  474. "four": 2,
  475. }
  476. c.files["four"] = map[string]int{
  477. "two": 7,
  478. "three": 3,
  479. "four": 1,
  480. }
  481. c.PeopleNumber = 1
  482. c.people = make([]map[string]int, 1)
  483. c.people[0] = map[string]int{}
  484. c.people[0]["one"] = 1
  485. c.people[0]["two"] = 2
  486. c.people[0]["three"] = 3
  487. c.people[0]["four"] = 4
  488. *c.renames = []rename{{ToName: "four", FromName: "one"}}
  489. files, people := c.propagateRenames(map[string]bool{"two": true, "three": true, "four": true})
  490. assert.Len(t, files, 3)
  491. assert.Len(t, people, 1)
  492. assert.Equal(t, files["two"], map[string]int{"two": 10, "three": 1, "four": 9})
  493. assert.Equal(t, files["three"], map[string]int{"two": 1, "three": 3, "four": 6})
  494. assert.Equal(t, files["four"], map[string]int{"two": 9, "three": 6, "four": 2})
  495. assert.Equal(t, people[0], map[string]int{"two": 2, "three": 3, "four": 5})
  496. }
  497. func getSlice(vals ...int) []int {
  498. return vals
  499. }
  500. func getCouplesMap(vals ...int) map[int]int64 {
  501. res := map[int]int64{}
  502. for i := 0; i < len(vals); i += 2 {
  503. res[vals[i]] = int64(vals[i+1])
  504. }
  505. return res
  506. }