messenger_test.go

package messenger

import (
	"math/rand"
	"net/http"
	"net/http/httptest"
	"net/url"
	"strings"
	"sync"
	"testing"
	"time"

	"github.com/drewolson/testflight"
)

func setClient(code int, body []byte) *httptest.Server {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(code)
		w.Write(body)
	}))

	http.DefaultClient.Transport = &http.Transport{
		Proxy: func(req *http.Request) (*url.URL, error) {
			return url.Parse(server.URL)
		},
	}
	return server
}

//seed the rand
func init() {
	rand.Seed(time.Now().UnixNano())
}

var letters = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")

func randString(leng uint) string {
	b := make([]rune, leng)
	for i := range b {
		b[i] = letters[rand.Intn(len(letters))]
	}
	return string(b)
}

func TestCheckIntegrity(t *testing.T) {
	if !checkIntegrity("e6af24be1d683c8c911949f897eea1f6", []byte(`{"object":"page","entry":[{"id":"1751036168465324","time":1460923697656,"messaging":[{"sender":{"id":"982337261802700"},"recipient":{"id":"1751036168465324"},"timestamp":1460923697635,"message":{"mid":"mid.1460923697625:5c96e8279b55505308","seq":614,"text":"Test \u00e4\u00eb\u00ef"}}]}]}`), "da611bd448dc12acdf0cd3ab33fdb3adaee26145") {
		t.Error("Message integrity verification does not work")
	}

	if checkIntegrity("e6af24be1d683c8c911949f897eea1f6", []byte(`{"object":"page","entry":[]}`), "f1a4569dcf02a9829a15696d949b386b7d6d0272") {
		t.Error("Message integrity verification does not work")
	}
}

