Add resource functions to send mime:Entity[] payloads in mock services

ballerina-platform · Nov 1, 2023 · a9fdd66 · a9fdd66
1 parent c99139f
commit a9fdd66
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 20 deletions.
diff --git a/ballerina/modules/soap11/tests/http_soap_service.bal b/ballerina/modules/soap11/tests/http_soap_service.bal
@@ -16,6 +16,7 @@
 
 import ballerina/crypto;
 import ballerina/http;
+import ballerina/mime;
 import ballerina/soap;
 
 const crypto:KeyStore serverKeyStore = {
@@ -34,6 +35,19 @@ service / on new http:Listener(9090) {
         return response;
     }
 
+    resource function post getMimePayload(http:Request request) returns http:Response|error {
+        http:Response response = new;
+        mime:Entity[] mtomMessage = [];
+        mime:Entity envelope = new;
+        check envelope.setContentType("application/xop+xml");
+        envelope.setContentId("<soap@envelope>");
+        envelope.setBody(check (check request.getBodyParts())[0].getXml());
+        mtomMessage.push(envelope);
+        response.setBodyParts(mtomMessage);
+        response.setPayload(mtomMessage);
+        return response;
+    }
+
     resource function post getSamePayload(http:Request request) returns http:Response|error {
         xml payload = check request.getXmlPayload();
         http:Response response = new;
@@ -45,18 +59,18 @@ service / on new http:Listener(9090) {
         xml payload = check request.getXmlPayload();
         xml applyOutboundConfig = check soap:applyOutboundConfig(
             {
-                verificationKey: clientPublicKey,
-                signatureAlgorithm: soap:RSA_SHA256,
-                decryptionAlgorithm: soap:RSA_ECB,
-                decryptionKey: serverPrivateKey
-            }, payload);
+            verificationKey: clientPublicKey,
+            signatureAlgorithm: soap:RSA_SHA256,
+            decryptionAlgorithm: soap:RSA_ECB,
+            decryptionKey: serverPrivateKey
+        }, payload);
         xml securedEnv = check soap:applySecurityPolicies(
             {
-                signatureAlgorithm: soap:RSA_SHA256,
-                encryptionAlgorithm: soap:RSA_ECB,
-                signatureKey: serverPrivateKey,
-                encryptionKey: clientPublicKey
-            }, applyOutboundConfig);
+            signatureAlgorithm: soap:RSA_SHA256,
+            encryptionAlgorithm: soap:RSA_ECB,
+            signatureKey: serverPrivateKey,
+            encryptionKey: clientPublicKey
+        }, applyOutboundConfig);
         http:Response response = new;
         response.setPayload(securedEnv);
         return response;

diff --git a/ballerina/modules/soap12/tests/http_soap_service.bal b/ballerina/modules/soap12/tests/http_soap_service.bal
@@ -16,6 +16,7 @@
 
 import ballerina/crypto;
 import ballerina/http;
+import ballerina/mime;
 import ballerina/soap;
 
 const crypto:KeyStore serverKeyStore = {
@@ -34,6 +35,19 @@ service / on new http:Listener(9090) {
         return response;
     }
 
+    resource function post getMimePayload(http:Request request) returns http:Response|error {
+        http:Response response = new;
+        mime:Entity[] mtomMessage = [];
+        mime:Entity envelope = new;
+        check envelope.setContentType("application/xop+xml");
+        envelope.setContentId("<soap@envelope>");
+        envelope.setBody(check (check request.getBodyParts())[0].getXml());
+        mtomMessage.push(envelope);
+        response.setBodyParts(mtomMessage);
+        response.setPayload(mtomMessage);
+        return response;
+    }
+
     resource function post getSamePayload(http:Request request) returns http:Response|error {
         xml payload = check request.getXmlPayload();
         http:Response response = new;
@@ -45,18 +59,18 @@ service / on new http:Listener(9090) {
         xml payload = check request.getXmlPayload();
         xml applyOutboundConfig = check soap:applyOutboundConfig(
             {
-                verificationKey: clientPublicKey,
-                signatureAlgorithm: soap:RSA_SHA256,
-                decryptionAlgorithm: soap:RSA_ECB,
-                decryptionKey: serverPrivateKey
-            }, payload);
+            verificationKey: clientPublicKey,
+            signatureAlgorithm: soap:RSA_SHA256,
+            decryptionAlgorithm: soap:RSA_ECB,
+            decryptionKey: serverPrivateKey
+        }, payload);
         xml securedEnv = check soap:applySecurityPolicies(
             {
-                signatureAlgorithm: soap:RSA_SHA256,
-                encryptionAlgorithm: soap:RSA_ECB,
-                signatureKey: serverPrivateKey,
-                encryptionKey: clientPublicKey
-            }, applyOutboundConfig);
+            signatureAlgorithm: soap:RSA_SHA256,
+            encryptionAlgorithm: soap:RSA_ECB,
+            signatureKey: serverPrivateKey,
+            encryptionKey: clientPublicKey
+        }, applyOutboundConfig);
         http:Response response = new;
         response.setPayload(securedEnv);
         return response;