From df68e139dca87992abaab0a51ce27d59e84ace10 Mon Sep 17 00:00:00 2001
From: xis <xis@schowek.net>
Date: Mon, 16 Oct 2023 20:50:26 +0200
Subject: [PATCH] StreamMessageMapperTest updated

---
 .../resources/application-integration.yml     |  1 -
 .../dlna/StreamMessageMapperTest.groovy       | 35 ++++++++++++++++++-
 2 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/src/integration/resources/application-integration.yml b/src/integration/resources/application-integration.yml
index dee3d54..4c4158b 100644
--- a/src/integration/resources/application-integration.yml
+++ b/src/integration/resources/application-integration.yml
@@ -16,4 +16,3 @@ spring:
       mode: always
       schema-locations: classpath:db/schema.sql
       data-locations: classpath:db/data.sql
-
diff --git a/src/test/groovy/net/schowek/nextclouddlna/dlna/StreamMessageMapperTest.groovy b/src/test/groovy/net/schowek/nextclouddlna/dlna/StreamMessageMapperTest.groovy
index e2ed49d..520f14e 100644
--- a/src/test/groovy/net/schowek/nextclouddlna/dlna/StreamMessageMapperTest.groovy
+++ b/src/test/groovy/net/schowek/nextclouddlna/dlna/StreamMessageMapperTest.groovy
@@ -1,7 +1,10 @@
 package net.schowek.nextclouddlna.dlna
 
-
+import org.jupnp.model.message.StreamResponseMessage
+import org.jupnp.model.message.UpnpHeaders
 import org.jupnp.model.message.UpnpRequest
+import org.jupnp.model.message.UpnpResponse
+import org.springframework.http.HttpStatus
 import org.springframework.mock.web.MockHttpServletRequest
 import spock.lang.Specification
 
@@ -59,5 +62,35 @@ class StreamMessageMapperTest extends Specification {
         "foo"  | _
     }
 
+    def "should map streamResponseMessage to ResponseEntity"() {
+        given:
+        def content = "some content"
+        def headers = [
+                "foo": ["bar"],
+                "baz": ["blah"]
+        ]
+
+        def response = new StreamResponseMessage(new UpnpResponse(responseStatus, "OK"))
+        response.headers = new UpnpHeaders(headers)
+        response.body = content
+
+        when:
+        def result = sut.map(response)
+
+        then:
+        result.statusCode == expectedHttpStatus
+        result.body == content
+        result.headers.each {
+            assert headers.keySet().contains(it.key.toLowerCase())
+            assert headers[it.key.toLowerCase()] == it.value
+        }
+
+        where:
+        responseStatus || expectedHttpStatus
+        200            || HttpStatus.OK
+        404            || HttpStatus.NOT_FOUND
+        500            || HttpStatus.INTERNAL_SERVER_ERROR
+        400            || HttpStatus.BAD_REQUEST
+    }
 
 }