hercules/internal/core/forks_test.go

package core

import (
	"testing"

	"github.com/stretchr/testify/assert"
	"gopkg.in/src-d/go-git.v4"
)

type testForkPipelineItem struct {
	NoopMerger
	Mutable   map[int]bool
	Immutable string
}

func (item *testForkPipelineItem) Name() string {
	return "Test"
}

func (item *testForkPipelineItem) Provides() []string {
	arr := [...]string{"test"}
	return arr[:]
}

func (item *testForkPipelineItem) Requires() []string {
	return []string{}
}

func (item *testForkPipelineItem) Configure(facts map[string]interface{}) error {
	return nil
}

func (item *testForkPipelineItem) ListConfigurationOptions() []ConfigurationOption {
	return nil
}

func (item *testForkPipelineItem) Flag() string {
	return "mytest"
}

func (item *testForkPipelineItem) Features() []string {
	return nil
}

func (item *testForkPipelineItem) Initialize(repository *git.Repository) error {
	item.Mutable = map[int]bool{}
	return nil
}

func (item *testForkPipelineItem) Consume(deps map[string]interface{}) (map[string]interface{}, error) {
	return map[string]interface{}{"test": "foo"}, nil
}

func (item *testForkPipelineItem) Fork(n int) []PipelineItem {
	return ForkCopyPipelineItem(item, n)
}

func TestForkCopyPipelineItem(t *testing.T) {
	origin := &testForkPipelineItem{}
	origin.Initialize(nil)
	origin.Mutable[2] = true
	origin.Immutable = "before"
	clone := origin.Fork(1)[0].(*testForkPipelineItem)
	origin.Immutable = "after"
	origin.Mutable[1] = true
	assert.True(t, clone.Mutable[1])
	assert.True(t, clone.Mutable[2])
	assert.Equal(t, "before", clone.Immutable)
}

func TestInsertHibernateBoot(t *testing.T) {
	plan := []runAction{
		{runActionEmerge, nil, []int{1, 2}},
		{runActionEmerge, nil, []int{3}},
		{runActionCommit, nil, []int{3}},
		{runActionCommit, nil, []int{3}},
		{runActionCommit, nil, []int{1}},
		{runActionFork, nil, []int{2, 4}},
		{runActionCommit, nil, []int{3}},
		{runActionCommit, nil, []int{3}},
		{runActionDelete, nil, []int{1}},
		{runActionMerge, nil, []int{2, 4}},
	}
	plan = insertHibernateBoot(plan, 2)
	assert.Equal(t, []runAction{
		{runActionEmerge, nil, []int{1, 2}},
		{runActionHibernate, nil, []int{1, 2}},
		{runActionEmerge, nil, []int{3}},
		{runActionCommit, nil, []int{3}},
		{runActionCommit, nil, []int{3}},
		{runActionBoot, nil, []int{1}},
		{runActionCommit, nil, []int{1}},
		{runActionBoot, nil, []int{2}},
		{runActionFork, nil, []int{2, 4}},
		{runActionHibernate, nil, []int{2, 4}},
		{runActionCommit, nil, []int{3}},
		{runActionCommit, nil, []int{3}},
		{runActionDelete, nil, []int{1}},
		{runActionBoot, nil, []int{2, 4}},
		{runActionMerge, nil, []int{2, 4}},
	}, plan)
}