| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891 | 
							- package core
 
- import (
 
- 	"bytes"
 
- 	"errors"
 
- 	"fmt"
 
- 	"io"
 
- 	"io/ioutil"
 
- 	"os"
 
- 	"testing"
 
- 	"time"
 
- 	"github.com/stretchr/testify/assert"
 
- 	"gopkg.in/src-d/go-git.v4"
 
- 	"gopkg.in/src-d/go-git.v4/plumbing"
 
- 	"gopkg.in/src-d/go-git.v4/plumbing/object"
 
- 	"gopkg.in/src-d/hercules.v10/internal/pb"
 
- 	"gopkg.in/src-d/hercules.v10/internal/test"
 
- )
 
- type testPipelineItem struct {
 
- 	Initialized      bool
 
- 	DepsConsumed     bool
 
- 	Disposed         bool
 
- 	Forked           bool
 
- 	Merged           *bool
 
- 	CommitMatches    bool
 
- 	IndexMatches     bool
 
- 	MergeState       *int
 
- 	TestError        bool
 
- 	ConfigureRaises  bool
 
- 	InitializeRaises bool
 
- 	InitializePanics bool
 
- 	ConsumePanics    bool
 
- }
 
- func (item *testPipelineItem) Name() string {
 
- 	return "Test"
 
- }
 
- func (item *testPipelineItem) Provides() []string {
 
- 	arr := [...]string{"test"}
 
- 	return arr[:]
 
- }
 
- func (item *testPipelineItem) Requires() []string {
 
- 	return []string{}
 
- }
 
- func (item *testPipelineItem) Configure(facts map[string]interface{}) error {
 
- 	if item.ConfigureRaises {
 
- 		return errors.New("test1")
 
- 	}
 
- 	return nil
 
- }
 
- func (item *testPipelineItem) ListConfigurationOptions() []ConfigurationOption {
 
- 	options := [...]ConfigurationOption{{
 
- 		Name:        "TestOption",
 
- 		Description: "The option description.",
 
- 		Flag:        "test-option",
 
- 		Type:        IntConfigurationOption,
 
- 		Default:     10,
 
- 	}}
 
- 	return options[:]
 
- }
 
- func (item *testPipelineItem) Flag() string {
 
- 	return "mytest"
 
- }
 
- func (item *testPipelineItem) Description() string {
 
- 	return "description!"
 
- }
 
- func (item *testPipelineItem) Features() []string {
 
- 	f := [...]string{"power"}
 
- 	return f[:]
 
- }
 
- func (item *testPipelineItem) Initialize(repository *git.Repository) error {
 
- 	if item.InitializePanics {
 
- 		panic("!")
 
- 	}
 
- 	item.Initialized = repository != nil
 
- 	item.Merged = new(bool)
 
- 	item.MergeState = new(int)
 
- 	if item.InitializeRaises {
 
- 		return errors.New("test2")
 
- 	}
 
- 	return nil
 
- }
 
- func (item *testPipelineItem) Consume(deps map[string]interface{}) (map[string]interface{}, error) {
 
- 	if item.TestError {
 
- 		return nil, errors.New("error")
 
- 	}
 
- 	if item.ConsumePanics {
 
- 		panic("!")
 
- 	}
 
- 	obj, exists := deps[DependencyCommit]
 
- 	item.DepsConsumed = exists
 
- 	if item.DepsConsumed {
 
- 		commit := obj.(*object.Commit)
 
- 		item.CommitMatches = commit.Hash == plumbing.NewHash(
 
- 			"af9ddc0db70f09f3f27b4b98e415592a7485171c")
 
- 		obj, item.DepsConsumed = deps[DependencyIndex]
 
- 		if item.DepsConsumed {
 
- 			item.IndexMatches = obj.(int) == 0
 
- 		}
 
- 	}
 
- 	obj, exists = deps[DependencyIsMerge]
 
- 	if exists {
 
- 		*item.MergeState++
 
- 		if obj.(bool) {
 
- 			*item.MergeState++
 
- 		}
 
- 	}
 
- 	return map[string]interface{}{"test": item}, nil
 
- }
 
- func (item *testPipelineItem) Dispose() {
 
- 	item.Disposed = true
 
- }
 
- func (item *testPipelineItem) Fork(n int) []PipelineItem {
 
- 	result := make([]PipelineItem, n)
 
- 	for i := 0; i < n; i++ {
 
- 		result[i] = &testPipelineItem{Merged: item.Merged, MergeState: item.MergeState}
 
- 	}
 
- 	item.Forked = true
 
- 	return result
 
- }
 
- func (item *testPipelineItem) Merge(branches []PipelineItem) {
 
- 	*item.Merged = true
 
- }
 
- func (item *testPipelineItem) Finalize() interface{} {
 
- 	return item
 
- }
 
- func (item *testPipelineItem) Serialize(result interface{}, binary bool, writer io.Writer) error {
 
- 	return nil
 
- }
 
- type dependingTestPipelineItem struct {
 
- 	DependencySatisfied  bool
 
- 	TestNilConsumeReturn bool
 
- 	Hibernated           bool
 
- 	Booted               bool
 
- 	RaiseHibernateError  bool
 
- 	RaiseBootError       bool
 
- }
 
