From e3eabbebfe2b3fafe182769d0e3cc19d92979b04 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Sat, 25 Mar 2017 12:02:37 +0100
Subject: [PATCH] Add Request utility methods

---
 .../net/pterodactylus/sone/utils/Requests.kt       |  8 ++++++
 .../net/pterodactylus/sone/utils/RequestsTest.kt   | 31 ++++++++++++++++++++++
 2 files changed, 39 insertions(+)
 create mode 100644 src/main/kotlin/net/pterodactylus/sone/utils/Requests.kt
 create mode 100644 src/test/kotlin/net/pterodactylus/sone/utils/RequestsTest.kt

diff --git a/src/main/kotlin/net/pterodactylus/sone/utils/Requests.kt b/src/main/kotlin/net/pterodactylus/sone/utils/Requests.kt
new file mode 100644
index 0000000..50f3957
--- /dev/null
+++ b/src/main/kotlin/net/pterodactylus/sone/utils/Requests.kt
@@ -0,0 +1,8 @@
+package net.pterodactylus.sone.utils
+
+import net.pterodactylus.util.web.Method.GET
+import net.pterodactylus.util.web.Method.POST
+import net.pterodactylus.util.web.Request
+
+val Request.isGET get() = this.method == GET
+val Request.isPOST get() = this.method == POST
diff --git a/src/test/kotlin/net/pterodactylus/sone/utils/RequestsTest.kt b/src/test/kotlin/net/pterodactylus/sone/utils/RequestsTest.kt
new file mode 100644
index 0000000..eb0bc43
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/utils/RequestsTest.kt
@@ -0,0 +1,31 @@
+package net.pterodactylus.sone.utils
+
+import net.pterodactylus.sone.test.mock
+import net.pterodactylus.sone.test.whenever
+import net.pterodactylus.util.web.Method.GET
+import net.pterodactylus.util.web.Method.POST
+import net.pterodactylus.util.web.Request
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.equalTo
+import org.junit.Test
+
+/**
+ * Unit test for the [Request] utilities.
+ */
+class RequestsTest {
+
+	@Test
+	fun `GET request is recognized correctly`() {
+		val request = mock<Request>().apply { whenever(method).thenReturn(GET) }
+		assertThat(request.isGET, equalTo(true))
+		assertThat(request.isPOST, equalTo(false))
+	}
+
+	@Test
+	fun `POST request is recognized correctly`() {
+		val request = mock<Request>().apply { whenever(method).thenReturn(POST) }
+		assertThat(request.isGET, equalTo(false))
+		assertThat(request.isPOST, equalTo(true))
+	}
+
+}
-- 
2.7.4