gophish/webhook/webhook_test.go

package webhook

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"net/http/httptest"
	"testing"

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

type WebhookSuite struct {
	suite.Suite
}

type mockSender struct {
	client *http.Client
}

func newMockSender() *mockSender {
	ms := &mockSender{
		client: &http.Client{},
	}
	return ms
}

func (ms mockSender) Send(endPoint EndPoint, data interface{}) error {
	log.Println("[test] mocked 'Send' function")
	return nil
}

func (s *WebhookSuite) TestSendMocked() {
	mcSnd := newMockSender()
	endp1 := EndPoint{URL: "http://example.com/a1", Secret: "s1"}
	d1 := map[string]string{
		"a1": "a11",
		"a2": "a22",
		"a3": "a33",
	}
	err := mcSnd.Send(endp1, d1)
	s.Nil(err)
}

func (s *WebhookSuite) TestSendReal() {
	expectedSign := "004b36ca3fcbc01a08b17bf5d4a7e1aa0b10e14f55f3f8bd9acac0c7e8d2635d"
	secret := "secret456"
	d1 := map[string]interface{}{
		"key1": "val1",
		"key2": "val2",
		"key3": "val3",
	}

	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fmt.Println("[test] running the server...")

		signStartIdx := len(Sha256Prefix) + 1
		realSignRaw := r.Header.Get(SignatureHeader)
		realSign := realSignRaw[signStartIdx:]
		assert.Equal(s.T(), expectedSign, realSign)

		contTypeJsonHeader := r.Header.Get("Content-Type")
		assert.Equal(s.T(), contTypeJsonHeader, "application/json")

		body, err := ioutil.ReadAll(r.Body)
		s.Nil(err)

		var d2 map[string]interface{}
		err = json.Unmarshal(body, &d2)
		s.Nil(err)
		assert.Equal(s.T(), d1, d2)
	}))

	defer ts.Close()
	endp1 := EndPoint{URL: ts.URL, Secret: secret}
	err := Send(endp1, d1)
	s.Nil(err)
}

func (s *WebhookSuite) TestSignature() {
	secret := "secret123"
	payload := []byte("some payload456")
	expectedSign := "ab7844c1e9149f8dc976c4188a72163c005930f3c2266a163ffe434230bdf761"
	realSign, err := sign(secret, payload)
	s.Nil(err)
	assert.Equal(s.T(), expectedSign, realSign)
}

func TestWebhookSuite(t *testing.T) {
	suite.Run(t, new(WebhookSuite))
}