- func (item *dependingTestPipelineItem) Name() string {
 
- 	return "Test2"
 
- }
 
- func (item *dependingTestPipelineItem) Provides() []string {
 
- 	arr := [...]string{"test2"}
 
- 	return arr[:]
 
- }
 
- func (item *dependingTestPipelineItem) Requires() []string {
 
- 	arr := [...]string{"test"}
 
- 	return arr[:]
 
- }
 
- func (item *dependingTestPipelineItem) ListConfigurationOptions() []ConfigurationOption {
 
- 	options := [...]ConfigurationOption{{
 
- 		Name:        "TestOption2",
 
- 		Description: "The option description.",
 
- 		Flag:        "test-option2",
 
- 		Type:        IntConfigurationOption,
 
- 		Default:     10,
 
- 	}}
 
- 	return options[:]
 
- }
 
- func (item *dependingTestPipelineItem) Configure(facts map[string]interface{}) error {
 
- 	return nil
 
- }
 
- func (item *dependingTestPipelineItem) Initialize(repository *git.Repository) error {
 
- 	return nil
 
- }
 
- func (item *dependingTestPipelineItem) Flag() string {
 
- 	return "depflag"
 
- }
 
- func (item *dependingTestPipelineItem) Description() string {
 
- 	return "another description"
 
- }
 
- func (item *dependingTestPipelineItem) Consume(deps map[string]interface{}) (map[string]interface{}, error) {
 
- 	_, exists := deps["test"]
 
- 	item.DependencySatisfied = exists
 
- 	if !item.TestNilConsumeReturn {
 
- 		return map[string]interface{}{"test2": item}, nil
 
- 	}
 
- 	return nil, nil
 
- }
 
- func (item *dependingTestPipelineItem) Fork(n int) []PipelineItem {
 
- 	clones := make([]PipelineItem, n)
 
- 	for i := range clones {
 
- 		clones[i] = item
 
- 	}
 
- 	return clones
 
- }
 
- func (item *dependingTestPipelineItem) Merge(branches []PipelineItem) {
 
- }
 
- func (item *dependingTestPipelineItem) Hibernate() error {
 
- 	item.Hibernated = true
 
- 	if item.RaiseHibernateError {
 
- 		return errors.New("error")
 
- 	}
 
- 	return nil
 
- }
 
- func (item *dependingTestPipelineItem) Boot() error {
 
- 	item.Booted = true
 
- 	if item.RaiseBootError {
 
- 		return errors.New("error")
 
- 	}
 
- 	return nil
 
- }
 
- func (item *dependingTestPipelineItem) Finalize() interface{} {
 
- 	return true
 
- }
 
- func (item *dependingTestPipelineItem) Serialize(result interface{}, binary bool, writer io.Writer) error {
 
- 	return nil
 
- }
 
- func TestPipelineFacts(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	pipeline.SetFact("fact", "value")
 
- 	assert.Equal(t, pipeline.GetFact("fact"), "value")
 
- }
 
- func TestPipelineFeatures(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	pipeline.SetFeature("feat")
 
- 	val, _ := pipeline.GetFeature("feat")
 
- 	assert.True(t, val)
 
- 	_, exists := pipeline.GetFeature("!")
 
- 	assert.False(t, exists)
 
- 	Registry.featureFlags.Set("777")
 
- 	defer func() {
 
- 		Registry.featureFlags = arrayFeatureFlags{Flags: []string{}, Choices: map[string]bool{}}
 
- 	}()
 
- 	pipeline.SetFeaturesFromFlags()
 
- 	_, exists = pipeline.GetFeature("777")
 
- 	assert.False(t, exists)
 
- 	assert.Panics(t, func() {
 
- 		pipeline.SetFeaturesFromFlags(
 
- 			&PipelineItemRegistry{}, &PipelineItemRegistry{})
 
- 	})
 
- }
 
- func TestPipelineErrors(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	pipeline.SetFact("fact", "value")
 
- 	assert.Equal(t, pipeline.GetFact("fact"), "value")
 
- 	item := &testPipelineItem{}
 
- 	pipeline.AddItem(item)
 
- 	item.ConfigureRaises = true
 
- 	err := pipeline.Initialize(map[string]interface{}{})
 
- 	assert.NotNil(t, err)
 
- 	assert.Contains(t, err.Error(), "configure")
 
- 	assert.Contains(t, err.Error(), "test1")
 
- 	item.ConfigureRaises = false
 
- 	item.InitializeRaises = true
 
- 	err = pipeline.Initialize(map[string]interface{}{})
 
- 	assert.NotNil(t, err)
 
- 	assert.Contains(t, err.Error(), "initialize")
 
- 	assert.Contains(t, err.Error(), "test2")
 
- 	item.InitializeRaises = false
 
- 	item.InitializePanics = true
 
- 	assert.Panics(t, func() { pipeline.Initialize(map[string]interface{}{}) })
 
- }
 
- func TestPipelineRun(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	item := &testPipelineItem{}
 
- 	pipeline.AddItem(item)
 
- 	assert.Nil(t, pipeline.Initialize(map[string]interface{}{}))
 
- 	assert.True(t, item.Initialized)
 
- 	commits := make([]*object.Commit, 1)
 
- 	commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
 
- 		"af9ddc0db70f09f3f27b4b98e415592a7485171c"))
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.Nil(t, err)
 
