webhooks_test.go

package slack

import (
	"encoding/json"
	"net/http"
	"reflect"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestPostWebhook_OK(t *testing.T) {
	once.Do(startServer)

	var receivedPayload WebhookMessage

	http.HandleFunc("/webhook", func(rw http.ResponseWriter, r *http.Request) {
		rw.Header().Set("Content-Type", "application/json")

		decoder := json.NewDecoder(r.Body)
		err := decoder.Decode(&receivedPayload)
		if err != nil {
			t.Errorf("Request contained invalid JSON, %s", err)
		}

		response := []byte(`{}`)
		rw.Write(response)
	})

	url := "http://" + serverAddr + "/webhook"

	payload := &WebhookMessage{
		Text: "Test Text",
		Attachments: []Attachment{
			{
				Text: "Foo",
			},
		},
	}

	err := PostWebhook(url, payload)

	if err != nil {
		t.Errorf("Expected not to receive error: %s", err)
	}

	if !reflect.DeepEqual(payload, &receivedPayload) {
		t.Errorf("Payload did not match\nwant: %#v\n got: %#v", payload, receivedPayload)
	}
}

func TestPostWebhook_NotOK(t *testing.T) {
	once.Do(startServer)

	http.HandleFunc("/webhook2", func(rw http.ResponseWriter, r *http.Request) {
		rw.WriteHeader(http.StatusInternalServerError)
		rw.Write([]byte("500 - Something bad happened!"))
	})

	url := "http://" + serverAddr + "/webhook2"

	err := PostWebhook(url, &WebhookMessage{})

	if err == nil {
		t.Errorf("Expected to receive error")
	}
}

func TestWebhookMessage_WithBlocks(t *testing.T) {
	textBlockObject := NewTextBlockObject("plain_text", "text", false, false)
	sectionBlock := NewSectionBlock(textBlockObject, nil, nil)

	singleBlock := &Blocks{BlockSet: []Block{sectionBlock}}
	twoBlocks := &Blocks{BlockSet: []Block{sectionBlock, sectionBlock}}

	msgSingleBlock := WebhookMessage{Blocks: singleBlock}
	assert.Equal(t, 1, len(msgSingleBlock.Blocks.BlockSet))

	msgJsonSingleBlock, _ := json.Marshal(msgSingleBlock)
	assert.Equal(t, `{"blocks":[{"type":"section","text":{"type":"plain_text","text":"text"}}],"replace_original":false,"delete_original":false}`, string(msgJsonSingleBlock))

	msgTwoBlocks := WebhookMessage{Blocks: twoBlocks}
	assert.Equal(t, 2, len(msgTwoBlocks.Blocks.BlockSet))

	msgJsonTwoBlocks, _ := json.Marshal(msgTwoBlocks)
	assert.Equal(t, `{"blocks":[{"type":"section","text":{"type":"plain_text","text":"text"}},{"type":"section","text":{"type":"plain_text","text":"text"}}],"replace_original":false,"delete_original":false}`, string(msgJsonTwoBlocks))

	msgNoBlocks := WebhookMessage{Text: "foo"}
	msgJsonNoBlocks, _ := json.Marshal(msgNoBlocks)
	assert.Equal(t, `{"text":"foo","replace_original":false,"delete_original":false}`, string(msgJsonNoBlocks))
}