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 owner/repo#abc (example) by Test User
- 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 owner/repo#abc (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 owner/repo#abc (example) by Test User", Notify: true, } if client.Request != expected { t.Errorf("Invalid hipchat payload. Expected: %v, got %v", expected, client.Request) } }