- 	assert.Equal(t, 2, len(result))
 
- 	assert.Equal(t, item, result[item].(*testPipelineItem))
 
- 	common := result[nil].(*CommonAnalysisResult)
 
- 	assert.Equal(t, common.BeginTime, int64(1481719198))
 
- 	assert.Equal(t, common.EndTime, int64(1481719198))
 
- 	assert.Equal(t, common.CommitsNumber, 1)
 
- 	assert.True(t, common.RunTime.Nanoseconds()/1e6 < 100)
 
- 	assert.Len(t, common.RunTimePerItem, 1)
 
- 	for key, val := range common.RunTimePerItem {
 
- 		assert.True(t, val >= 0, key)
 
- 	}
 
- 	assert.True(t, item.DepsConsumed)
 
- 	assert.True(t, item.Disposed)
 
- 	assert.True(t, item.CommitMatches)
 
- 	assert.True(t, item.IndexMatches)
 
- 	assert.Equal(t, 1, *item.MergeState)
 
- 	assert.True(t, item.Forked)
 
- 	assert.False(t, *item.Merged)
 
- 	pipeline.RemoveItem(item)
 
- 	result, err = pipeline.Run(commits)
 
- 	assert.Nil(t, err)
 
- 	assert.Equal(t, 1, len(result))
 
- }
 
- func TestPipelineRunBranches(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	item := &testPipelineItem{}
 
- 	pipeline.AddItem(item)
 
- 	pipeline.Initialize(map[string]interface{}{})
 
- 	assert.True(t, item.Initialized)
 
- 	hashes := []string{
 
- 		"6db8065cdb9bb0758f36a7e75fc72ab95f9e8145",
 
- 		"f30daba81ff2bf0b3ba02a1e1441e74f8a4f6fee",
 
- 		"8a03b5620b1caa72ec9cb847ea88332621e2950a",
 
- 		"dd9dd084d5851d7dc4399fc7dbf3d8292831ebc5",
 
- 		"f4ed0405b14f006c0744029d87ddb3245607587a",
 
- 	}
 
- 	commits := make([]*object.Commit, len(hashes))
 
- 	for i, h := range hashes {
 
- 		var err error
 
- 		commits[i], err = test.Repository.CommitObject(plumbing.NewHash(h))
 
- 		if err != nil {
 
- 			t.Fatal(err)
 
- 		}
 
- 	}
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.Nil(t, err)
 
- 	assert.True(t, item.Forked)
 
- 	assert.True(t, *item.Merged)
 
- 	assert.Equal(t, 2, len(result))
 
- 	assert.Equal(t, item, result[item].(*testPipelineItem))
 
- 	common := result[nil].(*CommonAnalysisResult)
 
- 	assert.Equal(t, common.CommitsNumber, 5)
 
- 	assert.Equal(t, *item.MergeState, 8)
 
- }
 
- func TestPipelineOnProgress(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	progressOk := 0
 
- 	onProgress := func(step int, total int, action string) {
 
- 		if step == 1 && total == 4 && action == "emerge" {
 
- 			progressOk++
 
- 		}
 
- 		if step == 2 && total == 4 && action == "af9ddc0" {
 
- 			progressOk++
 
- 		}
 
- 		if step == 3 && total == 4 && action == "finalize" {
 
- 			progressOk++
 
- 		}
 
- 		if step == 4 && total == 4 && action == "" {
 
- 			progressOk++
 
- 		}
 
- 	}
 
- 	pipeline.OnProgress = onProgress
 
- 	commits := make([]*object.Commit, 1)
 
- 	commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
 
- 		"af9ddc0db70f09f3f27b4b98e415592a7485171c"))
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.Nil(t, err)
 
- 	assert.Equal(t, 1, len(result))
 
- 	assert.Equal(t, 4, progressOk)
 
- }
 
