From b8425b053e0ac70e917076ef036661da382bd7c3 Mon Sep 17 00:00:00 2001
From: Henner <kein.regen@wolke7.net>
Date: Sat, 7 Oct 2017 18:06:20 +0200
Subject: [PATCH] Add missing tests

---
 types_test.go | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/types_test.go b/types_test.go
index ed8d625..9ef03b1 100644
--- a/types_test.go
+++ b/types_test.go
@@ -91,6 +91,15 @@ func TestCommandWithBotName(t *testing.T) {
 	}
 }
 
+func TestCommandWithAtWithBotName(t *testing.T) {
+	message := tgbotapi.Message{Text: "/command@testbot"}
+	message.Entities = &[]tgbotapi.MessageEntity{{Type: "bot_command", Offset: 0, Length: 16}}
+
+	if message.CommandWithAt() != "command@testbot" {
+		t.Fail()
+	}
+}
+
 func TestMessageCommandArgumentsWithArguments(t *testing.T) {
 	message := tgbotapi.Message{Text: "/command with arguments"}
 	message.Entities = &[]tgbotapi.MessageEntity{{Type: "bot_command", Offset: 0, Length: 8}}
@@ -99,6 +108,14 @@ func TestMessageCommandArgumentsWithArguments(t *testing.T) {
 	}
 }
 
+func TestMessageCommandArgumentsWithMalformedArguments(t *testing.T) {
+	message := tgbotapi.Message{Text: "/command-without argument space"}
+	message.Entities = &[]tgbotapi.MessageEntity{{Type: "bot_command", Offset: 0, Length: 8}}
+	if message.CommandArguments() != "without argument space" {
+		t.Fail()
+	}
+}
+
 func TestMessageCommandArgumentsWithoutArguments(t *testing.T) {
 	message := tgbotapi.Message{Text: "/command"}
 	if message.CommandArguments() != "" {