add more tests to mux (#446)

Author: topi314

handler/mux_test.go

@@ -245,3 +245,62 @@ func TestMiddlewareMux(t *testing.T) {
 		assert.Equal(t, d.expectedVars3, dataVars3)
 	}
 }
+
+func TestMux(t *testing.T) {
+	buttonFooData, err := os.ReadFile("testdata/mux/button_foo_component.json")
+	assert.NoError(t, err)
+
+	buttonFooBarData, err := os.ReadFile("testdata/mux/button_foo_bar_component.json")
+	assert.NoError(t, err)
+
+	data := []struct {
+		data     []byte
+		expected *discord.InteractionResponse
+	}{
+		{
+			data: buttonFooData,
+			expected: &discord.InteractionResponse{
+				Type: discord.InteractionResponseTypeCreateMessage,
+				Data: discord.MessageCreate{
+					Content: "/foo/",
+				},
+			},
+		},
+		{
+			data: buttonFooBarData,
+			expected: &discord.InteractionResponse{
+				Type: discord.InteractionResponseTypeCreateMessage,
+				Data: discord.MessageCreate{
+					Content: "/foo/bar",
+				},
+			},
+		},
+	}
+
+	mux := New()
+	mux.Route("/foo", func(r Router) {
+		r.ButtonComponent("/", func(data discord.ButtonInteractionData, e *ComponentEvent) error {
+			return e.CreateMessage(discord.MessageCreate{
+				Content: "/foo/",
+			})
+		})
+		r.ButtonComponent("/bar", func(data discord.ButtonInteractionData, e *ComponentEvent) error {
+			return e.CreateMessage(discord.MessageCreate{
+				Content: "/foo/bar",
+			})
+		})
+	})
+
+	for _, d := range data {
+		interaction, err := discord.UnmarshalInteraction(d.data)
+		assert.NoError(t, err)
+
+		recorder := NewRecorder()
+		mux.OnEvent(&events.InteractionCreate{
+			GenericEvent: events.NewGenericEvent(nil, 0, 0),
+			Interaction:  interaction,
+			Respond:      recorder.Respond,
+		})
+		assert.Equal(t, d.expected, recorder.Response)
+	}
+}

handler/testdata/mux/button_foo_bar_component.json

@@ -0,0 +1,70 @@
+{
+    "version": 1,
+    "type": 3,
+    "token": "unique_interaction_token",
+    "message": {
+        "type": 0,
+        "tts": false,
+        "timestamp": "2021-05-19T02:12:51.710000+00:00",
+        "pinned": false,
+        "mentions": [],
+        "mention_roles": [],
+        "mention_everyone": false,
+        "id": "844397162624450620",
+        "flags": 0,
+        "embeds": [],
+        "edited_timestamp": null,
+        "content": "This is a message with components.",
+        "components": [
+            {
+                "type": 1,
+                "components": [
+                    {
+                        "type": 2,
+                        "label": "Click me!",
+                        "style": 1,
+                        "custom_id": "/foo/bar"
+                    }
+                ]
+            }
+        ],
+        "channel_id": "345626669114982402",
+        "author": {
+            "username": "Mason",
+            "public_flags": 131141,
+            "id": "53908232506183680",
+            "discriminator": "1337",
+            "avatar": "a_d5efa99b3eeaa7dd43acca82f5692432"
+        },
+        "attachments": []
+    },
+    "member": {
+        "user": {
+            "username": "Mason",
+            "public_flags": 131141,
+            "id": "53908232506183680",
+            "discriminator": "1337",
+            "avatar": "a_d5efa99b3eeaa7dd43acca82f5692432"
+        },
+        "roles": [
+            "290926798626357999"
+        ],
+        "premium_since": null,
+        "permissions": "17179869183",
+        "pending": false,
+        "nick": null,
+        "mute": false,
+        "joined_at": "2017-03-13T19:19:14.040000+00:00",
+        "is_pending": false,
+        "deaf": false,
+        "avatar": null
+    },
+    "id": "846462639134605312",
+    "guild_id": "290926798626357999",
+    "data": {
+        "custom_id": "/foo/bar",
+        "component_type": 2
+    },
+    "channel_id": "345626669114982999",
+    "application_id": "290926444748734465"
+}

handler/testdata/mux/button_foo_component.json

@@ -0,0 +1,70 @@
+{
+    "version": 1,
+    "type": 3,
+    "token": "unique_interaction_token",
+    "message": {
+        "type": 0,
+        "tts": false,
+        "timestamp": "2021-05-19T02:12:51.710000+00:00",
+        "pinned": false,
+        "mentions": [],
+        "mention_roles": [],
+        "mention_everyone": false,
+        "id": "844397162624450620",
+        "flags": 0,
+        "embeds": [],
+        "edited_timestamp": null,
+        "content": "This is a message with components.",
+        "components": [
+            {
+                "type": 1,
+                "components": [
+                    {
+                        "type": 2,
+                        "label": "Click me!",
+                        "style": 1,
+                        "custom_id": "/foo"
+                    }
+                ]
+            }
+        ],
+        "channel_id": "345626669114982402",
+        "author": {
+            "username": "Mason",
+            "public_flags": 131141,
+            "id": "53908232506183680",
+            "discriminator": "1337",
+            "avatar": "a_d5efa99b3eeaa7dd43acca82f5692432"
+        },
+        "attachments": []
+    },
+    "member": {
+        "user": {
+            "username": "Mason",
+            "public_flags": 131141,
+            "id": "53908232506183680",
+            "discriminator": "1337",
+            "avatar": "a_d5efa99b3eeaa7dd43acca82f5692432"
+        },
+        "roles": [
+            "290926798626357999"
+        ],
+        "premium_since": null,
+        "permissions": "17179869183",
+        "pending": false,
+        "nick": null,
+        "mute": false,
+        "joined_at": "2017-03-13T19:19:14.040000+00:00",
+        "is_pending": false,
+        "deaf": false,
+        "avatar": null
+    },
+    "id": "846462639134605312",
+    "guild_id": "290926798626357999",
+    "data": {
+        "custom_id": "/foo",
+        "component_type": 2
+    },
+    "channel_id": "345626669114982999",
+    "application_id": "290926444748734465"
+}