atlantis/server/pull_closed_executor_test.go

package server_test

import (
	"errors"
	"reflect"
	"testing"

	ghmocks "github.com/hootsuite/atlantis/server/github/mocks"
	lockmocks "github.com/hootsuite/atlantis/server/locking/mocks"
	"github.com/hootsuite/atlantis/server/models"
	"github.com/hootsuite/atlantis/server/models/fixtures"
	"github.com/hootsuite/atlantis/server"
	"github.com/hootsuite/atlantis/server/mocks"
	. "github.com/hootsuite/atlantis/testing_util"
	. "github.com/petergtz/pegomock"
)

func TestCleanUpPullWorkspaceErr(t *testing.T) {
	t.Log("when workspace.Delete returns an error, we return it")
	RegisterMockTestingT(t)
	w := mocks.NewMockWorkspace()
	pce := server.PullClosedExecutor{
		Workspace: w,
	}
	err := errors.New("err")
	When(w.Delete(fixtures.Repo, fixtures.Pull)).ThenReturn(err)
	actualErr := pce.CleanUpPull(fixtures.Repo, fixtures.Pull)
	Equals(t, "cleaning workspace: err", actualErr.Error())
}

func TestCleanUpPullUnlockErr(t *testing.T) {
	t.Log("when locker.UnlockByPull returns an error, we return it")
	RegisterMockTestingT(t)
	w := mocks.NewMockWorkspace()
	l := lockmocks.NewMockLocker()
	pce := server.PullClosedExecutor{
		Locker:    l,
		Workspace: w,
	}
	err := errors.New("err")
	When(l.UnlockByPull(fixtures.Repo.FullName, fixtures.Pull.Num)).ThenReturn(nil, err)
	actualErr := pce.CleanUpPull(fixtures.Repo, fixtures.Pull)
	Equals(t, "cleaning up locks: err", actualErr.Error())
}

func TestCleanUpPullNoLocks(t *testing.T) {
	t.Log("when there are no locks to clean up, we don't comment")
	RegisterMockTestingT(t)
	w := mocks.NewMockWorkspace()
	l := lockmocks.NewMockLocker()
	gh := ghmocks.NewMockClient()
	pce := server.PullClosedExecutor{
		Locker:    l,
		Github:    gh,
		Workspace: w,
	}
	When(l.UnlockByPull(fixtures.Repo.FullName, fixtures.Pull.Num)).ThenReturn(nil, nil)
	err := pce.CleanUpPull(fixtures.Repo, fixtures.Pull)
	Ok(t, err)
	gh.VerifyWasCalled(Never()).CreateComment(AnyRepo(), AnyPullRequest(), AnyString())
}

func TestCleanUpPullComments(t *testing.T) {
	t.Log("should comment correctly")
	RegisterMockTestingT(t)
	cases := []struct {
		Description string
		Locks       []models.ProjectLock
		Exp         string
	}{
		{
			"single lock, empty path",
			[]models.ProjectLock{
				{
					Project: models.NewProject("owner/repo", ""),
					Env:     "default",
				},
			},
			"- path: `owner/repo/.` environment: `default`",
		},
		{
			"single lock, non-empty path",
			[]models.ProjectLock{
				{
					Project: models.NewProject("owner/repo", "path"),
					Env:     "default",
				},
			},
			"- path: `owner/repo/path` environment: `default`",
		},
		{
			"single path, multiple environments",
			[]models.ProjectLock{
				{
					Project: models.NewProject("owner/repo", "path"),
					Env:     "env1",
				},
				{
					Project: models.NewProject("owner/repo", "path"),
					Env:     "env2",
				},
			},
			"- path: `owner/repo/path` environments: `env1`, `env2`",
		},
		{
			"multiple paths, multiple environments",
			[]models.ProjectLock{
				{
					Project: models.NewProject("owner/repo", "path"),
					Env:     "env1",
				},
				{
					Project: models.NewProject("owner/repo", "path"),
					Env:     "env2",
				},
				{
					Project: models.NewProject("owner/repo", "path2"),
					Env:     "env1",
				},
				{
					Project: models.NewProject("owner/repo", "path2"),
					Env:     "env2",
				},
			},
			"- path: `owner/repo/path` environments: `env1`, `env2`\n- path: `owner/repo/path2` environments: `env1`, `env2`",
		},
	}
	for _, c := range cases {
		w := mocks.NewMockWorkspace()
		gh := ghmocks.NewMockClient()
		l := lockmocks.NewMockLocker()
		pce := server.PullClosedExecutor{
			Locker:    l,
			Github:    gh,
			Workspace: w,
		}
		t.Log("testing: " + c.Description)
		When(l.UnlockByPull(fixtures.Repo.FullName, fixtures.Pull.Num)).ThenReturn(c.Locks, nil)
		err := pce.CleanUpPull(fixtures.Repo, fixtures.Pull)
		Ok(t, err)
		_, _, comment := gh.VerifyWasCalledOnce().CreateComment(AnyRepo(), AnyPullRequest(), AnyString()).GetCapturedArguments()

		expected := "Locks and plans deleted for the projects and environments modified in this pull request:\n\n" + c.Exp
		Equals(t, expected, comment)
	}
}

func AnyRepo() models.Repo {
	RegisterMatcher(NewAnyMatcher(reflect.TypeOf(models.Repo{})))
	return models.Repo{}
}

func AnyPullRequest() models.PullRequest {
	RegisterMatcher(NewAnyMatcher(reflect.TypeOf(models.PullRequest{})))
	return models.PullRequest{}
}