- func TestPipelineCommitsFull(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	commits, err := pipeline.Commits(false)
 
- 	assert.Nil(t, err)
 
- 	assert.True(t, len(commits) >= 100)
 
- 	hashMap := map[plumbing.Hash]bool{}
 
- 	for _, c := range commits {
 
- 		hashMap[c.Hash] = true
 
- 	}
 
- 	assert.Equal(t, len(commits), len(hashMap))
 
- 	assert.Contains(t, hashMap, plumbing.NewHash(
 
- 		"cce947b98a050c6d356bc6ba95030254914027b1"))
 
- 	assert.Contains(t, hashMap, plumbing.NewHash(
 
- 		"a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
 
- }
 
- func TestPipelineCommitsFirstParent(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	commits, err := pipeline.Commits(true)
 
- 	assert.Nil(t, err)
 
- 	assert.True(t, len(commits) >= 100)
 
- 	hashMap := map[plumbing.Hash]bool{}
 
- 	for _, c := range commits {
 
- 		hashMap[c.Hash] = true
 
- 	}
 
- 	assert.Equal(t, len(commits), len(hashMap))
 
- 	assert.Contains(t, hashMap, plumbing.NewHash(
 
- 		"cce947b98a050c6d356bc6ba95030254914027b1"))
 
- 	assert.NotContains(t, hashMap, plumbing.NewHash(
 
- 		"a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
 
- }
 
- func TestLoadCommitsFromFile(t *testing.T) {
 
- 	tmp, err := ioutil.TempFile("", "hercules-test-")
 
- 	assert.Nil(t, err)
 
- 	tmp.WriteString("cce947b98a050c6d356bc6ba95030254914027b1\n6db8065cdb9bb0758f36a7e75fc72ab95f9e8145")
 
- 	tmp.Close()
 
- 	defer os.Remove(tmp.Name())
 
- 	commits, err := LoadCommitsFromFile(tmp.Name(), test.Repository)
 
- 	assert.Nil(t, err)
 
- 	assert.Equal(t, len(commits), 2)
 
- 	assert.Equal(t, commits[0].Hash, plumbing.NewHash(
 
- 		"cce947b98a050c6d356bc6ba95030254914027b1"))
 
- 	assert.Equal(t, commits[1].Hash, plumbing.NewHash(
 
- 		"6db8065cdb9bb0758f36a7e75fc72ab95f9e8145"))
 
- 	commits, err = LoadCommitsFromFile("/WAT?xxx!", test.Repository)
 
- 	assert.Nil(t, commits)
 
- 	assert.NotNil(t, err)
 
- 	tmp, err = ioutil.TempFile("", "hercules-test-")
 
- 	assert.Nil(t, err)
 
- 	tmp.WriteString("WAT")
 
- 	tmp.Close()
 
- 	defer os.Remove(tmp.Name())
 
- 	commits, err = LoadCommitsFromFile(tmp.Name(), test.Repository)
 
- 	assert.Nil(t, commits)
 
- 	assert.NotNil(t, err)
 
- 	tmp, err = ioutil.TempFile("", "hercules-test-")
 
- 	assert.Nil(t, err)
 
- 	tmp.WriteString("ffffffffffffffffffffffffffffffffffffffff")
 
- 	tmp.Close()
 
- 	defer os.Remove(tmp.Name())
 
- 	commits, err = LoadCommitsFromFile(tmp.Name(), test.Repository)
 
- 	assert.Nil(t, commits)
 
- 	assert.NotNil(t, err)
 
- }
 
- func TestPipelineDeps(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	item1 := &dependingTestPipelineItem{}
 
- 	item2 := &testPipelineItem{}
 
- 	pipeline.AddItem(item1)
 
- 	pipeline.AddItem(item2)
 
- 	assert.Equal(t, pipeline.Len(), 2)
 
- 	pipeline.Initialize(map[string]interface{}{})
 
- 	commits := make([]*object.Commit, 1)
 
- 	commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
 
- 		"af9ddc0db70f09f3f27b4b98e415592a7485171c"))
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.NoError(t, err)
 
- 	assert.True(t, result[item1].(bool))
 
- 	assert.Equal(t, result[item2], item2)
 
- 	item1.TestNilConsumeReturn = true
 
- 	_, err = pipeline.Run(commits)
 
- 	assert.Error(t, err)
 
- }
 
- func TestPipelineDeployFeatures(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	pipeline.DeployItem(&testPipelineItem{})
 
- 	f, _ := pipeline.GetFeature("power")
 
- 	assert.True(t, f)
 
- }
 
- func TestPipelineError(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	item := &testPipelineItem{}
 
- 	item.TestError = true
 
- 	pipeline.AddItem(item)
 
- 	pipeline.Initialize(map[string]interface{}{})
 
- 	commits := make([]*object.Commit, 1)
 
- 	commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
 
- 		"af9ddc0db70f09f3f27b4b98e415592a7485171c"))
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.Nil(t, result)
 
- 	assert.NotNil(t, err)
 
- }
 
- func TestPipelineDryRun(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	item := &testPipelineItem{}
 
- 	item.TestError = true
 
- 	pipeline.AddItem(item)
 
- 	pipeline.DryRun = true
 
- 	pipeline.Initialize(map[string]interface{}{})
 
- 	assert.True(t, pipeline.DryRun)
 
- 	pipeline.DryRun = false
 
- 	pipeline.Initialize(map[string]interface{}{ConfigPipelineDryRun: true})
 
- 	assert.True(t, pipeline.DryRun)
 
- 	commits := make([]*object.Commit, 1)
 
- 	commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
 
- 		"af9ddc0db70f09f3f27b4b98e415592a7485171c"))
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.NotNil(t, result)
 
- 	assert.Len(t, result, 1)
 
- 	assert.Contains(t, result, nil)
 
- 	assert.Nil(t, err)
 
- }
 
- func TestPipelineDryRunFalse(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	item := &testPipelineItem{}
 
- 	pipeline.AddItem(item)
 
- 	pipeline.Initialize(map[string]interface{}{ConfigPipelineDryRun: false})
 
- 	commits := make([]*object.Commit, 1)
 
- 	commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
 
- 		"af9ddc0db70f09f3f27b4b98e415592a7485171c"))
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.NotNil(t, result)
 
- 	assert.Len(t, result, 2)
 
- 	assert.Contains(t, result, nil)
 
- 	assert.Contains(t, result, item)
 
