Adding slack integration for posting to channel on applies. Fixes #179

cmd/server.go

@@ -84,7 +84,7 @@ var stringFlags = []stringFlag{
 		env:         "ATLANTIS_GITLAB_TOKEN",
 	},
 	{
-		name:        GitlabWebHookSecret,
+		name: GitlabWebHookSecret,
 		description: "Optional secret used to validate GitLab webhooks." +
 			" If not specified, Atlantis won't be able to validate that the incoming webhook call came from GitLab. " +
 			"Can also be specified via the ATLANTIS_GITLAB_WEBHOOK_SECRET environment variable.",

server/events/apply_executor.go

@@ -12,6 +12,7 @@ import (
 	"github.com/hootsuite/atlantis/server/events/run"
 	"github.com/hootsuite/atlantis/server/events/terraform"
 	"github.com/hootsuite/atlantis/server/events/vcs"
+	"github.com/hootsuite/atlantis/server/events/webhooks"
 )
 
 type ApplyExecutor struct {
@@ -21,6 +22,7 @@ type ApplyExecutor struct {
 	Run               *run.Run
 	Workspace         Workspace
 	ProjectPreExecute *ProjectPreExecute
+	Webhooks          webhooks.WebhooksSender
 }
 
 func (a *ApplyExecutor) Execute(ctx *CommandContext) CommandResponse {
@@ -92,6 +94,15 @@ func (a *ApplyExecutor) apply(ctx *CommandContext, repoDir string, plan models.P
 	env := ctx.Command.Environment
 	tfApplyCmd := append(append(append([]string{"apply", "-no-color"}, applyExtraArgs...), ctx.Command.Flags...), plan.LocalPath)
 	output, err := a.Terraform.RunCommandWithVersion(ctx.Log, absolutePath, tfApplyCmd, terraformVersion, env)
+
+	a.Webhooks.Send(ctx.Log, webhooks.ApplyResult{
+		Workspace: env,
+		User:      ctx.User,
+		Repo:      ctx.BaseRepo,
+		Pull:      ctx.Pull,
+		Success:   err == nil,
+	})
+
 	if err != nil {
 		return ProjectResult{Error: fmt.Errorf("%s\n%s", err.Error(), output)}
 	}

server/events/command_handler_test.go

@@ -57,7 +57,7 @@ func setup(t *testing.T) {
 		MarkdownRenderer:         &events.MarkdownRenderer{},
 		GithubPullGetter:         githubGetter,
 		GitlabMergeRequestGetter: gitlabGetter,
-		Logger:                   logger,
+		Logger: logger,
 	}
 }
 

server/events/pull_closed_executor_test.go

@@ -3,17 +3,18 @@ package events_test
 import (
 	"reflect"
 
-	"github.com/hootsuite/atlantis/server/events/models"
-	. "github.com/hootsuite/atlantis/testing"
-	. "github.com/petergtz/pegomock"
+	"errors"
 	"testing"
-	"github.com/hootsuite/atlantis/server/events/mocks"
-	lockmocks "github.com/hootsuite/atlantis/server/events/locking/mocks"
-	vcsmocks "github.com/hootsuite/atlantis/server/events/vcs/mocks"
+
 	"github.com/hootsuite/atlantis/server/events"
-	"errors"
-	"github.com/hootsuite/atlantis/server/events/vcs"
+	lockmocks "github.com/hootsuite/atlantis/server/events/locking/mocks"
+	"github.com/hootsuite/atlantis/server/events/mocks"
+	"github.com/hootsuite/atlantis/server/events/models"
 	"github.com/hootsuite/atlantis/server/events/models/fixtures"
+	"github.com/hootsuite/atlantis/server/events/vcs"
+	vcsmocks "github.com/hootsuite/atlantis/server/events/vcs/mocks"
+	. "github.com/hootsuite/atlantis/testing"
+	. "github.com/petergtz/pegomock"
 )
 
 func TestCleanUpPullWorkspaceErr(t *testing.T) {
@@ -155,4 +156,3 @@ func AnyPullRequest() models.PullRequest {
 	RegisterMatcher(NewAnyMatcher(reflect.TypeOf(models.PullRequest{})))
 	return models.PullRequest{}
 }
-

server/events/webhooks/mocks/mock_slack_client.go

@@ -0,0 +1,180 @@
+// Automatically generated by pegomock. DO NOT EDIT!
+// Source: github.com/hootsuite/atlantis/server/events/webhooks (interfaces: SlackClient)
+
+package mocks
+
+import (
+	webhooks "github.com/hootsuite/atlantis/server/events/webhooks"
+	pegomock "github.com/petergtz/pegomock"
+	"reflect"
+)
+
+type MockSlackClient struct {
+	fail func(message string, callerSkip ...int)
+}
+
+func NewMockSlackClient() *MockSlackClient {
+	return &MockSlackClient{fail: pegomock.GlobalFailHandler}
+}
+
+func (mock *MockSlackClient) AuthTest() error {
+	params := []pegomock.Param{}
+	result := pegomock.GetGenericMockFrom(mock).Invoke("AuthTest", params, []reflect.Type{reflect.TypeOf((*error)(nil)).Elem()})
+	var ret0 error
+	if len(result) != 0 {
+		if result[0] != nil {
+			ret0 = result[0].(error)
+		}
+	}
+	return ret0
+}
+
+func (mock *MockSlackClient) TokenIsSet() bool {
+	params := []pegomock.Param{}
+	result := pegomock.GetGenericMockFrom(mock).Invoke("TokenIsSet", params, []reflect.Type{reflect.TypeOf((*bool)(nil)).Elem()})
+	var ret0 bool
+	if len(result) != 0 {
+		if result[0] != nil {
+			ret0 = result[0].(bool)
+		}
+	}
+	return ret0
+}
+
+func (mock *MockSlackClient) ChannelExists(channelName string) (bool, error) {
+	params := []pegomock.Param{channelName}
+	result := pegomock.GetGenericMockFrom(mock).Invoke("ChannelExists", params, []reflect.Type{reflect.TypeOf((*bool)(nil)).Elem(), reflect.TypeOf((*error)(nil)).Elem()})
+	var ret0 bool
+	var ret1 error
+	if len(result) != 0 {
+		if result[0] != nil {
+			ret0 = result[0].(bool)
+		}
+		if result[1] != nil {
+			ret1 = result[1].(error)
+		}
+	}
+	return ret0, ret1
+}
+
+func (mock *MockSlackClient) PostMessage(channel string, applyResult webhooks.ApplyResult) error {
+	params := []pegomock.Param{channel, applyResult}
+	result := pegomock.GetGenericMockFrom(mock).Invoke("PostMessage", params, []reflect.Type{reflect.TypeOf((*error)(nil)).Elem()})
+	var ret0 error
+	if len(result) != 0 {
+		if result[0] != nil {
+			ret0 = result[0].(error)
+		}
+	}
+	return ret0
+}
+
+func (mock *MockSlackClient) VerifyWasCalledOnce() *VerifierSlackClient {
+	return &VerifierSlackClient{mock, pegomock.Times(1), nil}
+}
+
+func (mock *MockSlackClient) VerifyWasCalled(invocationCountMatcher pegomock.Matcher) *VerifierSlackClient {
+	return &VerifierSlackClient{mock, invocationCountMatcher, nil}
+}
+
+func (mock *MockSlackClient) VerifyWasCalledInOrder(invocationCountMatcher pegomock.Matcher, inOrderContext *pegomock.InOrderContext) *VerifierSlackClient {
+	return &VerifierSlackClient{mock, invocationCountMatcher, inOrderContext}
+}
+
+type VerifierSlackClient struct {
+	mock                   *MockSlackClient
+	invocationCountMatcher pegomock.Matcher
+	inOrderContext         *pegomock.InOrderContext
+}
+
+func (verifier *VerifierSlackClient) AuthTest() *SlackClient_AuthTest_OngoingVerification {
+	params := []pegomock.Param{}
+	methodInvocations := pegomock.GetGenericMockFrom(verifier.mock).Verify(verifier.inOrderContext, verifier.invocationCountMatcher, "AuthTest", params)
+	return &SlackClient_AuthTest_OngoingVerification{mock: verifier.mock, methodInvocations: methodInvocations}
+}
+
+type SlackClient_AuthTest_OngoingVerification struct {
+	mock              *MockSlackClient
+	methodInvocations []pegomock.MethodInvocation
+}
+
+func (c *SlackClient_AuthTest_OngoingVerification) GetCapturedArguments() {
+}
+
+func (c *SlackClient_AuthTest_OngoingVerification) GetAllCapturedArguments() {
+}
+
+func (verifier *VerifierSlackClient) TokenIsSet() *SlackClient_TokenIsSet_OngoingVerification {
+	params := []pegomock.Param{}
+	methodInvocations := pegomock.GetGenericMockFrom(verifier.mock).Verify(verifier.inOrderContext, verifier.invocationCountMatcher, "TokenIsSet", params)
+	return &SlackClient_TokenIsSet_OngoingVerification{mock: verifier.mock, methodInvocations: methodInvocations}
+}
+
+type SlackClient_TokenIsSet_OngoingVerification struct {
+	mock              *MockSlackClient
+	methodInvocations []pegomock.MethodInvocation
+}
+
+func (c *SlackClient_TokenIsSet_OngoingVerification) GetCapturedArguments() {
+}
+
+func (c *SlackClient_TokenIsSet_OngoingVerification) GetAllCapturedArguments() {
+}
+
+func (verifier *VerifierSlackClient) ChannelExists(channelName string) *SlackClient_ChannelExists_OngoingVerification {
+	params := []pegomock.Param{channelName}
+	methodInvocations := pegomock.GetGenericMockFrom(verifier.mock).Verify(verifier.inOrderContext, verifier.invocationCountMatcher, "ChannelExists", params)
+	return &SlackClient_ChannelExists_OngoingVerification{mock: verifier.mock, methodInvocations: methodInvocations}
+}
+
+type SlackClient_ChannelExists_OngoingVerification struct {
+	mock              *MockSlackClient
+	methodInvocations []pegomock.MethodInvocation
+}
+
+func (c *SlackClient_ChannelExists_OngoingVerification) GetCapturedArguments() string {
+	channelName := c.GetAllCapturedArguments()
+	return channelName[len(channelName)-1]
+}
+
+func (c *SlackClient_ChannelExists_OngoingVerification) GetAllCapturedArguments() (_param0 []string) {
+	params := pegomock.GetGenericMockFrom(c.mock).GetInvocationParams(c.methodInvocations)
+	if len(params) > 0 {
+		_param0 = make([]string, len(params[0]))
+		for u, param := range params[0] {
+			_param0[u] = param.(string)
+		}
+	}
+	return
+}
+
+func (verifier *VerifierSlackClient) PostMessage(channel string, applyResult webhooks.ApplyResult) *SlackClient_PostMessage_OngoingVerification {
+	params := []pegomock.Param{channel, applyResult}
+	methodInvocations := pegomock.GetGenericMockFrom(verifier.mock).Verify(verifier.inOrderContext, verifier.invocationCountMatcher, "PostMessage", params)
+	return &SlackClient_PostMessage_OngoingVerification{mock: verifier.mock, methodInvocations: methodInvocations}
+}
+
+type SlackClient_PostMessage_OngoingVerification struct {
+	mock              *MockSlackClient
+	methodInvocations []pegomock.MethodInvocation
+}
+
+func (c *SlackClient_PostMessage_OngoingVerification) GetCapturedArguments() (string, webhooks.ApplyResult) {
+	channel, applyResult := c.GetAllCapturedArguments()
+	return channel[len(channel)-1], applyResult[len(applyResult)-1]
+}
+
+func (c *SlackClient_PostMessage_OngoingVerification) GetAllCapturedArguments() (_param0 []string, _param1 []webhooks.ApplyResult) {
+	params := pegomock.GetGenericMockFrom(c.mock).GetInvocationParams(c.methodInvocations)
+	if len(params) > 0 {
+		_param0 = make([]string, len(params[0]))
+		for u, param := range params[0] {
+			_param0[u] = param.(string)
+		}
+		_param1 = make([]webhooks.ApplyResult, len(params[1]))
+		for u, param := range params[1] {
+			_param1[u] = param.(webhooks.ApplyResult)
+		}
+	}
+	return
+}