harness-drone/plugin/notify/hipchat_test.go

package notify

import (
	"testing"

	"github.com/drone/drone/shared/model"
)

type MockHipchatClient struct {
	Request HipchatMessageRequest
}

func (c *MockHipchatClient) PostMessage(req HipchatMessageRequest) error {
	c.Request = req
	return nil
}

var client = &MockHipchatClient{}

var subject = &Hipchat{
	Room:    "SampleRoom",
	Token:   "foo",
	Started: true,
	Success: true,
	Failure: true,
}

var request = &model.Request{
	Host: "http://examplehost.com",
	Repo: &model.Repo{
		Host:  "examplegit.com",
		Owner: "owner",
		Name:  "repo",
	},
	Commit: &model.Commit{
		Sha:     "abc",
		Branch:  "example",
		Status:  "Started",
		Message: "Test Commit",
		Author:  "Test User",
	},
	User: &model.User{
		Login: "TestUser",
	},
}

func Test_SendStarted(t *testing.T) {
	request.Commit.Status = "Started"

	subject.SendWithClient(client, request)
	expected := HipchatMessageRequest{
		RoomId:    "SampleRoom",
		AuthToken: "foo",
		Color:     "yellow",
		Message:   "Building <a href=\"http://examplehost.com/examplegit.com/owner/repo/example/abc\">owner/repo#abc</a> (example) by Test User <br> - Test Commit",
		Notify:    false,
	}

	if client.Request != expected {
		t.Errorf("Invalid hipchat payload. Expected: %v, got %v", expected, client.Request)
	}
}

func Test_SendSuccess(t *testing.T) {
	request.Commit.Status = "Success"

	subject.SendWithClient(client, request)
	expected := HipchatMessageRequest{
		RoomId:    "SampleRoom",
		AuthToken: "foo",
		Color:     "green",
		Message:   "Success <a href=\"http://examplehost.com/examplegit.com/owner/repo/example/abc\">owner/repo#abc</a> (example) by Test User",
		Notify:    false,
	}

	if client.Request != expected {
		t.Errorf("Invalid hipchat payload. Expected: %v, got %v", expected, client.Request)
	}
}

func Test_SendFailure(t *testing.T) {
	request.Commit.Status = "Failure"

	subject.SendWithClient(client, request)
	expected := HipchatMessageRequest{
		RoomId:    "SampleRoom",
		AuthToken: "foo",
		Color:     "red",
		Message:   "Failed <a href=\"http://examplehost.com/examplegit.com/owner/repo/example/abc\">owner/repo#abc</a> (example) by Test User",
		Notify:    true,
	}

	if client.Request != expected {
		t.Errorf("Invalid hipchat payload. Expected: %v, got %v", expected, client.Request)
	}
}