- 	assert.Nil(t, err)
 
- 	assert.True(t, item.DepsConsumed)
 
- 	assert.True(t, item.CommitMatches)
 
- 	assert.True(t, item.IndexMatches)
 
- 	assert.Equal(t, 1, *item.MergeState)
 
- 	assert.True(t, item.Forked)
 
- 	assert.False(t, *item.Merged)
 
- }
 
- func TestPipelineDumpPlanConfigure(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	item := &testPipelineItem{}
 
- 	pipeline.AddItem(item)
 
- 	pipeline.DumpPlan = true
 
- 	pipeline.DryRun = true
 
- 	pipeline.Initialize(map[string]interface{}{})
 
- 	assert.True(t, pipeline.DumpPlan)
 
- 	pipeline.DumpPlan = false
 
- 	pipeline.Initialize(map[string]interface{}{ConfigPipelineDumpPlan: true})
 
- 	assert.True(t, pipeline.DumpPlan)
 
- 	stream := &bytes.Buffer{}
 
- 	backupPlanPrintFunc := planPrintFunc
 
- 	planPrintFunc = func(args ...interface{}) {
 
- 		fmt.Fprintln(stream, args...)
 
- 	}
 
- 	defer func() {
 
- 		planPrintFunc = backupPlanPrintFunc
 
- 	}()
 
- 	commits := make([]*object.Commit, 1)
 
- 	commits[0], _ = test.Repository.CommitObject(plumbing.NewHash(
 
- 		"af9ddc0db70f09f3f27b4b98e415592a7485171c"))
 
- 	result, err := pipeline.Run(commits)
 
- 	assert.NotNil(t, result)
 
- 	assert.Len(t, result, 1)
 
- 	assert.Contains(t, result, nil)
 
- 	assert.Nil(t, err)
 
- 	assert.Equal(t, `E [1]
 
- C 1 af9ddc0db70f09f3f27b4b98e415592a7485171c
 
- `, stream.String())
 
- }
 
- func TestCommonAnalysisResultCopy(t *testing.T) {
 
- 	c1 := CommonAnalysisResult{
 
- 		BeginTime: 1513620635, EndTime: 1513720635, CommitsNumber: 1, RunTime: 100,
 
- 		RunTimePerItem: map[string]float64{"one": 1, "two": 2}}
 
- 	c2 := c1.Copy()
 
- 	assert.Equal(t, c1, c2)
 
- 	c2.RunTimePerItem["one"] = 100500
 
- 	assert.Equal(t, c1.RunTimePerItem["one"], float64(1))
 
- }
 
- func TestCommonAnalysisResultMerge(t *testing.T) {
 
- 	c1 := CommonAnalysisResult{
 
- 		BeginTime: 1513620635, EndTime: 1513720635, CommitsNumber: 1, RunTime: 100,
 
- 		RunTimePerItem: map[string]float64{"one": 1, "two": 2}}
 
- 	assert.Equal(t, c1.BeginTimeAsTime().Unix(), int64(1513620635))
 
- 	assert.Equal(t, c1.EndTimeAsTime().Unix(), int64(1513720635))
 
- 	c2 := CommonAnalysisResult{
 
- 		BeginTime: 1513620535, EndTime: 1513730635, CommitsNumber: 2, RunTime: 200,
 
- 		RunTimePerItem: map[string]float64{"two": 4, "three": 8}}
 
- 	c1.Merge(&c2)
 
- 	assert.Equal(t, c1.BeginTime, int64(1513620535))
 
- 	assert.Equal(t, c1.EndTime, int64(1513730635))
 
- 	assert.Equal(t, c1.CommitsNumber, 3)
 
- 	assert.Equal(t, c1.RunTime.Nanoseconds(), int64(300))
 
- 	assert.Equal(t, c1.RunTimePerItem, map[string]float64{"one": 1, "two": 6, "three": 8})
 
- }
 
- func TestCommonAnalysisResultMetadata(t *testing.T) {
 
- 	c1 := &CommonAnalysisResult{
 
- 		BeginTime: 1513620635, EndTime: 1513720635, CommitsNumber: 1, RunTime: 100 * 1e6,
 
- 		RunTimePerItem: map[string]float64{"one": 1, "two": 2}}
 
- 	meta := &pb.Metadata{}
 
- 	c1 = MetadataToCommonAnalysisResult(c1.FillMetadata(meta))
 
- 	assert.Equal(t, c1.BeginTimeAsTime().Unix(), int64(1513620635))
 
- 	assert.Equal(t, c1.EndTimeAsTime().Unix(), int64(1513720635))
 
- 	assert.Equal(t, c1.CommitsNumber, 1)
 
- 	assert.Equal(t, c1.RunTime.Nanoseconds(), int64(100*1e6))
 
- 	assert.Equal(t, c1.RunTimePerItem, map[string]float64{"one": 1, "two": 2})
 
- }
 
