From 01488228f2004269ef338210c71f043115395844 Mon Sep 17 00:00:00 2001
From: Samuel El-Borai <samuel.elborai@gmail.com>
Date: Sun, 11 Mar 2018 20:48:59 +0100
Subject: [PATCH] Add tests for (*Messenger).classify

---
 messenger_test.go | 62 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100644 messenger_test.go

diff --git a/messenger_test.go b/messenger_test.go
new file mode 100644
index 0000000..af59b5e
--- /dev/null
+++ b/messenger_test.go
@@ -0,0 +1,62 @@
+package messenger
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestMessenger_Classify(t *testing.T) {
+	m := New(Options{})
+
+	for name, test := range map[string]struct {
+		msgInfo  MessageInfo
+		expected Action
+	}{
+		"unknown": {
+			msgInfo:  MessageInfo{},
+			expected: UnknownAction,
+		},
+		"message": {
+			msgInfo: MessageInfo{
+				Message: &Message{},
+			},
+			expected: TextAction,
+		},
+		"delivery": {
+			msgInfo: MessageInfo{
+				Delivery: &Delivery{},
+			},
+			expected: DeliveryAction,
+		},
+		"read": {
+			msgInfo: MessageInfo{
+				Read: &Read{},
+			},
+			expected: ReadAction,
+		},
+		"postback": {
+			msgInfo: MessageInfo{
+				PostBack: &PostBack{},
+			},
+			expected: PostBackAction,
+		},
+		"optin": {
+			msgInfo: MessageInfo{
+				OptIn: &OptIn{},
+			},
+			expected: OptInAction,
+		},
+		"referral": {
+			msgInfo: MessageInfo{
+				ReferralMessage: &ReferralMessage{},
+			},
+			expected: ReferralAction,
+		},
+	} {
+		t.Run("action "+name, func(t *testing.T) {
+			action := m.classify(test.msgInfo, Entry{})
+			assert.Exactly(t, action, test.expected)
+		})
+	}
+}