* Licensed to the Apache Software Foundation (ASF) under one

* or more contributor license agreements.  See the NOTICE file

* distributed with this work for additional information

* regarding copyright ownership.  The ASF licenses this file

* to you under the Apache License, Version 2.0 (the

* "License"); you may not use this file except in compliance

* with the License.  You may obtain a copy of the License at

*

*   http://www.apache.org/licenses/LICENSE-2.0

*

* Unless required by applicable law or agreed to in writing,

* software distributed under the License is distributed on an

* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

* KIND, either express or implied.  See the License for the

* specific language governing permissions and limitations

* under the License.

*/

    private ResourceResolver resolver;

    private Resource testRoot;

    private MessageStoreImpl store;

    /**

     * Some code is taken from http://svn.apache.org/repos/asf/sling/trunk/launchpad/test-services/src/main/java/org/apache/sling/launchpad/testservices/serversidetests/WriteableResourcesTest.java

     */

    @Before

    public void setup() throws Exception {

        resolver = new MockedResourceResolver();

        assertNotNull("Expecting non-null ResourceResolver", resolver);

        final Resource root = resolver.getResource("/");

        assertNotNull("Expecting non-null root Resource", root);

        final String path = getClass().getSimpleName() + "_" + System.currentTimeMillis();

        testRoot = resolver.create(root, path, null);

        resolver.commit();

        store = new MessageStoreImpl() {

            protected ResourceResolver getResourceResolver() {

                return resolver;

            }

        };

        store.threadKeyGen = new ThreadKeyGeneratorImpl();

        store.attachmentFilter = new AttachmentFilterImpl();

        store.archivePath = testRoot.getPath() + "/";

        store.resourceTypeKey = MessageStoreImplRepositoryTest.TEST_RT_KEY;

    }

    @After

    public void cleanup() throws Exception {

        resolver.close();

        resolver = null;

        testRoot = null;

        store = null;

    }

    @Test

    public void simpleMultipartMessageTest() throws IOException {

        Multipart multipart = new MultipartImpl("mixed");

        BodyPart att0 = createTextBody("This is the first part of the template..", "plain", true);

        multipart.addBodyPart(att0);

        BodyPart att1 = createRandomBinaryAttachment(200);

        multipart.addBodyPart(att1);

        BodyPart att2 = createRandomBinaryAttachment(300);

        multipart.addBodyPart(att2);

        BodyPart att3 = createTextBody("Some sample text here...?!", "html", true);

        multipart.addBodyPart(att3);

        BodyPart att4 = createRandomBinaryAttachment(100);

        multipart.addBodyPart(att4);

        BodyPart att5 = createTextBody("Some other text here...?!", "plain", true);

        multipart.addBodyPart(att5);

       

        MessageImpl message = new MessageImpl();

        message.setMultipart(multipart);

        message.setSubject("Template message");

        message.setDate(new Date());

        message.getHeader().setField(new RawField(LIST_ID, "<list.example.com>"));

        assertSaveMessageWithAttachments(message, 6);

    }

   

    @Test

    public void recursiveMultipartMessageTest() throws IOException {

        Multipart multipart = new MultipartImpl("mixed");

        BodyPart att1 = createRandomBinaryAttachment(100);

        multipart.addBodyPart(att1);

        BodyPart att2 = createRandomBinaryAttachment(133);

        multipart.addBodyPart(att2);

       

        Multipart nestedMultipart = new MultipartImpl("mixed");

        BodyPart nBody = createTextBody("Some sample text here...?!", "plain", false);

        nestedMultipart.addBodyPart(nBody);

        BodyPart nAtt1 = createRandomBinaryAttachment(300);

        nestedMultipart.addBodyPart(nAtt1);

        BodyPart NAtt2 = createRandomBinaryAttachment(100);

        nestedMultipart.addBodyPart(NAtt2);

        BodyPart nAtt3 = createTextBody("Some other text here...<br>?!", "html", true);

        nestedMultipart.addBodyPart(nAtt3);

       

        BodyPart nestedMessage = new BodyPart();

        nestedMessage.setMultipart(nestedMultipart);

        multipart.addBodyPart(nestedMessage);

        MessageImpl message = new MessageImpl();

        message.setMultipart(multipart);

        message.setSubject("Template message");

        message.setDate(new Date());

        message.getHeader().setField(new RawField(LIST_ID, "<list.example.com>"));

        assertSaveMessageWithAttachments(message, 5);

    }

    private void assertSaveMessageWithAttachments(Message msg, int num) throws IOException {

        store.save(msg);

        List<BodyPart> attList = new LinkedList<BodyPart>();

        MessageStoreImpl.recursiveMultipartProcessing((Multipart) msg.getBody(), new StringBuilder(), new StringBuilder(), false, attList);

        @SuppressWarnings("unchecked")

        Queue<BodyPart> attachmentsMsg = (Queue<BodyPart>) attList;

        assertTrue("No attachments found", attachmentsMsg.size() > 0);

        assertEquals("", num, attachmentsMsg.size());

       

        final Resource r = resolver.getResource(getResourcePath(msg, store));

        assertNotNull("Expecting non-null Resource", r);

        for (Resource aRes : r.getChildren()) {

            final ModifiableValueMap aMap = aRes.adaptTo(ModifiableValueMap.class);

            BodyPart aMsg = attachmentsMsg.poll();

            assertNotNull("JCR contains more attachments", aMsg);

            for (Field f : aMsg.getHeader().getFields()) {

                String name = f.getName();

                assertEquals("Field "+name+" is different", (aMap.get(name, String.class)), f.getBody());

            }

           

            if (aMsg.getBody() instanceof TextBody) {

                assertEquals("Content is not the same", MessageStoreImpl.getTextPart(aMsg), aMap.get(CONTENT, String.class));

            } else if (aMsg.getBody() instanceof BinaryBody) {

                assertEquals("Content is not the same", getBinPart(aMsg), aMap.get(CONTENT, String.class));

            } else {

                fail("Unknown type of attachment body");

            }

        }

        assertEquals("Message contains more attachments", attachmentsMsg.poll(), null);

    }

    private String getBinPart(Entity part) throws IOException {

        BinaryBody bb = (BinaryBody) part.getBody();

        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        bb.writeTo(baos);

        return new String(baos.toByteArray());

    }

    /*

     * taken from http://svn.apache.org/repos/asf/james/mime4j/trunk/examples/src/main/java/org/apache/james/mime4j/samples/transform/TransformMessage.java

     */

    private static BodyPart createTextBody(String text, String subtype, boolean isAttachment) {

        TextBody body = new StorageBodyFactory().textBody(text, MailArchiveServerConstants.DEFAULT_ENCODER.charset().name());

        BodyPart bodyPart = new BodyPart();

        if (isAttachment) {

            bodyPart.setContentDisposition("attachment", "file"+Math.random());

        }

        bodyPart.setText(body, subtype);

        return bodyPart;

    }

    /*

     * taken from http://svn.apache.org/repos/asf/james/mime4j/trunk/examples/src/main/java/org/apache/james/mime4j/samples/transform/TransformMessage.java

     */

    private static BodyPart createRandomBinaryAttachment(int numberOfBytes) throws IOException {

        byte[] data = new byte[numberOfBytes];

        new Random().nextBytes(data);

        Body body = new StorageBodyFactory().binaryBody(new ByteArrayInputStream(data));

        BodyPart bodyPart = new BodyPart();

        bodyPart.setContentDisposition("attachment", "file"+Math.random());

        bodyPart.setBody(body, "application/octet-stream");

        return bodyPart;

    }

}