pipeline_test.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921
  1. package core
  2. import (
  3. "bytes"
  4. "errors"
  5. "fmt"
  6. "io"
  7. "io/ioutil"
  8. "os"
  9. "testing"
  10. "time"
  11. "github.com/stretchr/testify/assert"
  12. "github.com/stretchr/testify/require"
  13. "gopkg.in/src-d/go-git.v4"
  14. "gopkg.in/src-d/go-git.v4/plumbing"
  15. "gopkg.in/src-d/go-git.v4/plumbing/object"
  16. "gopkg.in/src-d/hercules.v10/internal/pb"
  17. "gopkg.in/src-d/hercules.v10/internal/test"
  18. )
  19. type testPipelineItem struct {
  20. Initialized bool
  21. DepsConsumed bool
  22. Disposed bool
  23. Forked bool
  24. Merged *bool
  25. CommitMatches bool
  26. IndexMatches bool
  27. MergeState *int
  28. TestError bool
  29. ConfigureRaises bool
  30. InitializeRaises bool
  31. InitializePanics bool
  32. ConsumePanics bool
  33. Logger Logger
  34. }
  35. func (item *testPipelineItem) Name() string {
  36. return "Test"
  37. }
  38. func (item *testPipelineItem) Provides() []string {
  39. arr := [...]string{"test"}
  40. return arr[:]
  41. }
  42. func (item *testPipelineItem) Requires() []string {
  43. return []string{}
  44. }
  45. func (item *testPipelineItem) Configure(facts map[string]interface{}) error {
  46. if item.ConfigureRaises {
  47. return errors.New("test1")
  48. }
  49. if l, ok := facts[ConfigLogger].(Logger); ok {
  50. item.Logger = l
  51. }
  52. return nil
  53. }
  54. func (item *testPipelineItem) ListConfigurationOptions() []ConfigurationOption {
  55. options := [...]ConfigurationOption{{
  56. Name: "TestOption",
  57. Description: "The option description.",
  58. Flag: "test-option",
  59. Type: IntConfigurationOption,
  60. Default: 10,
  61. }}
  62. return options[:]
  63. }
  64. func (item *testPipelineItem) Flag() string {
  65. return "mytest"
  66. }
  67. func (item *testPipelineItem) Description() string {
  68. return "description!"
  69. }
  70. func (item *testPipelineItem) Features() []string {
  71. f := [...]string{"power"}
  72. return f[:]
  73. }
  74. func (item *testPipelineItem) Initialize(repository *git.Repository) error {
  75. if item.InitializePanics {
  76. panic("!")
  77. }
  78. item.Initialized = repository != nil
  79. item.Merged = new(bool)
  80. item.MergeState = new(int)
  81. if item.InitializeRaises {
  82. return errors.New("test2")
  83. }
  84. return nil
  85. }
  86. func (item *testPipelineItem) Consume(deps map[string]interface{}) (map[string]interface{}, error) {
  87. if item.TestError {
  88. return nil, errors.New("error")
  89. }
  90. if item.ConsumePanics {
  91. panic("!")
  92. }
  93. obj, exists := deps[DependencyCommit]
  94. item.DepsConsumed = exists
  95. if item.DepsConsumed {
  96. commit := obj.(*object.Commit)
  97. item.CommitMatches = commit.Hash == plumbing.NewHash(
  98. "af9ddc0db70f09f3f27b4b98e415592a7485171c")
  99. obj, item.DepsConsumed = deps[DependencyIndex]
  100. if item.DepsConsumed {
  101. item.IndexMatches = obj.(int) == 0
  102. }
  103. }
  104. obj, exists = deps[DependencyIsMerge]
  105. if exists {
  106. *item.MergeState++
  107. if obj.(bool) {
  108. *item.MergeState++
  109. }
  110. }
  111. return map[string]interface{}{"test": item}, nil
  112. }
  113. func (item *testPipelineItem) Dispose() {
  114. item.Disposed = true
  115. }
  116. func (item *testPipelineItem) Fork(n int) []PipelineItem {
  117. result := make([]PipelineItem, n)
  118. for i := 0; i < n; i++ {
  119. result[i] = &testPipelineItem{Merged: item.Merged, MergeState: item.MergeState}
  120. }
  121. item.Forked = true
  122. return result
  123. }
  124. func (item *testPipelineItem) Merge(branches []PipelineItem) {
  125. *item.Merged = true
  126. }
  127. func (item *testPipelineItem) Finalize() interface{} {
  128. return item
  129. }
  130. func (item *testPipelineItem) Serialize(result interface{}, binary bool, writer io.Writer) error {
  131. return nil
  132. }
  133. type dependingTestPipelineItem struct {
  134. DependencySatisfied bool
  135. TestNilConsumeReturn bool
  136. Hibernated bool
  137. Booted bool
  138. RaiseHibernateError bool
  139. RaiseBootError bool
  140. }
  141. func (item *dependingTestPipelineItem) Name() string {
  142. return "Test2"
  143. }
  144. func (item *dependingTestPipelineItem) Provides() []string {
  145. arr := [...]string{"test2"}
  146. return arr[:]
  147. }
  148. func (item *dependingTestPipelineItem) Requires() []string {
  149. arr := [...]string{"test"}
  150. return arr[:]
  151. }
  152. func (item *dependingTestPipelineItem) ListConfigurationOptions() []ConfigurationOption {
  153. options := [...]ConfigurationOption{{
  154. Name: "TestOption2",
  155. Description: "The option description.",
  156. Flag: "test-option2",
  157. Type: IntConfigurationOption,
  158. Default: 10,
  159. }}
  160. return options[:]
  161. }
  162. func (item *dependingTestPipelineItem) Configure(facts map[string]interface{}) error {
  163. return nil
  164. }
  165. func (item *dependingTestPipelineItem) Initialize(repository *git.Repository) error {
  166. return nil
  167. }
  168. func (item *dependingTestPipelineItem) Flag() string {
  169. return "depflag"
  170. }
  171. func (item *dependingTestPipelineItem) Description() string {
  172. return "another description"
  173. }
  174. func (item *dependingTestPipelineItem) Consume(deps map[string]interface{}) (map[string]interface{}, error) {
  175. _, exists := deps["test"]
  176. item.DependencySatisfied = exists
  177. if !item.TestNilConsumeReturn {
  178. return map[string]interface{}{"test2": item}, nil
  179. }
  180. return nil, nil
  181. }
  182. func (item *dependingTestPipelineItem) Fork(n int) []PipelineItem {
  183. clones := make([]PipelineItem, n)
  184. for i := range clones {
  185. clones[i] = item
  186. }
  187. return clones
  188. }
  189. func (item *dependingTestPipelineItem) Merge(branches []PipelineItem) {
  190. }
  191. func (item *dependingTestPipelineItem) Hibernate() error {
  192. item.Hibernated = true
  193. if item.RaiseHibernateError {
  194. return errors.New("error")
  195. }
  196. return nil
  197. }
  198. func (item *dependingTestPipelineItem) Boot() error {
  199. item.Booted = true
  200. if item.RaiseBootError {
  201. return errors.New("error")
  202. }
  203. return nil
  204. }
  205. func (item *dependingTestPipelineItem) Finalize() interface{} {
  206. return true
  207. }
  208. func (item *dependingTestPipelineItem) Serialize(result interface{}, binary bool, writer io.Writer) error {
  209. return nil
  210. }
  211. func TestPipelineFacts(t *testing.T) {
  212. pipeline := NewPipeline(test.Repository)
  213. pipeline.SetFact("fact", "value")
  214. assert.Equal(t, pipeline.GetFact("fact"), "value")
  215. }
  216. func TestPipelineFeatures(t *testing.T) {
  217. pipeline := NewPipeline(test.Repository)
  218. pipeline.SetFeature("feat")
  219. val, _ := pipeline.GetFeature("feat")
  220. assert.True(t, val)
  221. _, exists := pipeline.GetFeature("!")
  222. assert.False(t, exists)
  223. Registry.featureFlags.Set("777")
  224. defer func() {
  225. Registry.featureFlags = arrayFeatureFlags{Flags: []string{}, Choices: map[string]bool{}}
  226. }()
  227. pipeline.SetFeaturesFromFlags()
  228. _, exists = pipeline.GetFeature("777")
  229. assert.False(t, exists)
  230. assert.Panics(t, func() {
  231. pipeline.SetFeaturesFromFlags(
  232. &PipelineItemRegistry{}, &PipelineItemRegistry{})
  233. })
  234. }
  235. func TestPipelineErrors(t *testing.T) {
  236. pipeline := NewPipeline(test.Repository)
  237. pipeline.SetFact("fact", "value")
  238. assert.Equal(t, pipeline.GetFact("fact"), "value")
  239. item := &testPipelineItem{}
  240. pipeline.AddItem(item)
  241. item.ConfigureRaises = true
  242. err := pipeline.Initialize(map[string]interface{}{})
  243. assert.Error(t, err)
  244. assert.Contains(t, err.Error(), "configure")
  245. assert.Contains(t, err.Error(), "test1")
  246. item.ConfigureRaises = false
  247. item.InitializeRaises = true
  248. err = pipeline.Initialize(map[string]interface{}{})
  249. assert.Error(t, err)
  250. assert.Contains(t, err.Error(), "initialize")
  251. assert.Contains(t, err.Error(), "test2")
  252. item.InitializeRaises = false
  253. item.InitializePanics = true
  254. assert.Panics(t, func() { pipeline.Initialize(map[string]interface{}{}) })
  255. }
  256. func TestPipelineInitialize(t *testing.T) {
  257. t.Run("without logger fact", func(t *testing.T) {
  258. pipeline := NewPipeline(test.Repository)
  259. item := &testPipelineItem{}
  260. pipeline.AddItem(item)
  261. require.NoError(t, pipeline.Initialize(map[string]interface{}{}))
  262. // pipeline logger should be initialized, and item logger should be the same
  263. require.NotNil(t, pipeline.l)
  264. assert.Equal(t, pipeline.l, item.Logger)
  265. })
  266. t.Run("with logger fact", func(t *testing.T) {
  267. pipeline := NewPipeline(test.Repository)
  268. item := &testPipelineItem{}
  269. logger := NewLogger()
  270. pipeline.AddItem(item)
  271. require.NoError(t, pipeline.Initialize(map[string]interface{}{
  272. ConfigLogger: logger,
  273. }))
  274. // pipeline logger should be set, and the item logger should be the same
  275. assert.Equal(t, logger, pipeline.l)
  276. assert.Equal(t, logger, item.Logger)
  277. })
  278. }
  279. func TestPipelineRun(t *testing.T) {
  280. pipeline := NewPipeline(test.Repository)
  281. item := &testPipelineItem{}
  282. pipeline.AddItem(item)
  283. assert.NoError(t, pipeline.Initialize(map[string]interface{}{}))
  284. assert.True(t, item.Initialized)
  285. commits := make([]*object.Commit, 1)
  286. commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
  287. "af9ddc0db70f09f3f27b4b98e415592a7485171c"))
  288. result, err := pipeline.Run(commits)
  289. assert.NoError(t, err)
  290. assert.Equal(t, 2, len(result))
  291. assert.Equal(t, item, result[item].(*testPipelineItem))
  292. common := result[nil].(*CommonAnalysisResult)
  293. assert.Equal(t, common.BeginTime, int64(1481719198))
  294. assert.Equal(t, common.EndTime, int64(1481719198))
  295. assert.Equal(t, common.CommitsNumber, 1)
  296. assert.True(t, common.RunTime.Nanoseconds()/1e6 < 100)
  297. assert.Len(t, common.RunTimePerItem, 1)
  298. for key, val := range common.RunTimePerItem {
  299. assert.True(t, val >= 0, key)
  300. }
  301. assert.True(t, item.DepsConsumed)
  302. assert.True(t, item.Disposed)
  303. assert.True(t, item.CommitMatches)
  304. assert.True(t, item.IndexMatches)
  305. assert.Equal(t, 1, *item.MergeState)
  306. assert.True(t, item.Forked)
  307. assert.False(t, *item.Merged)
  308. pipeline.RemoveItem(item)
  309. result, err = pipeline.Run(commits)
  310. assert.Nil(t, err)
  311. assert.Equal(t, 1, len(result))
  312. }
  313. func TestPipelineRunBranches(t *testing.T) {
  314. pipeline := NewPipeline(test.Repository)
  315. item := &testPipelineItem{}
  316. pipeline.AddItem(item)
  317. pipeline.Initialize(map[string]interface{}{})
  318. assert.True(t, item.Initialized)
  319. hashes := []string{
  320. "6db8065cdb9bb0758f36a7e75fc72ab95f9e8145",
  321. "f30daba81ff2bf0b3ba02a1e1441e74f8a4f6fee",
  322. "8a03b5620b1caa72ec9cb847ea88332621e2950a",
  323. "dd9dd084d5851d7dc4399fc7dbf3d8292831ebc5",
  324. "f4ed0405b14f006c0744029d87ddb3245607587a",
  325. }
  326. commits := make([]*object.Commit, len(hashes))
  327. for i, h := range hashes {
  328. var err error
  329. commits[i], err = test.Repository.CommitObject(plumbing.NewHash(h))
  330. if err != nil {
  331. t.Fatal(err)
  332. }
  333. }
  334. result, err := pipeline.Run(commits)
  335. assert.Nil(t, err)
  336. assert.True(t, item.Forked)
  337. assert.True(t, *item.Merged)
  338. assert.Equal(t, 2, len(result))
  339. assert.Equal(t, item, result[item].(*testPipelineItem))
  340. common := result[nil].(*CommonAnalysisResult)
  341. assert.Equal(t, common.CommitsNumber, 5)
  342. assert.Equal(t, *item.MergeState, 8)
  343. }
  344. func TestPipelineOnProgress(t *testing.T) {
  345. pipeline := NewPipeline(test.Repository)
  346. progressOk := 0
  347. onProgress := func(step int, total int, action string) {
  348. if step == 1 && total == 4 && action == "emerge" {
  349. progressOk++
  350. }
  351. if step == 2 && total == 4 && action == "af9ddc0" {
  352. progressOk++
  353. }
  354. if step == 3 && total == 4 && action == "finalize" {
  355. progressOk++
  356. }
  357. if step == 4 && total == 4 && action == "" {
  358. progressOk++
  359. }
  360. }
  361. pipeline.OnProgress = onProgress
  362. commits := make([]*object.Commit, 1)
  363. commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
  364. "af9ddc0db70f09f3f27b4b98e415592a7485171c"))
  365. result, err := pipeline.Run(commits)
  366. assert.Nil(t, err)
  367. assert.Equal(t, 1, len(result))
  368. assert.Equal(t, 4, progressOk)
  369. }
  370. func TestPipelineCommitsFull(t *testing.T) {
  371. pipeline := NewPipeline(test.Repository)
  372. commits, err := pipeline.Commits(false)
  373. assert.Nil(t, err)
  374. assert.True(t, len(commits) >= 100)
  375. hashMap := map[plumbing.Hash]bool{}
  376. for _, c := range commits {
  377. hashMap[c.Hash] = true
  378. }
  379. assert.Equal(t, len(commits), len(hashMap))
  380. assert.Contains(t, hashMap, plumbing.NewHash(
  381. "cce947b98a050c6d356bc6ba95030254914027b1"))
  382. assert.Contains(t, hashMap, plumbing.NewHash(
  383. "a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
  384. }
  385. func TestPipelineCommitsFirstParent(t *testing.T) {
  386. pipeline := NewPipeline(test.Repository)
  387. commits, err := pipeline.Commits(true)
  388. assert.Nil(t, err)
  389. assert.True(t, len(commits) >= 100)
  390. hashMap := map[plumbing.Hash]bool{}
  391. for _, c := range commits {
  392. hashMap[c.Hash] = true
  393. }
  394. assert.Equal(t, len(commits), len(hashMap))
  395. assert.Contains(t, hashMap, plumbing.NewHash(
  396. "cce947b98a050c6d356bc6ba95030254914027b1"))
  397. assert.NotContains(t, hashMap, plumbing.NewHash(
  398. "a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
  399. }
  400. func TestLoadCommitsFromFile(t *testing.T) {
  401. tmp, err := ioutil.TempFile("", "hercules-test-")
  402. assert.Nil(t, err)
  403. tmp.WriteString("cce947b98a050c6d356bc6ba95030254914027b1\n6db8065cdb9bb0758f36a7e75fc72ab95f9e8145")
  404. tmp.Close()
  405. defer os.Remove(tmp.Name())
  406. commits, err := LoadCommitsFromFile(tmp.Name(), test.Repository)
  407. assert.Nil(t, err)
  408. assert.Equal(t, len(commits), 2)
  409. assert.Equal(t, commits[0].Hash, plumbing.NewHash(
  410. "cce947b98a050c6d356bc6ba95030254914027b1"))
  411. assert.Equal(t, commits[1].Hash, plumbing.NewHash(
  412. "6db8065cdb9bb0758f36a7e75fc72ab95f9e8145"))
  413. commits, err = LoadCommitsFromFile("/WAT?xxx!", test.Repository)
  414. assert.Nil(t, commits)
  415. assert.NotNil(t, err)
  416. tmp, err = ioutil.TempFile("", "hercules-test-")
  417. assert.Nil(t, err)
  418. tmp.WriteString("WAT")
  419. tmp.Close()
  420. defer os.Remove(tmp.Name())
  421. commits, err = LoadCommitsFromFile(tmp.Name(), test.Repository)
  422. assert.Nil(t, commits)
  423. assert.NotNil(t, err)
  424. tmp, err = ioutil.TempFile("", "hercules-test-")
  425. assert.Nil(t, err)
  426. tmp.WriteString("ffffffffffffffffffffffffffffffffffffffff")
  427. tmp.Close()
  428. defer os.Remove(tmp.Name())
  429. commits, err = LoadCommitsFromFile(tmp.Name(), test.Repository)
  430. assert.Nil(t, commits)
  431. assert.NotNil(t, err)
  432. }
  433. func TestPipelineDeps(t *testing.T) {
  434. pipeline := NewPipeline(test.Repository)
  435. item1 := &dependingTestPipelineItem{}
  436. item2 := &testPipelineItem{}
  437. pipeline.AddItem(item1)
  438. pipeline.AddItem(item2)
  439. assert.Equal(t, pipeline.Len(), 2)
  440. pipeline.Initialize(map[string]interface{}{})
  441. commits := make([]*object.Commit, 1)
  442. commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
  443. "af9ddc0db70f09f3f27b4b98e415592a7485171c"))
  444. result, err := pipeline.Run(commits)
  445. assert.NoError(t, err)
  446. assert.True(t, result[item1].(bool))
  447. assert.Equal(t, result[item2], item2)
  448. item1.TestNilConsumeReturn = true
  449. _, err = pipeline.Run(commits)
  450. assert.Error(t, err)
  451. }
  452. func TestPipelineDeployFeatures(t *testing.T) {
  453. pipeline := NewPipeline(test.Repository)
  454. pipeline.DeployItem(&testPipelineItem{})
  455. f, _ := pipeline.GetFeature("power")
  456. assert.True(t, f)
  457. }
  458. func TestPipelineError(t *testing.T) {
  459. pipeline := NewPipeline(test.Repository)
  460. item := &testPipelineItem{}
  461. item.TestError = true
  462. pipeline.AddItem(item)
  463. pipeline.Initialize(map[string]interface{}{})
  464. commits := make([]*object.Commit, 1)
  465. commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
  466. "af9ddc0db70f09f3f27b4b98e415592a7485171c"))
  467. result, err := pipeline.Run(commits)
  468. assert.Nil(t, result)
  469. assert.NotNil(t, err)
  470. }
  471. func TestPipelineDryRun(t *testing.T) {
  472. pipeline := NewPipeline(test.Repository)
  473. item := &testPipelineItem{}
  474. item.TestError = true
  475. pipeline.AddItem(item)
  476. pipeline.DryRun = true
  477. pipeline.Initialize(map[string]interface{}{})
  478. assert.True(t, pipeline.DryRun)
  479. pipeline.DryRun = false
  480. pipeline.Initialize(map[string]interface{}{ConfigPipelineDryRun: true})
  481. assert.True(t, pipeline.DryRun)
  482. commits := make([]*object.Commit, 1)
  483. commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
  484. "af9ddc0db70f09f3f27b4b98e415592a7485171c"))
  485. result, err := pipeline.Run(commits)
  486. assert.NotNil(t, result)
  487. assert.Len(t, result, 1)
  488. assert.Contains(t, result, nil)
  489. assert.Nil(t, err)
  490. }
  491. func TestPipelineDryRunFalse(t *testing.T) {
  492. pipeline := NewPipeline(test.Repository)
  493. item := &testPipelineItem{}
  494. pipeline.AddItem(item)
  495. pipeline.Initialize(map[string]interface{}{ConfigPipelineDryRun: false})
  496. commits := make([]*object.Commit, 1)
  497. commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
  498. "af9ddc0db70f09f3f27b4b98e415592a7485171c"))
  499. result, err := pipeline.Run(commits)
  500. assert.NotNil(t, result)
  501. assert.Len(t, result, 2)
  502. assert.Contains(t, result, nil)
  503. assert.Contains(t, result, item)
  504. assert.Nil(t, err)
  505. assert.True(t, item.DepsConsumed)
  506. assert.True(t, item.CommitMatches)
  507. assert.True(t, item.IndexMatches)
  508. assert.Equal(t, 1, *item.MergeState)
  509. assert.True(t, item.Forked)
  510. assert.False(t, *item.Merged)
  511. }
  512. func TestPipelineDumpPlanConfigure(t *testing.T) {
  513. pipeline := NewPipeline(test.Repository)
  514. item := &testPipelineItem{}
  515. pipeline.AddItem(item)
  516. pipeline.DumpPlan = true
  517. pipeline.DryRun = true
  518. pipeline.Initialize(map[string]interface{}{})
  519. assert.True(t, pipeline.DumpPlan)
  520. pipeline.DumpPlan = false
  521. pipeline.Initialize(map[string]interface{}{ConfigPipelineDumpPlan: true})
  522. assert.True(t, pipeline.DumpPlan)
  523. stream := &bytes.Buffer{}
  524. backupPlanPrintFunc := planPrintFunc
  525. planPrintFunc = func(args ...interface{}) {
  526. fmt.Fprintln(stream, args...)
  527. }
  528. defer func() {
  529. planPrintFunc = backupPlanPrintFunc
  530. }()
  531. commits := make([]*object.Commit, 1)
  532. commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
  533. "af9ddc0db70f09f3f27b4b98e415592a7485171c"))
  534. result, err := pipeline.Run(commits)
  535. assert.NotNil(t, result)
  536. assert.Len(t, result, 1)
  537. assert.Contains(t, result, nil)
  538. assert.Nil(t, err)
  539. assert.Equal(t, `E [1]
  540. C 1 af9ddc0db70f09f3f27b4b98e415592a7485171c
  541. `, stream.String())
  542. }
  543. func TestCommonAnalysisResultCopy(t *testing.T) {
  544. c1 := CommonAnalysisResult{
  545. BeginTime: 1513620635, EndTime: 1513720635, CommitsNumber: 1, RunTime: 100,
  546. RunTimePerItem: map[string]float64{"one": 1, "two": 2}}
  547. c2 := c1.Copy()
  548. assert.Equal(t, c1, c2)
  549. c2.RunTimePerItem["one"] = 100500
  550. assert.Equal(t, c1.RunTimePerItem["one"], float64(1))
  551. }
  552. func TestCommonAnalysisResultMerge(t *testing.T) {
  553. c1 := CommonAnalysisResult{
  554. BeginTime: 1513620635, EndTime: 1513720635, CommitsNumber: 1, RunTime: 100,
  555. RunTimePerItem: map[string]float64{"one": 1, "two": 2}}
  556. assert.Equal(t, c1.BeginTimeAsTime().Unix(), int64(1513620635))
  557. assert.Equal(t, c1.EndTimeAsTime().Unix(), int64(1513720635))
  558. c2 := CommonAnalysisResult{
  559. BeginTime: 1513620535, EndTime: 1513730635, CommitsNumber: 2, RunTime: 200,
  560. RunTimePerItem: map[string]float64{"two": 4, "three": 8}}
  561. c1.Merge(&c2)
  562. assert.Equal(t, c1.BeginTime, int64(1513620535))
  563. assert.Equal(t, c1.EndTime, int64(1513730635))
  564. assert.Equal(t, c1.CommitsNumber, 3)
  565. assert.Equal(t, c1.RunTime.Nanoseconds(), int64(300))
  566. assert.Equal(t, c1.RunTimePerItem, map[string]float64{"one": 1, "two": 6, "three": 8})
  567. }
  568. func TestCommonAnalysisResultMetadata(t *testing.T) {
  569. c1 := &CommonAnalysisResult{
  570. BeginTime: 1513620635, EndTime: 1513720635, CommitsNumber: 1, RunTime: 100 * 1e6,
  571. RunTimePerItem: map[string]float64{"one": 1, "two": 2}}
  572. meta := &pb.Metadata{}
  573. c1 = MetadataToCommonAnalysisResult(c1.FillMetadata(meta))
  574. assert.Equal(t, c1.BeginTimeAsTime().Unix(), int64(1513620635))
  575. assert.Equal(t, c1.EndTimeAsTime().Unix(), int64(1513720635))
  576. assert.Equal(t, c1.CommitsNumber, 1)
  577. assert.Equal(t, c1.RunTime.Nanoseconds(), int64(100*1e6))
  578. assert.Equal(t, c1.RunTimePerItem, map[string]float64{"one": 1, "two": 2})
  579. }
  580. func TestConfigurationOptionTypeString(t *testing.T) {
  581. opt := ConfigurationOptionType(0)
  582. assert.Equal(t, opt.String(), "")
  583. opt = ConfigurationOptionType(1)
  584. assert.Equal(t, opt.String(), "int")
  585. opt = ConfigurationOptionType(2)
  586. assert.Equal(t, opt.String(), "string")
  587. opt = ConfigurationOptionType(3)
  588. assert.Equal(t, opt.String(), "float")
  589. opt = ConfigurationOptionType(4)
  590. assert.Equal(t, opt.String(), "string")
  591. opt = ConfigurationOptionType(5)
  592. assert.Equal(t, opt.String(), "path")
  593. opt = ConfigurationOptionType(6)
  594. assert.Panics(t, func() { _ = opt.String() })
  595. }
  596. func TestConfigurationOptionFormatDefault(t *testing.T) {
  597. opt := ConfigurationOption{Type: StringConfigurationOption, Default: "ololo"}
  598. assert.Equal(t, opt.FormatDefault(), "\"ololo\"")
  599. opt = ConfigurationOption{Type: IntConfigurationOption, Default: 7}
  600. assert.Equal(t, opt.FormatDefault(), "7")
  601. opt = ConfigurationOption{Type: BoolConfigurationOption, Default: false}
  602. assert.Equal(t, opt.FormatDefault(), "false")
  603. opt = ConfigurationOption{Type: FloatConfigurationOption, Default: 0.5}
  604. assert.Equal(t, opt.FormatDefault(), "0.5")
  605. }
  606. func TestPrepareRunPlanTiny(t *testing.T) {
  607. rootCommit, err := test.Repository.CommitObject(plumbing.NewHash(
  608. "cce947b98a050c6d356bc6ba95030254914027b1"))
  609. if err != nil {
  610. t.Fatal(err)
  611. }
  612. plan := prepareRunPlan([]*object.Commit{rootCommit}, 0, true)
  613. assert.Len(t, plan, 2)
  614. assert.Equal(t, runActionEmerge, plan[0].Action)
  615. assert.Equal(t, rootBranchIndex, plan[0].Items[0])
  616. assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[0].Commit.Hash.String())
  617. assert.Equal(t, runActionCommit, plan[1].Action)
  618. assert.Equal(t, rootBranchIndex, plan[1].Items[0])
  619. assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[1].Commit.Hash.String())
  620. }
  621. func TestPrepareRunPlanSmall(t *testing.T) {
  622. cit, err := test.Repository.Log(&git.LogOptions{From: plumbing.ZeroHash})
  623. if err != nil {
  624. panic(err)
  625. }
  626. defer cit.Close()
  627. var commits []*object.Commit
  628. timeCutoff := time.Date(2016, 12, 15, 0, 0, 0, 0, time.FixedZone("CET", 7200))
  629. cit.ForEach(func(commit *object.Commit) error {
  630. reliableTime := time.Date(commit.Author.When.Year(), commit.Author.When.Month(),
  631. commit.Author.When.Day(), commit.Author.When.Hour(), commit.Author.When.Minute(),
  632. commit.Author.When.Second(), 0, time.FixedZone("CET", 7200))
  633. if reliableTime.Before(timeCutoff) {
  634. commits = append(commits, commit)
  635. }
  636. return nil
  637. })
  638. plan := prepareRunPlan(commits, 0, false)
  639. /*for _, p := range plan {
  640. if p.Commit != nil {
  641. fmt.Println(p.Action, p.Commit.Hash.String(), p.Items)
  642. } else {
  643. fmt.Println(p.Action, strings.Repeat(" ", 40), p.Items)
  644. }
  645. }*/
  646. // fork, merge and one artificial commit per branch
  647. assert.Len(t, plan, len(commits)+1)
  648. assert.Equal(t, runActionEmerge, plan[0].Action)
  649. assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[0].Commit.Hash.String())
  650. assert.Equal(t, rootBranchIndex, plan[0].Items[0])
  651. assert.Equal(t, runActionCommit, plan[1].Action)
  652. assert.Equal(t, rootBranchIndex, plan[1].Items[0])
  653. assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[1].Commit.Hash.String())
  654. assert.Equal(t, runActionCommit, plan[2].Action)
  655. assert.Equal(t, rootBranchIndex, plan[2].Items[0])
  656. assert.Equal(t, "a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3", plan[2].Commit.Hash.String())
  657. assert.Equal(t, runActionCommit, plan[10].Action)
  658. assert.Equal(t, rootBranchIndex, plan[10].Items[0])
  659. assert.Equal(t, "a28e9064c70618dc9d68e1401b889975e0680d11", plan[10].Commit.Hash.String())
  660. }
  661. func TestMergeDag(t *testing.T) {
  662. cit, err := test.Repository.Log(&git.LogOptions{From: plumbing.ZeroHash})
  663. if err != nil {
  664. panic(err)
  665. }
  666. defer cit.Close()
  667. var commits []*object.Commit
  668. timeCutoff := time.Date(2017, 8, 12, 0, 0, 0, 0, time.FixedZone("CET", 7200))
  669. cit.ForEach(func(commit *object.Commit) error {
  670. reliableTime := time.Date(commit.Author.When.Year(), commit.Author.When.Month(),
  671. commit.Author.When.Day(), commit.Author.When.Hour(), commit.Author.When.Minute(),
  672. commit.Author.When.Second(), 0, time.FixedZone("CET", 7200))
  673. if reliableTime.Before(timeCutoff) {
  674. commits = append(commits, commit)
  675. }
  676. return nil
  677. })
  678. hashes, dag := buildDag(commits)
  679. leaveRootComponent(hashes, dag)
  680. mergedDag, _ := mergeDag(hashes, dag)
  681. for key, vals := range mergedDag {
  682. if key != plumbing.NewHash("a28e9064c70618dc9d68e1401b889975e0680d11") &&
  683. key != plumbing.NewHash("db325a212d0bc99b470e000641d814745024bbd5") {
  684. assert.Len(t, vals, len(dag[key]), key.String())
  685. } else {
  686. mvals := map[string]bool{}
  687. for _, val := range vals {
  688. mvals[val.Hash.String()] = true
  689. }
  690. if key == plumbing.NewHash("a28e9064c70618dc9d68e1401b889975e0680d11") {
  691. assert.Contains(t, mvals, "db325a212d0bc99b470e000641d814745024bbd5")
  692. assert.Contains(t, mvals, "be9b61e09b08b98e64ed461a4004c9e2412f78ee")
  693. }
  694. if key == plumbing.NewHash("db325a212d0bc99b470e000641d814745024bbd5") {
  695. assert.Contains(t, mvals, "f30daba81ff2bf0b3ba02a1e1441e74f8a4f6fee")
  696. assert.Contains(t, mvals, "8a03b5620b1caa72ec9cb847ea88332621e2950a")
  697. }
  698. }
  699. }
  700. assert.Len(t, mergedDag, 8)
  701. assert.Contains(t, mergedDag, plumbing.NewHash("cce947b98a050c6d356bc6ba95030254914027b1"))
  702. assert.Contains(t, mergedDag, plumbing.NewHash("a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
  703. assert.Contains(t, mergedDag, plumbing.NewHash("a28e9064c70618dc9d68e1401b889975e0680d11"))
  704. assert.Contains(t, mergedDag, plumbing.NewHash("be9b61e09b08b98e64ed461a4004c9e2412f78ee"))
  705. assert.Contains(t, mergedDag, plumbing.NewHash("db325a212d0bc99b470e000641d814745024bbd5"))
  706. assert.Contains(t, mergedDag, plumbing.NewHash("f30daba81ff2bf0b3ba02a1e1441e74f8a4f6fee"))
  707. assert.Contains(t, mergedDag, plumbing.NewHash("8a03b5620b1caa72ec9cb847ea88332621e2950a"))
  708. assert.Contains(t, mergedDag, plumbing.NewHash("dd9dd084d5851d7dc4399fc7dbf3d8292831ebc5"))
  709. queue := []plumbing.Hash{plumbing.NewHash("cce947b98a050c6d356bc6ba95030254914027b1")}
  710. visited := map[plumbing.Hash]bool{}
  711. for len(queue) > 0 {
  712. head := queue[len(queue)-1]
  713. queue = queue[:len(queue)-1]
  714. if visited[head] {
  715. continue
  716. }
  717. visited[head] = true
  718. for _, child := range mergedDag[head] {
  719. queue = append(queue, child.Hash)
  720. }
  721. }
  722. assert.Len(t, visited, 8)
  723. }
  724. func TestPrepareRunPlanBig(t *testing.T) {
  725. cases := [][7]int{
  726. {2017, 8, 9, 0, 0, 0, 0},
  727. {2017, 8, 10, 0, 0, 0, 0},
  728. {2017, 8, 24, 1, 1, 1, 1},
  729. {2017, 9, 19, 1 - 2, 1, 1, 1},
  730. {2017, 9, 23, 1 - 2, 1, 1, 1},
  731. {2017, 12, 8, 1, 1, 1, 1},
  732. {2017, 12, 9, 1, 1, 1, 1},
  733. {2017, 12, 10, 1, 1, 1, 1},
  734. {2017, 12, 11, 2, 2, 2, 2},
  735. {2017, 12, 19, 3, 3, 3, 3},
  736. {2017, 12, 27, 3, 3, 3, 3},
  737. {2018, 1, 10, 3, 3, 3, 3},
  738. {2018, 1, 16, 3, 3, 3, 3},
  739. {2018, 1, 18, 4, 5, 4, 4},
  740. {2018, 1, 23, 5, 5, 5, 5},
  741. {2018, 3, 12, 6, 6, 6, 6},
  742. {2018, 5, 13, 6, 6, 6, 6},
  743. {2018, 5, 16, 7, 7, 7, 7},
  744. }
  745. for _, testCase := range cases {
  746. func() {
  747. cit, err := test.Repository.Log(&git.LogOptions{From: plumbing.ZeroHash})
  748. if err != nil {
  749. panic(err)
  750. }
  751. defer cit.Close()
  752. var commits []*object.Commit
  753. timeCutoff := time.Date(
  754. testCase[0], time.Month(testCase[1]), testCase[2], 0, 0, 0, 0, time.FixedZone("CET", 7200))
  755. cit.ForEach(func(commit *object.Commit) error {
  756. reliableTime := time.Date(commit.Author.When.Year(), commit.Author.When.Month(),
  757. commit.Author.When.Day(), commit.Author.When.Hour(), commit.Author.When.Minute(),
  758. commit.Author.When.Second(), 0, time.FixedZone("CET", 7200))
  759. if reliableTime.Before(timeCutoff) {
  760. commits = append(commits, commit)
  761. }
  762. return nil
  763. })
  764. plan := prepareRunPlan(commits, 0, false)
  765. /*for _, p := range plan {
  766. if p.Commit != nil {
  767. fmt.Println(p.Action, p.Commit.Hash.String(), p.Items)
  768. } else {
  769. fmt.Println(p.Action, strings.Repeat(" ", 40), p.Items)
  770. }
  771. }*/
  772. numCommits := 0
  773. numForks := 0
  774. numMerges := 0
  775. numDeletes := 0
  776. numEmerges := 0
  777. processed := map[plumbing.Hash]map[int]int{}
  778. for _, p := range plan {
  779. switch p.Action {
  780. case runActionCommit:
  781. branches := processed[p.Commit.Hash]
  782. if branches == nil {
  783. branches = map[int]int{}
  784. processed[p.Commit.Hash] = branches
  785. }
  786. branches[p.Items[0]]++
  787. for _, parent := range p.Commit.ParentHashes {
  788. assert.Contains(t, processed, parent)
  789. }
  790. numCommits++
  791. case runActionFork:
  792. numForks++
  793. case runActionMerge:
  794. counts := map[int]int{}
  795. for _, i := range p.Items {
  796. counts[i]++
  797. }
  798. for x, v := range counts {
  799. assert.Equal(t, 1, v, x)
  800. }
  801. numMerges++
  802. case runActionDelete:
  803. numDeletes++
  804. case runActionEmerge:
  805. numEmerges++
  806. }
  807. }
  808. for c, branches := range processed {
  809. for b, v := range branches {
  810. assert.Equal(t, 1, v, fmt.Sprint(c.String(), b))
  811. }
  812. }
  813. assert.Equal(t, numCommits, len(commits)+testCase[3], fmt.Sprintf("commits %v", testCase))
  814. assert.Equal(t, numForks, testCase[4], fmt.Sprintf("forks %v", testCase))
  815. assert.Equal(t, numMerges, testCase[5], fmt.Sprintf("merges %v", testCase))
  816. assert.Equal(t, numDeletes, testCase[6], fmt.Sprintf("deletes %v", testCase))
  817. assert.Equal(t, numEmerges, 1, fmt.Sprintf("emerges %v", testCase))
  818. }()
  819. }
  820. }
  821. func TestPipelineRunHibernation(t *testing.T) {
  822. pipeline := NewPipeline(test.Repository)
  823. pipeline.HibernationDistance = 2
  824. pipeline.AddItem(&testPipelineItem{})
  825. item := &dependingTestPipelineItem{}
  826. pipeline.AddItem(item)
  827. pipeline.Initialize(map[string]interface{}{})
  828. hashes := []string{
  829. "0183e08978007c746468fca9f68e6e2fbf32100c",
  830. "b467a682f680a4dcfd74869480a52f8be3a4fdf0",
  831. "31c9f752f9ce103e85523442fa3f05b1ff4ea546",
  832. "6530890fcd02fb5e6e85ce2951fdd5c555f2c714",
  833. "feb2d230777cbb492ecbc27dea380dc1e7b8f437",
  834. "9b30d2abc043ab59aa7ec7b50970c65c90b98853",
  835. }
  836. commits := make([]*object.Commit, len(hashes))
  837. for i, h := range hashes {
  838. var err error
  839. commits[i], err = test.Repository.CommitObject(plumbing.NewHash(h))
  840. if err != nil {
  841. t.Fatal(err)
  842. }
  843. }
  844. pipeline.PrintActions = true
  845. _, err := pipeline.Run(commits)
  846. assert.NoError(t, err)
  847. assert.True(t, item.Hibernated)
  848. assert.True(t, item.Booted)
  849. item.RaiseHibernateError = true
  850. _, err = pipeline.Run(commits)
  851. assert.Error(t, err)
  852. item.RaiseHibernateError = false
  853. pipeline.Run(commits)
  854. item.RaiseBootError = true
  855. _, err = pipeline.Run(commits)
  856. assert.Error(t, err)
  857. }