Added post method to DavResource

Signed-off-by: Arnau Mora <arnyminerz@proton.me>
bitfireAT · Jun 4, 2024 · c7205f9 · c7205f9
1 parent b87d772
commit c7205f9
Show file tree

Hide file tree

Showing 3 changed files with 115 additions and 33 deletions.
diff --git a/src/main/kotlin/at/bitfire/dav4jvm/DavCollection.kt b/src/main/kotlin/at/bitfire/dav4jvm/DavCollection.kt
@@ -11,14 +11,12 @@ import at.bitfire.dav4jvm.exception.DavException
 import at.bitfire.dav4jvm.exception.HttpException
 import at.bitfire.dav4jvm.property.webdav.NS_WEBDAV
 import at.bitfire.dav4jvm.property.webdav.SyncToken
+import java.io.StringWriter
+import java.util.logging.Logger
 import okhttp3.HttpUrl
 import okhttp3.OkHttpClient
 import okhttp3.Request
-import okhttp3.RequestBody
 import okhttp3.RequestBody.Companion.toRequestBody
-import java.io.IOException
-import java.io.StringWriter
-import java.util.logging.Logger
 
 /**
  * Represents a WebDAV collection.
@@ -36,27 +34,6 @@ open class DavCollection @JvmOverloads constructor(
         val NRESULTS = Property.Name(NS_WEBDAV, "nresults")
     }
 
-    /**
-     * Sends a POST request. Primarily intended to be used with an Add-Member URL (RFC 5995).
-     */
-    @Throws(IOException::class, HttpException::class)
-    fun post(body: RequestBody, ifNoneMatch: Boolean = false, callback: ResponseCallback) {
-        followRedirects {
-            val builder = Request.Builder()
-                    .post(body)
-                    .url(location)
-
-            if (ifNoneMatch)
-                // don't overwrite anything existing
-                builder.header("If-None-Match", "*")
-
-            httpClient.newCall(builder.build()).execute()
-        }.use { response ->
-            checkStatus(response)
-            callback.onResponse(response)
-        }
-    }
-
     /**
      * Sends a REPORT sync-collection request.
      *

diff --git a/src/main/kotlin/at/bitfire/dav4jvm/DavResource.kt b/src/main/kotlin/at/bitfire/dav4jvm/DavResource.kt
@@ -20,6 +20,13 @@ import at.bitfire.dav4jvm.property.caldav.NS_CALDAV
 import at.bitfire.dav4jvm.property.carddav.NS_CARDDAV
 import at.bitfire.dav4jvm.property.webdav.NS_WEBDAV
 import at.bitfire.dav4jvm.property.webdav.SyncToken
+import java.io.EOFException
+import java.io.IOException
+import java.io.Reader
+import java.io.StringWriter
+import java.net.HttpURLConnection
+import java.util.logging.Level
+import java.util.logging.Logger
 import okhttp3.Headers
 import okhttp3.HttpUrl
 import okhttp3.MediaType.Companion.toMediaType
@@ -30,13 +37,6 @@ import okhttp3.RequestBody.Companion.toRequestBody
 import okhttp3.Response
 import org.xmlpull.v1.XmlPullParser
 import org.xmlpull.v1.XmlPullParserException
-import java.io.EOFException
-import java.io.IOException
-import java.io.Reader
-import java.io.StringWriter
-import java.net.HttpURLConnection
-import java.util.logging.Level
-import java.util.logging.Logger
 import at.bitfire.dav4jvm.Response as DavResponse
 
 /**
@@ -412,6 +412,30 @@ open class DavResource @JvmOverloads constructor(
         }
     }
 
+    /**
+     * Sends a GET request to the resource. Follows up to [MAX_REDIRECTS] redirects.
+     */
+    @Throws(IOException::class, HttpException::class)
+    fun post(body: RequestBody, ifNoneMatch: Boolean = false, headers: Headers? = null, callback: ResponseCallback) {
+        followRedirects {
+            val builder = Request.Builder()
+                .post(body)
+                .url(location)
+
+            if (ifNoneMatch)
+            // don't overwrite anything existing
+                builder.header("If-None-Match", "*")
+
+            if (headers != null)
+                builder.headers(headers)
+
+            httpClient.newCall(builder.build()).execute()
+        }.use { response ->
+            checkStatus(response)
+            callback.onResponse(response)
+        }
+    }
+
     /**
      * Sends a PUT request to the resource. Follows up to [MAX_REDIRECTS] redirects.
      *

diff --git a/src/test/kotlin/at/bitfire/dav4jvm/DavResourceTest.kt b/src/test/kotlin/at/bitfire/dav4jvm/DavResourceTest.kt
@@ -13,6 +13,7 @@ import at.bitfire.dav4jvm.property.webdav.DisplayName
 import at.bitfire.dav4jvm.property.webdav.GetContentType
 import at.bitfire.dav4jvm.property.webdav.GetETag
 import at.bitfire.dav4jvm.property.webdav.ResourceType
+import java.net.HttpURLConnection
 import okhttp3.HttpUrl.Companion.toHttpUrl
 import okhttp3.MediaType.Companion.toMediaType
 import okhttp3.OkHttpClient
@@ -30,7 +31,6 @@ import org.junit.Assert.assertTrue
 import org.junit.Assert.fail
 import org.junit.Before
 import org.junit.Test
-import java.net.HttpURLConnection
 
 class DavResourceTest {
 
@@ -310,6 +310,87 @@ class DavResourceTest {
         assertTrue(called)
     }
 
+    @Test
+    fun testPost() {
+        val url = sampleUrl()
+        val dav = DavResource(httpClient, url)
+
+        /* POSITIVE TEST CASES */
+
+        // 200 OK
+        mockServer.enqueue(
+            MockResponse()
+                .setResponseCode(HttpURLConnection.HTTP_OK)
+                .setHeader("ETag", "W/\"My Weak ETag\"")
+                .setHeader("Content-Type", "application/x-test-result")
+                .setBody(sampleText)
+        )
+        var called = false
+        dav.post(
+            body = "body".toRequestBody("application/x-test-result".toMediaType())
+        ) { response ->
+            called = true
+            assertEquals(sampleText, response.body!!.string())
+
+            val eTag = GetETag.fromResponse(response)
+            assertEquals("My Weak ETag", eTag!!.eTag)
+            assertTrue(eTag.weak)
+            assertEquals("application/x-test-result".toMediaType(), GetContentType(response.body!!.contentType()!!).type)
+        }
+        assertTrue(called)
+
+        var rq = mockServer.takeRequest()
+        assertEquals("POST", rq.method)
+        assertEquals(url.encodedPath, rq.path)
+        assertTrue(rq.getHeader("Content-Type")?.contains("application/x-test-result") == true)
+        assertEquals("body", rq.body.readUtf8())
+
+        // 302 Moved Temporarily + 200 OK
+        mockServer.enqueue(
+            MockResponse()
+                .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP)
+                .setHeader("Location", "/target")
+                .setBody("This resource was moved.")
+        )
+        mockServer.enqueue(
+            MockResponse()
+                .setResponseCode(HttpURLConnection.HTTP_OK)
+                .setHeader("ETag", "\"StrongETag\"")
+                .setBody(sampleText)
+        )
+        called = false
+        dav.post(
+            body = "body".toRequestBody("application/x-test-result".toMediaType())
+        ) { response ->
+            called = true
+            assertEquals(sampleText, response.body!!.string())
+            val eTag = GetETag(response.header("ETag")!!)
+            assertEquals("StrongETag", eTag.eTag)
+            assertFalse(eTag.weak)
+        }
+        assertTrue(called)
+
+        mockServer.takeRequest()
+        rq = mockServer.takeRequest()
+        assertEquals("POST", rq.method)
+        assertEquals("/target", rq.path)
+
+        // 200 OK without ETag in response
+        mockServer.enqueue(
+            MockResponse()
+                .setResponseCode(HttpURLConnection.HTTP_OK)
+                .setBody(sampleText)
+        )
+        called = false
+        dav.post(
+            body = "body".toRequestBody("application/x-test-result".toMediaType())
+        ) { response ->
+            called = true
+            assertNull(response.header("ETag"))
+        }
+        assertTrue(called)
+    }
+
     @Test
     fun testMove() {
         val url = sampleUrl()