func TestHandler(t *testing.T) {
	mess := &Messenger{
		AccessToken: "foo",
		VerifyToken: "bar",
	}
	testflight.WithServer(http.HandlerFunc(mess.Handler), func(r *testflight.Requester) {
		// Legit verify request
		response := r.Get("/?hub.verify_token=bar&hub.challenge=zoo")
		if response.StatusCode != http.StatusOK {
			t.Errorf("Invalid status code, expected %d, got: %d", http.StatusOK, response.StatusCode)
		}
		if response.Body != "zoo" {
			t.Error("Invalid body.")
		}
		// Invalid verify_token
		response = r.Get("/?hub.verify_token=abba&hub.challenge=zoo")
		if response.StatusCode != http.StatusUnauthorized {
			t.Errorf("Invalid status code, expected %d, got: %d", http.StatusUnauthorized, response.StatusCode)
		}
		if response.Body != "" {
			t.Error("Invalid body, expected to be empty.")
		}
		// Invalid method
		response = r.Put("/", "application/json", "foo-bar")
		if response.StatusCode != http.StatusMethodNotAllowed {
			t.Errorf("Invalid status code, expected %d, got: %d", http.StatusMethodNotAllowed, response.StatusCode)
		}

		mess.AppSecret = "e6af24be1d683c8c911949f897eea1f6"
		// Legit Post request
		postRequest, err := http.NewRequest("POST", "/", strings.NewReader(`{"object":"page","entry":[{"id":"1751036168465324","time":1460923697656,"messaging":[{"sender":{"id":"982337261802700"},"recipient":{"id":"1751036168465324"},"timestamp":1460923697635,"message":{"mid":"mid.1460923697625:5c96e8279b55505308","seq":614,"text":"Test \u00e4\u00eb\u00ef"}}]}]}`))
		if err != nil {
			t.Error(err)
		}
		postRequest.Header.Add("x-hub-signature", "sha1=da611bd448dc12acdf0cd3ab33fdb3adaee26145")
		response = r.Do(postRequest)
		if response.StatusCode != http.StatusOK {
			t.Errorf("Invalid status code, expected %d, got: %d", http.StatusOK, response.StatusCode)
		}

		// Invalid signature
		mess.AppSecret = "abc"

		// Invalid signature
		response = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"1751036168465324","time":1460923697656,"messaging":[{"sender":{"id":"982337261802700"},"recipient":{"id":"1751036168465324"},"timestamp":1460923697635,"message":{"mid":"mid.1460923697625:5c96e8279b55505308","seq":614,"text":"Test \u00e4\u00eb\u00ef"}}]}]}`)
		if response.StatusCode != http.StatusBadRequest {
			t.Errorf("Invalid status code, expected %d, got: %d", http.StatusBadRequest, response.StatusCode)
		}

		mess.AppSecret = ""
		// Invalid request
		response = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"1751036168465324","time":1460923697656,"messaging":[{"sender":{"id":"982337261802701751036168465324"},"timestamp":1460923697635,"message":{"mid":"mid.1460923697625:5c96e8279b55505308","seq":614,"text":"Test \u00e4\u00eb\u00e`)
		if response.StatusCode != http.StatusBadRequest {
			t.Errorf("Invalid status code, expected %d, got: %d", http.StatusBadRequest, response.StatusCode)
		}
		wg := sync.WaitGroup{}
		mess.MessageReceived = func(Event, MessageOpts, ReceivedMessage) {
			wg.Done()
		}
		mess.MessageDelivered = func(Event, MessageOpts, Delivery) {
			wg.Done()
		}
		mess.Postback = func(Event, MessageOpts, Postback) {
			wg.Done()
		}
		mess.Authentication = func(event Event, opts MessageOpts, optin *Optin) {
			wg.Done()
		}
		mess.MessageRead = func(Event, MessageOpts, Read) {
			wg.Done()
		}
		mess.MessageEcho = func(Event, MessageOpts, MessageEcho) {
			wg.Done()
		}

		wg.Add(6)
		// received
		_ = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"510249619162304","time":1468152703635,"messaging":[{"sender":{"id":"1066835436691078"},"recipient":{"id":"510249619162304"},"timestamp":1468152703534,"message":{"mid":"mid.1468152703527:6600c706f15a292027","seq":414,"text":"test"}}]}]}`)
		// echo
		_ = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"510249619162304","time":1468152706999,"messaging":[{"sender":{"id":"510249619162304"},"recipient":{"id":"1066835436691078"},"timestamp":1468152706921,"message":{"is_echo":true,"app_id":1708188056130407,"mid":"mid.1468152706286:e7982eb7618034a075","seq":415,"attachments":[{"title":"abc","url":null,"type":"template","payload":{"template_type":"generic","sharable":false,"elements":[{"title":"abc","buttons":[{"type":"postback","url":null,"title":"abecad\u0142o","payload":"test"}]}]}}]}}]}]}`)
		// delivery
		_ = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"510249619162304","time":1468152707422,"messaging":[{"sender":{"id":"1066835436691078"},"recipient":{"id":"510249619162304"},"timestamp":0,"delivery":{"mids":["mid.1468152706286:e7982eb7618034a075"],"watermark":1468152706921,"seq":416}}]}]}`)
		// read
		_ = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"510249619162304","time":1468152897133,"messaging":[{"sender":{"id":"1066835436691078"},"recipient":{"id":"510249619162304"},"timestamp":1468152897070,"read":{"watermark":1468152706921,"seq":417}}]}]}`)
		// postback
		_ = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"510249619162304","time":1468152897212,"messaging":[{"sender":{"id":"1066835436691078"},"recipient":{"id":"510249619162304"},"timestamp":1468152897212,"postback":{"payload":"test"}}]}]}`)
		// authentication
		_ = r.Post("/", "application/json", `{"object":"page","entry":[{"id":"510249619162304","time":1468152897212,"messaging":[{"sender":{"id":"1066835436691078"},"recipient":{"id":"510249619162304"},"timestamp":1468152897212,"optin":{"ref":"test"}}]}]}`)

		c := make(chan bool)
		go func() {
			wg.Wait()
			c <- true
		}()

		select {
		case <-c:
		case <-time.After(time.Second * 1):
			t.Error("Some handlers were not executed")
		}
	})
}