- func TestConfigurationOptionTypeString(t *testing.T) {
 
- 	opt := ConfigurationOptionType(0)
 
- 	assert.Equal(t, opt.String(), "")
 
- 	opt = ConfigurationOptionType(1)
 
- 	assert.Equal(t, opt.String(), "int")
 
- 	opt = ConfigurationOptionType(2)
 
- 	assert.Equal(t, opt.String(), "string")
 
- 	opt = ConfigurationOptionType(3)
 
- 	assert.Equal(t, opt.String(), "float")
 
- 	opt = ConfigurationOptionType(4)
 
- 	assert.Equal(t, opt.String(), "string")
 
- 	opt = ConfigurationOptionType(5)
 
- 	assert.Equal(t, opt.String(), "path")
 
- 	opt = ConfigurationOptionType(6)
 
- 	assert.Panics(t, func() { _ = opt.String() })
 
- }
 
- func TestConfigurationOptionFormatDefault(t *testing.T) {
 
- 	opt := ConfigurationOption{Type: StringConfigurationOption, Default: "ololo"}
 
- 	assert.Equal(t, opt.FormatDefault(), "\"ololo\"")
 
- 	opt = ConfigurationOption{Type: IntConfigurationOption, Default: 7}
 
- 	assert.Equal(t, opt.FormatDefault(), "7")
 
- 	opt = ConfigurationOption{Type: BoolConfigurationOption, Default: false}
 
- 	assert.Equal(t, opt.FormatDefault(), "false")
 
- 	opt = ConfigurationOption{Type: FloatConfigurationOption, Default: 0.5}
 
- 	assert.Equal(t, opt.FormatDefault(), "0.5")
 
- }
 
- func TestPrepareRunPlanTiny(t *testing.T) {
 
- 	rootCommit, err := test.Repository.CommitObject(plumbing.NewHash(
 
- 		"cce947b98a050c6d356bc6ba95030254914027b1"))
 
- 	if err != nil {
 
- 		t.Fatal(err)
 
- 	}
 
- 	plan := prepareRunPlan([]*object.Commit{rootCommit}, 0, true)
 
- 	assert.Len(t, plan, 2)
 
- 	assert.Equal(t, runActionEmerge, plan[0].Action)
 
- 	assert.Equal(t, rootBranchIndex, plan[0].Items[0])
 
- 	assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[0].Commit.Hash.String())
 
- 	assert.Equal(t, runActionCommit, plan[1].Action)
 
- 	assert.Equal(t, rootBranchIndex, plan[1].Items[0])
 
- 	assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[1].Commit.Hash.String())
 
- }
 
- func TestPrepareRunPlanSmall(t *testing.T) {
 
- 	cit, err := test.Repository.Log(&git.LogOptions{From: plumbing.ZeroHash})
 
- 	if err != nil {
 
- 		panic(err)
 
- 	}
 
- 	defer cit.Close()
 
- 	var commits []*object.Commit
 
- 	timeCutoff := time.Date(2016, 12, 15, 0, 0, 0, 0, time.FixedZone("CET", 7200))
 
- 	cit.ForEach(func(commit *object.Commit) error {
 
- 		reliableTime := time.Date(commit.Author.When.Year(), commit.Author.When.Month(),
 
- 			commit.Author.When.Day(), commit.Author.When.Hour(), commit.Author.When.Minute(),
 
- 			commit.Author.When.Second(), 0, time.FixedZone("CET", 7200))
 
- 		if reliableTime.Before(timeCutoff) {
 
- 			commits = append(commits, commit)
 
- 		}
 
- 		return nil
 
- 	})
 
- 	plan := prepareRunPlan(commits, 0, false)
 
- 	/*for _, p := range plan {
 
- 		if p.Commit != nil {
 
- 			fmt.Println(p.Action, p.Commit.Hash.String(), p.Items)
 
- 		} else {
 
- 			fmt.Println(p.Action, strings.Repeat(" ", 40), p.Items)
 
- 		}
 
- 	}*/
 
- 	// fork, merge and one artificial commit per branch
 
- 	assert.Len(t, plan, len(commits)+1)
 
- 	assert.Equal(t, runActionEmerge, plan[0].Action)
 
- 	assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[0].Commit.Hash.String())
 
- 	assert.Equal(t, rootBranchIndex, plan[0].Items[0])
 
- 	assert.Equal(t, runActionCommit, plan[1].Action)
 
- 	assert.Equal(t, rootBranchIndex, plan[1].Items[0])
 
- 	assert.Equal(t, "cce947b98a050c6d356bc6ba95030254914027b1", plan[1].Commit.Hash.String())
 
- 	assert.Equal(t, runActionCommit, plan[2].Action)
 
- 	assert.Equal(t, rootBranchIndex, plan[2].Items[0])
 
- 	assert.Equal(t, "a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3", plan[2].Commit.Hash.String())
 
- 	assert.Equal(t, runActionCommit, plan[10].Action)
 
- 	assert.Equal(t, rootBranchIndex, plan[10].Items[0])
 
- 	assert.Equal(t, "a28e9064c70618dc9d68e1401b889975e0680d11", plan[10].Commit.Hash.String())
 
- }
 
