renames_test.go 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. package hercules
  2. import (
  3. "github.com/stretchr/testify/assert"
  4. "gopkg.in/src-d/go-git.v4/plumbing"
  5. "gopkg.in/src-d/go-git.v4/plumbing/object"
  6. "testing"
  7. )
  8. func fixtureRenameAnalysis() *RenameAnalysis {
  9. ra := RenameAnalysis{SimilarityThreshold: 80}
  10. ra.Initialize(testRepository)
  11. return &ra
  12. }
  13. func TestRenameAnalysisMeta(t *testing.T) {
  14. ra := fixtureRenameAnalysis()
  15. assert.Equal(t, ra.Name(), "RenameAnalysis")
  16. assert.Equal(t, len(ra.Provides()), 1)
  17. assert.Equal(t, ra.Provides()[0], "changes")
  18. assert.Equal(t, len(ra.Requires()), 2)
  19. assert.Equal(t, ra.Requires()[0], "blob_cache")
  20. assert.Equal(t, ra.Requires()[1], "changes")
  21. opts := ra.ListConfigurationOptions()
  22. assert.Len(t, opts, 1)
  23. assert.Equal(t, opts[0].Name, ConfigRenameAnalysisSimilarityThreshold)
  24. ra.SimilarityThreshold = 0
  25. facts := map[string]interface{}{}
  26. facts[ConfigRenameAnalysisSimilarityThreshold] = 70
  27. ra.Configure(facts)
  28. assert.Equal(t, ra.SimilarityThreshold, 70)
  29. delete(facts, ConfigRenameAnalysisSimilarityThreshold)
  30. ra.Configure(facts)
  31. assert.Equal(t, ra.SimilarityThreshold, 70)
  32. }
  33. func TestRenameAnalysisRegistration(t *testing.T) {
  34. tp, exists := Registry.registered[(&RenameAnalysis{}).Name()]
  35. assert.True(t, exists)
  36. assert.Equal(t, tp.Elem().Name(), "RenameAnalysis")
  37. tps, exists := Registry.provided[(&RenameAnalysis{}).Provides()[0]]
  38. assert.True(t, exists)
  39. assert.True(t, len(tps) >= 1)
  40. matched := false
  41. for _, tp := range tps {
  42. matched = matched || tp.Elem().Name() == "RenameAnalysis"
  43. }
  44. assert.True(t, matched)
  45. }
  46. func TestRenameAnalysisInitializeInvalidThreshold(t *testing.T) {
  47. ra := RenameAnalysis{SimilarityThreshold: -10}
  48. ra.Initialize(testRepository)
  49. assert.Equal(t, ra.SimilarityThreshold, RENAME_ANALYSIS_DEFAULT_THRESHOLD)
  50. ra = RenameAnalysis{SimilarityThreshold: 110}
  51. ra.Initialize(testRepository)
  52. assert.Equal(t, ra.SimilarityThreshold, RENAME_ANALYSIS_DEFAULT_THRESHOLD)
  53. ra = RenameAnalysis{SimilarityThreshold: 0}
  54. ra.Initialize(testRepository)
  55. ra = RenameAnalysis{SimilarityThreshold: 100}
  56. ra.Initialize(testRepository)
  57. }
  58. func TestRenameAnalysisConsume(t *testing.T) {
  59. ra := fixtureRenameAnalysis()
  60. changes := make(object.Changes, 3)
  61. // 2b1ed978194a94edeabbca6de7ff3b5771d4d665
  62. treeFrom, _ := testRepository.TreeObject(plumbing.NewHash(
  63. "96c6ece9b2f3c7c51b83516400d278dea5605100"))
  64. treeTo, _ := testRepository.TreeObject(plumbing.NewHash(
  65. "251f2094d7b523d5bcc60e663b6cf38151bf8844"))
  66. changes[0] = &object.Change{From: object.ChangeEntry{
  67. Name: "analyser.go",
  68. Tree: treeFrom,
  69. TreeEntry: object.TreeEntry{
  70. Name: "analyser.go",
  71. Mode: 0100644,
  72. Hash: plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1"),
  73. },
  74. }, To: object.ChangeEntry{},
  75. }
  76. changes[1] = &object.Change{From: object.ChangeEntry{}, To: object.ChangeEntry{
  77. Name: "burndown.go",
  78. Tree: treeTo,
  79. TreeEntry: object.TreeEntry{
  80. Name: "burndown.go",
  81. Mode: 0100644,
  82. Hash: plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2"),
  83. },
  84. },
  85. }
  86. changes[2] = &object.Change{From: object.ChangeEntry{
  87. Name: "cmd/hercules/main.go",
  88. Tree: treeFrom,
  89. TreeEntry: object.TreeEntry{
  90. Name: "cmd/hercules/main.go",
  91. Mode: 0100644,
  92. Hash: plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9"),
  93. },
  94. }, To: object.ChangeEntry{
  95. Name: "cmd/hercules/main.go",
  96. Tree: treeTo,
  97. TreeEntry: object.TreeEntry{
  98. Name: "cmd/hercules/main.go",
  99. Mode: 0100644,
  100. Hash: plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72"),
  101. },
  102. },
  103. }
  104. cache := map[plumbing.Hash]*object.Blob{}
  105. hash := plumbing.NewHash("baa64828831d174f40140e4b3cfa77d1e917a2c1")
  106. cache[hash], _ = testRepository.BlobObject(hash)
  107. hash = plumbing.NewHash("29c9fafd6a2fae8cd20298c3f60115bc31a4c0f2")
  108. cache[hash], _ = testRepository.BlobObject(hash)
  109. hash = plumbing.NewHash("c29112dbd697ad9b401333b80c18a63951bc18d9")
  110. cache[hash], _ = testRepository.BlobObject(hash)
  111. hash = plumbing.NewHash("f7d918ec500e2f925ecde79b51cc007bac27de72")
  112. cache[hash], _ = testRepository.BlobObject(hash)
  113. deps := map[string]interface{}{}
  114. deps["blob_cache"] = cache
  115. deps["changes"] = changes
  116. ra.SimilarityThreshold = 33
  117. res, err := ra.Consume(deps)
  118. assert.Nil(t, err)
  119. renamed := res["changes"].(object.Changes)
  120. assert.Equal(t, len(renamed), 2)
  121. ra.SimilarityThreshold = 35
  122. res, err = ra.Consume(deps)
  123. assert.Nil(t, err)
  124. renamed = res["changes"].(object.Changes)
  125. assert.Equal(t, len(renamed), 3)
  126. }
  127. func TestSortableChanges(t *testing.T) {
  128. changes := sortableChanges{
  129. sortableChange{
  130. nil, plumbing.NewHash("0000000000000000000000000000000000000000"),
  131. }, sortableChange{
  132. nil, plumbing.NewHash("ffffffffffffffffffffffffffffffffffffffff"),
  133. },
  134. }
  135. assert.True(t, changes.Less(0, 1))
  136. assert.False(t, changes.Less(1, 0))
  137. assert.False(t, changes.Less(0, 0))
  138. changes.Swap(0, 1)
  139. assert.Equal(t, changes[0].hash.String(), "ffffffffffffffffffffffffffffffffffffffff")
  140. assert.Equal(t, changes[1].hash.String(), "0000000000000000000000000000000000000000")
  141. }
  142. func TestSortableBlobs(t *testing.T) {
  143. blobs := sortableBlobs{
  144. sortableBlob{
  145. nil, int64(0),
  146. }, sortableBlob{
  147. nil, int64(1),
  148. },
  149. }
  150. assert.True(t, blobs.Less(0, 1))
  151. assert.False(t, blobs.Less(1, 0))
  152. assert.False(t, blobs.Less(0, 0))
  153. blobs.Swap(0, 1)
  154. assert.Equal(t, blobs[0].size, int64(1))
  155. assert.Equal(t, blobs[1].size, int64(0))
  156. }