1
2
3
4
5
6
7
8
9
10
11 package org.mule.module.pgp;
12
13 import org.mule.RequestContext;
14 import org.mule.api.MuleEvent;
15 import org.mule.tck.testmodels.fruit.Orange;
16 import org.mule.transformer.encryption.EncryptionTransformer;
17 import org.mule.transformer.simple.ByteArrayToObject;
18
19 import java.io.InputStream;
20
21 import org.apache.commons.io.IOUtils;
22 import org.junit.Test;
23
24 import static org.junit.Assert.assertNotNull;
25 import static org.junit.Assert.assertTrue;
26
27 public class KBEStrategyUsingEncryptionTransformerTestCase extends AbstractEncryptionStrategyTestCase
28 {
29 @Test
30 public void testEncrypt() throws Exception
31 {
32 String msg = "Test Message";
33
34 MuleEvent event = getTestEvent(msg, getTestService("orange", Orange.class));
35 event = RequestContext.setEvent(event);
36
37 EncryptionTransformer etrans = new EncryptionTransformer();
38 etrans.setStrategy(kbStrategy);
39 Object result = etrans.doTransform(msg.getBytes(), "UTF-8");
40
41 assertNotNull(result);
42 InputStream inputStream = (InputStream) result;
43 String message = IOUtils.toString(inputStream);
44 String encrypted = (String) new ByteArrayToObject().doTransform(message.getBytes(), "UTF-8");
45 assertTrue(encrypted.startsWith("-----BEGIN PGP MESSAGE-----"));
46 }
47 }