- func TestMergeDag(t *testing.T) {
 
- 	cit, err := test.Repository.Log(&git.LogOptions{From: plumbing.ZeroHash})
 
- 	if err != nil {
 
- 		panic(err)
 
- 	}
 
- 	defer cit.Close()
 
- 	var commits []*object.Commit
 
- 	timeCutoff := time.Date(2017, 8, 12, 0, 0, 0, 0, time.FixedZone("CET", 7200))
 
- 	cit.ForEach(func(commit *object.Commit) error {
 
- 		reliableTime := time.Date(commit.Author.When.Year(), commit.Author.When.Month(),
 
- 			commit.Author.When.Day(), commit.Author.When.Hour(), commit.Author.When.Minute(),
 
- 			commit.Author.When.Second(), 0, time.FixedZone("CET", 7200))
 
- 		if reliableTime.Before(timeCutoff) {
 
- 			commits = append(commits, commit)
 
- 		}
 
- 		return nil
 
- 	})
 
- 	hashes, dag := buildDag(commits)
 
- 	leaveRootComponent(hashes, dag)
 
- 	mergedDag, _ := mergeDag(hashes, dag)
 
- 	for key, vals := range mergedDag {
 
- 		if key != plumbing.NewHash("a28e9064c70618dc9d68e1401b889975e0680d11") &&
 
- 			key != plumbing.NewHash("db325a212d0bc99b470e000641d814745024bbd5") {
 
- 			assert.Len(t, vals, len(dag[key]), key.String())
 
- 		} else {
 
- 			mvals := map[string]bool{}
 
- 			for _, val := range vals {
 
- 				mvals[val.Hash.String()] = true
 
- 			}
 
- 			if key == plumbing.NewHash("a28e9064c70618dc9d68e1401b889975e0680d11") {
 
- 				assert.Contains(t, mvals, "db325a212d0bc99b470e000641d814745024bbd5")
 
- 				assert.Contains(t, mvals, "be9b61e09b08b98e64ed461a4004c9e2412f78ee")
 
- 			}
 
- 			if key == plumbing.NewHash("db325a212d0bc99b470e000641d814745024bbd5") {
 
- 				assert.Contains(t, mvals, "f30daba81ff2bf0b3ba02a1e1441e74f8a4f6fee")
 
- 				assert.Contains(t, mvals, "8a03b5620b1caa72ec9cb847ea88332621e2950a")
 
- 			}
 
- 		}
 
- 	}
 
- 	assert.Len(t, mergedDag, 8)
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("cce947b98a050c6d356bc6ba95030254914027b1"))
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("a3ee37f91f0d705ec9c41ae88426f0ae44b2fbc3"))
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("a28e9064c70618dc9d68e1401b889975e0680d11"))
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("be9b61e09b08b98e64ed461a4004c9e2412f78ee"))
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("db325a212d0bc99b470e000641d814745024bbd5"))
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("f30daba81ff2bf0b3ba02a1e1441e74f8a4f6fee"))
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("8a03b5620b1caa72ec9cb847ea88332621e2950a"))
 
- 	assert.Contains(t, mergedDag, plumbing.NewHash("dd9dd084d5851d7dc4399fc7dbf3d8292831ebc5"))
 
- 	queue := []plumbing.Hash{plumbing.NewHash("cce947b98a050c6d356bc6ba95030254914027b1")}
 
- 	visited := map[plumbing.Hash]bool{}
 
- 	for len(queue) > 0 {
 
- 		head := queue[len(queue)-1]
 
- 		queue = queue[:len(queue)-1]
 
- 		if visited[head] {
 
- 			continue
 
- 		}
 
- 		visited[head] = true
 
- 		for _, child := range mergedDag[head] {
 
- 			queue = append(queue, child.Hash)
 
- 		}
 
- 	}
 
- 	assert.Len(t, visited, 8)
 
- }
 
- func TestPrepareRunPlanBig(t *testing.T) {
 
- 	cases := [][7]int{
 
- 		{2017, 8, 9, 0, 0, 0, 0},
 
- 		{2017, 8, 10, 0, 0, 0, 0},
 
- 		{2017, 8, 24, 1, 1, 1, 1},
 
- 		{2017, 9, 19, 1 - 2, 1, 1, 1},
 
- 		{2017, 9, 23, 1 - 2, 1, 1, 1},
 
- 		{2017, 12, 8, 1, 1, 1, 1},
 
- 		{2017, 12, 9, 1, 1, 1, 1},
 
- 		{2017, 12, 10, 1, 1, 1, 1},
 
- 		{2017, 12, 11, 2, 2, 2, 2},
 
- 		{2017, 12, 19, 3, 3, 3, 3},
 
- 		{2017, 12, 27, 3, 3, 3, 3},
 
- 		{2018, 1, 10, 3, 3, 3, 3},
 
- 		{2018, 1, 16, 3, 3, 3, 3},
 
- 		{2018, 1, 18, 4, 5, 4, 4},
 
- 		{2018, 1, 23, 5, 5, 5, 5},
 
- 		{2018, 3, 12, 6, 6, 6, 6},
 
- 		{2018, 5, 13, 6, 6, 6, 6},
 
- 		{2018, 5, 16, 7, 7, 7, 7},
 
- 	}
 
- 	for _, testCase := range cases {
 
- 		func() {
 
- 			cit, err := test.Repository.Log(&git.LogOptions{From: plumbing.ZeroHash})
 
- 			if err != nil {
 
- 				panic(err)
 
- 			}
 
- 			defer cit.Close()
 
- 			var commits []*object.Commit
 
- 			timeCutoff := time.Date(
 
- 				testCase[0], time.Month(testCase[1]), testCase[2], 0, 0, 0, 0, time.FixedZone("CET", 7200))
 
- 			cit.ForEach(func(commit *object.Commit) error {
 
- 				reliableTime := time.Date(commit.Author.When.Year(), commit.Author.When.Month(),
 
- 					commit.Author.When.Day(), commit.Author.When.Hour(), commit.Author.When.Minute(),
 
- 					commit.Author.When.Second(), 0, time.FixedZone("CET", 7200))
 
- 				if reliableTime.Before(timeCutoff) {
 
- 					commits = append(commits, commit)
 
- 				}
 
- 				return nil
 
- 			})
 
- 			plan := prepareRunPlan(commits, 0, false)
 
- 			/*for _, p := range plan {
 
- 				if p.Commit != nil {
 
- 					fmt.Println(p.Action, p.Commit.Hash.String(), p.Items)
 
- 				} else {
 
- 					fmt.Println(p.Action, strings.Repeat(" ", 40), p.Items)
 
- 				}
 
- 			}*/
 
- 			numCommits := 0
 
- 			numForks := 0
 
- 			numMerges := 0
 
- 			numDeletes := 0
 
- 			numEmerges := 0
 
- 			processed := map[plumbing.Hash]map[int]int{}
 
- 			for _, p := range plan {
 
- 				switch p.Action {
 
- 				case runActionCommit:
 
- 					branches := processed[p.Commit.Hash]
 
- 					if branches == nil {
 
- 						branches = map[int]int{}
 
- 						processed[p.Commit.Hash] = branches
 
- 					}
 
- 					branches[p.Items[0]]++
 
- 					for _, parent := range p.Commit.ParentHashes {
 
- 						assert.Contains(t, processed, parent)
 
- 					}
 
- 					numCommits++
 
- 				case runActionFork:
 
- 					numForks++
 
- 				case runActionMerge:
 
- 					counts := map[int]int{}
 
- 					for _, i := range p.Items {
 
- 						counts[i]++
 
- 					}
 
- 					for x, v := range counts {
 
- 						assert.Equal(t, 1, v, x)
 
- 					}
 
- 					numMerges++
 
- 				case runActionDelete:
 
- 					numDeletes++
 
- 				case runActionEmerge:
 
- 					numEmerges++
 
- 				}
 
- 			}
 
- 			for c, branches := range processed {
 
- 				for b, v := range branches {
 
- 					assert.Equal(t, 1, v, fmt.Sprint(c.String(), b))
 
- 				}
 
- 			}
 
- 			assert.Equal(t, numCommits, len(commits)+testCase[3], fmt.Sprintf("commits %v", testCase))
 
- 			assert.Equal(t, numForks, testCase[4], fmt.Sprintf("forks %v", testCase))
 
- 			assert.Equal(t, numMerges, testCase[5], fmt.Sprintf("merges %v", testCase))
 
- 			assert.Equal(t, numDeletes, testCase[6], fmt.Sprintf("deletes %v", testCase))
 
- 			assert.Equal(t, numEmerges, 1, fmt.Sprintf("emerges %v", testCase))
 
- 		}()
 
- 	}
 
- }
 
- func TestPipelineRunHibernation(t *testing.T) {
 
- 	pipeline := NewPipeline(test.Repository)
 
- 	pipeline.HibernationDistance = 2
 
- 	pipeline.AddItem(&testPipelineItem{})
 
- 	item := &dependingTestPipelineItem{}
 
- 	pipeline.AddItem(item)
 
- 	pipeline.Initialize(map[string]interface{}{})
 
- 	hashes := []string{
 
- 		"0183e08978007c746468fca9f68e6e2fbf32100c",
 
- 		"b467a682f680a4dcfd74869480a52f8be3a4fdf0",
 
- 		"31c9f752f9ce103e85523442fa3f05b1ff4ea546",
 
- 		"6530890fcd02fb5e6e85ce2951fdd5c555f2c714",
 
- 		"feb2d230777cbb492ecbc27dea380dc1e7b8f437",
 
- 		"9b30d2abc043ab59aa7ec7b50970c65c90b98853",
 
- 	}
 
- 	commits := make([]*object.Commit, len(hashes))
 
- 	for i, h := range hashes {
 
- 		var err error
 
- 		commits[i], err = test.Repository.CommitObject(plumbing.NewHash(h))
 
- 		if err != nil {
 
- 			t.Fatal(err)
 
- 		}
 
- 	}
 
- 	pipeline.PrintActions = true
 
- 	_, err := pipeline.Run(commits)
 
- 	assert.NoError(t, err)
 
- 	assert.True(t, item.Hibernated)
 
- 	assert.True(t, item.Booted)
 
- 	item.RaiseHibernateError = true
 
- 	_, err = pipeline.Run(commits)
 
- 	assert.Error(t, err)
 
- 	item.RaiseHibernateError = false
 
- 	pipeline.Run(commits)
 
- 	item.RaiseBootError = true
 
- 	_, err = pipeline.Run(commits)
 
- 	assert.Error(t, err)
 
- }
 
 
  |