How to use marshaller method of com.consol.citrus.mail.message.MailMessage class

Best Citrus code snippet using com.consol.citrus.mail.message.MailMessage.marshaller

Source:MailServer.java

...49public class MailServer extends AbstractServer implements SimpleMessageListener, InitializingBean {50    /** Server port */51    private int port = 25;52    /** XML message mapper */53    private MailMarshaller marshaller = new MailMarshaller();54    /** Mail message converter */55    private MailMessageConverter messageConverter = new MailMessageConverter();56    /** Java mail session */57    private Session mailSession;58    /** Java mail properties */59    private Properties javaMailProperties = new Properties();60    /** Should accept automatically or handled via test case */61    private boolean autoAccept = true;62    /** Should split multipart messages for each mime part */63    private boolean splitMultipart = false;64    /** Smtp server instance */65    private SMTPServer smtpServer;66    @Override67    protected void startup() {68        smtpServer = new SMTPServer(new SimpleMessageListenerAdapter(this));69        smtpServer.setSoftwareName(getName());70        smtpServer.setPort(port);71        smtpServer.start();72    }73    @Override74    protected void shutdown() {75        smtpServer.stop();76    }77    @Override78    public boolean accept(String from, String recipient) {79        if (autoAccept) {80            return true;81        }82        Message response = getEndpointAdapter().handleMessage(83                MailMessage.accept(from, recipient)84                           .marshaller(marshaller));85        if (response == null || response.getPayload() == null) {86            throw new CitrusRuntimeException("Did not receive accept response. Missing accept response because autoAccept is disabled.");87        }88        AcceptResponse acceptResponse = null;89        if (response.getPayload() instanceof AcceptResponse) {90            acceptResponse = (AcceptResponse) response.getPayload();91        } else if (response.getPayload() instanceof String) {92            acceptResponse = (AcceptResponse) marshaller.unmarshal(response.getPayload(Source.class));93        }94        if (acceptResponse == null) {95            throw new CitrusRuntimeException("Unable to read accept response from payload: " + response);96        }97        return acceptResponse.isAccept();98    }99    @Override100    public void deliver(String from, String recipient, InputStream data) {101        try {102            MimeMailMessage mimeMailMessage = new MimeMailMessage(new MimeMessage(getSession(), data));103            MailMessage request = messageConverter.convertInbound(mimeMailMessage, getEndpointConfiguration(), null);104            Message response = invokeEndpointAdapter(request);105            if (response != null && response.getPayload() != null) {106                MailResponse mailResponse = null;107                if (response.getPayload() instanceof MailResponse) {108                    mailResponse = (MailResponse) response.getPayload();109                } else if (response.getPayload() instanceof String) {110                    mailResponse = (MailResponse) marshaller.unmarshal(response.getPayload(Source.class));111                }112                if (mailResponse != null && mailResponse.getCode() != MailResponse.OK_CODE) {113                    throw new RejectException(mailResponse.getCode(), mailResponse.getMessage());114                }115            }116        } catch (MessagingException e) {117            throw new CitrusRuntimeException(e);118        }119    }120    /**121     * Invokes the endpoint adapter with constructed mail message and headers.122     * @param mail123     */124    protected Message invokeEndpointAdapter(MailMessage mail) {125        if (splitMultipart) {126            return split(mail.getPayload(MailRequest.class).getBody(), mail.getHeaders());127        } else {128            return getEndpointAdapter().handleMessage(mail);129        }130    }131    /**132     * Split mail message into several messages. Each body and each attachment results in separate message133     * invoked on endpoint adapter. Mail message response if any should be sent only once within test case.134     * However latest mail response sent by test case is returned, others are ignored.135     *136     * @param bodyPart137     * @param messageHeaders138     */139    private Message split(BodyPart bodyPart, Map<String, Object> messageHeaders) {140        MailMessage mailRequest = createMailMessage(messageHeaders, bodyPart.getContent(), bodyPart.getContentType());141        Stack<Message> responseStack = new Stack<>();142        if (bodyPart instanceof AttachmentPart) {143            fillStack(getEndpointAdapter().handleMessage(mailRequest144                    .setHeader(CitrusMailMessageHeaders.MAIL_CONTENT_TYPE, bodyPart.getContentType())145                    .setHeader(CitrusMailMessageHeaders.MAIL_FILENAME, ((AttachmentPart) bodyPart).getFileName())), responseStack);146        } else {147            fillStack(getEndpointAdapter().handleMessage(mailRequest148                    .setHeader(CitrusMailMessageHeaders.MAIL_CONTENT_TYPE, bodyPart.getContentType())), responseStack);149        }150        if (bodyPart.hasAttachments()) {151            for (AttachmentPart attachmentPart : bodyPart.getAttachments().getAttachments()) {152                fillStack(split(attachmentPart, messageHeaders), responseStack);153            }154        }155        return responseStack.isEmpty() ? null : responseStack.pop();156    }157    private void fillStack(Message message, Stack<Message> responseStack) {158        if (message != null) {159            responseStack.push(message);160        }161    }162    /**163     * Creates a new mail message model object from message headers.164     * @param messageHeaders165     * @param body166     * @param contentType167     * @return168     */169    protected MailMessage createMailMessage(Map<String, Object> messageHeaders, String body, String contentType) {170        return MailMessage.request(messageHeaders)171                .marshaller(marshaller)172                .from(messageHeaders.get(CitrusMailMessageHeaders.MAIL_FROM).toString())173                .to(messageHeaders.get(CitrusMailMessageHeaders.MAIL_TO).toString())174                .cc(messageHeaders.get(CitrusMailMessageHeaders.MAIL_CC).toString())175                .bcc(messageHeaders.get(CitrusMailMessageHeaders.MAIL_BCC).toString())176                .subject(messageHeaders.get(CitrusMailMessageHeaders.MAIL_SUBJECT).toString())177                .body(body, contentType);178    }179    @Override180    public MailEndpointConfiguration getEndpointConfiguration() {181        MailEndpointConfiguration endpointConfiguration = new MailEndpointConfiguration();182        endpointConfiguration.setMessageConverter(messageConverter);183        endpointConfiguration.setMarshaller(marshaller);184        endpointConfiguration.setJavaMailProperties(javaMailProperties);185        return endpointConfiguration;186    }187    /**188     * Return new mail session if not already created before.189     * @return190     */191    public synchronized Session getSession() {192        if (this.mailSession == null) {193            this.mailSession = Session.getInstance(this.javaMailProperties);194        }195        return this.mailSession;196    }197    /**198     * Is auto accept enabled.199     * @return200     */201    public boolean isAutoAccept() {202        return autoAccept;203    }204    /**205     * Enable/disable auto accept feature.206     * @param autoAccept207     */208    public void setAutoAccept(boolean autoAccept) {209        this.autoAccept = autoAccept;210    }211    /**212     * Gets the mail message marshaller.213     * @return214     */215    public MailMarshaller getMarshaller() {216        return marshaller;217    }218    /**219     * Sets the mail message marshaller.220     * @param marshaller221     */222    public void setMarshaller(MailMarshaller marshaller) {223        this.marshaller = marshaller;224    }225    /**226     * Gets the Java mail properties.227     * @return228     */229    public Properties getJavaMailProperties() {230        return javaMailProperties;231    }232    /**233     * Sets the Java mail properties.234     * @param javaMailProperties235     */236    public void setJavaMailProperties(Properties javaMailProperties) {237        this.javaMailProperties = javaMailProperties;...

Source:MailMessage.java

...23 * @author Christoph Deppisch24 * @since 2.725 */26public class MailMessage extends DefaultMessage {27    private MailMarshaller marshaller = new MailMarshaller();28    private MailRequest mailRequest;29    private MailResponse mailResponse;30    private AcceptRequest acceptRequest;31    private AcceptResponse acceptResponse;32    /**33     * Prevent traditional instantiation.34     */35    private MailMessage() { super(); }36    /**37     * Constructor initializes new request message.38     * @param mailRequest39     */40    private MailMessage(MailRequest mailRequest) {41        super(mailRequest);42        this.mailRequest = mailRequest;43    }44    /**45     * Constructor initializes new response message.46     * @param mailResponse47     */48    private MailMessage(MailResponse mailResponse) {49        super(mailResponse);50        this.mailResponse = mailResponse;51    }52    /**53     * Constructor initializes new request message.54     * @param acceptRequest55     */56    private MailMessage(AcceptRequest acceptRequest) {57        super(acceptRequest);58        this.acceptRequest = acceptRequest;59    }60    /**61     * Constructor initializes new response message.62     * @param acceptResponse63     */64    private MailMessage(AcceptResponse acceptResponse) {65        super(acceptResponse);66        this.acceptResponse = acceptResponse;67    }68    /**69     * Constructor initializes new request message.70     * @param mailRequest71     * @param headers72     */73    private MailMessage(MailRequest mailRequest, Map<String, Object> headers) {74        super(mailRequest, headers);75        this.mailRequest = mailRequest;76    }77    public static MailMessage request(Map<String, Object> headers) {78        MailRequest request = new MailRequest();79        if (headers.containsKey(CitrusMailMessageHeaders.MAIL_FROM)) {80            request.setFrom(headers.get(CitrusMailMessageHeaders.MAIL_FROM).toString());81        }82        if (headers.containsKey(CitrusMailMessageHeaders.MAIL_TO)) {83            request.setTo(headers.get(CitrusMailMessageHeaders.MAIL_TO).toString());84        }85        if (headers.containsKey(CitrusMailMessageHeaders.MAIL_CC)) {86            request.setCc(headers.get(CitrusMailMessageHeaders.MAIL_CC).toString());87        }88        if (headers.containsKey(CitrusMailMessageHeaders.MAIL_BCC)) {89            request.setBcc(headers.get(CitrusMailMessageHeaders.MAIL_BCC).toString());90        }91        if (headers.containsKey(CitrusMailMessageHeaders.MAIL_SUBJECT)) {92            request.setSubject(headers.get(CitrusMailMessageHeaders.MAIL_SUBJECT).toString());93        }94        return new MailMessage(request, headers);95    }96    public static MailMessage request() {97        return request(null, null, null);98    }99    public static MailMessage request(String from) {100        return request(from, null, null);101    }102    public static MailMessage request(String from, String to) {103        return request(from, to, null);104    }105    public static MailMessage request(String from, String to, String subject) {106        MailRequest request = new MailRequest();107        request.setFrom(from);108        request.setTo(to);109        request.setSubject(subject);110        return new MailMessage(request);111    }112    public static MailMessage response() {113        return response(250);114    }115    public static MailMessage response(int code) {116        return response(code, "OK");117    }118    public static MailMessage response(int code, String message) {119        MailResponse response = new MailResponse();120        response.setCode(code);121        response.setMessage(message);122        return new MailMessage(response);123    }124    public static MailMessage accept(String from, String to) {125        AcceptRequest accept = new AcceptRequest();126        accept.setFrom(from);127        accept.setTo(to);128        return new MailMessage(accept);129    }130    public static MailMessage accept() {131        return accept(true);132    }133    public static MailMessage accept(boolean result) {134        AcceptResponse acceptResponse = new AcceptResponse();135        acceptResponse.setAccept(result);136        return new MailMessage(acceptResponse);137    }138    public MailMessage subject(String subject) {139        if (mailRequest != null) {140            mailRequest.setSubject(subject);141        } else {142            throw new CitrusRuntimeException("Invalid access to method 'subject' for mail message");143        }144        return this;145    }146    public MailMessage from(String from) {147        if (mailRequest == null && acceptRequest == null) {148            throw new CitrusRuntimeException("Invalid access to method 'from' for mail message");149        }150        if (mailRequest != null) {151            mailRequest.setFrom(from);152        }153        if (acceptRequest != null) {154            acceptRequest.setFrom(from);155        }156        return this;157    }158    public MailMessage to(String to) {159        if (mailRequest == null && acceptRequest == null) {160            throw new CitrusRuntimeException("Invalid access to method 'to' for mail message");161        }162        if (mailRequest != null) {163            mailRequest.setTo(to);164        }165        if (acceptRequest != null) {166            acceptRequest.setTo(to);167        }168        return this;169    }170    public MailMessage cc(String cc) {171        if (mailRequest != null) {172            mailRequest.setCc(cc);173        } else {174            throw new CitrusRuntimeException("Invalid access to method 'cc' for mail message");175        }176        return this;177    }178    public MailMessage bcc(String bcc) {179        if (mailRequest != null) {180            mailRequest.setBcc(bcc);181        } else {182            throw new CitrusRuntimeException("Invalid access to method 'bcc' for mail message");183        }184        return this;185    }186    public MailMessage body(BodyPart bodyPart) {187        if (mailRequest != null) {188            mailRequest.setBody(bodyPart);189        } else {190            throw new CitrusRuntimeException("Invalid access to method 'subject' for mail message");191        }192        return this;193    }194    public MailMessage body(String body) {195        return body(body, "text/plain");196    }197    public MailMessage body(String body, String contentType) {198        if (mailRequest != null) {199            mailRequest.setBody(new BodyPart(body, contentType));200        } else {201            throw new CitrusRuntimeException("Invalid access to method 'subject' for mail message");202        }203        return this;204    }205    public MailMessage marshaller(MailMarshaller mailMarshaller) {206        this.marshaller = mailMarshaller;207        return this;208    }209    @Override210    public <T> T getPayload(Class<T> type) {211        if (String.class.equals(type)) {212            return (T) getPayload();213        } else if (MailRequest.class.equals(type) && mailRequest != null) {214            return (T) mailRequest;215        } else if (MailResponse.class.equals(type) && mailResponse != null) {216            return (T) mailResponse;217        } else if (AcceptRequest.class.equals(type) && acceptRequest != null) {218            return (T) acceptRequest;219        } else if (AcceptResponse.class.equals(type) && acceptResponse != null) {220            return (T) acceptResponse;221        }222        return super.getPayload(type);223    }224    @Override225    public Object getPayload() {226        StringResult payloadResult = new StringResult();227        if (mailRequest != null) {228            marshaller.marshal(mailRequest, payloadResult);229            return payloadResult.toString();230        } else if (mailResponse != null) {231            marshaller.marshal(mailResponse, payloadResult);232            return payloadResult.toString();233        } else if (acceptRequest != null) {234            marshaller.marshal(acceptRequest, payloadResult);235            return payloadResult.toString();236        } else if (acceptResponse != null) {237            marshaller.marshal(acceptResponse, payloadResult);238            return payloadResult.toString();239        }240        return super.getPayload();241    }242    /**243     * Gets the marshaller.244     *245     * @return246     */247    public MailMarshaller getMarshaller() {248        return marshaller;249    }250    /**251     * Sets the marshaller.252     *253     * @param marshaller254     */255    public void setMarshaller(MailMarshaller marshaller) {256        this.marshaller = marshaller;257    }258}...

marshaller

Using AI Code Generation

1package com.consol.citrus.mail.message;2import com.consol.citrus.message.Message;3import com.consol.citrus.message.MessageType;4import com.consol.citrus.message.MessageTypeResolver;5import com.consol.citrus.message.MessageTypeResolverChain;6import com.consol.cit

marshaller

Using AI Code Generation

1import com.consol.citrus.mail.message.MailMessage;2import com.consol.citrus.message.MessageType;3import com.consol.citrus.message.builder.MarshallingPayloadBuilder;4import com.consol.citrus.message.builder.PayloadTemplateMessageBuilder;5import com.consol.citrus.message.builder.TemplateMessageBuilder;6import com.consol.citrus.message.builder.TextPayloadBuilder;7import com.consol.citrus.message.builder.XmlMessagePayloadBuilder;8import com.consol.citrus.message.builder.XmlPayloadTemplateMessageBuilder;9import com.consol.ci

marshaller

Using AI Code Generation

1package com.consol.citrus.samples;2import java.io.File;3import java.io.IOException;4import java.util.Arrays;5import org.springframework.beans.factory.annotation.Autowired;6import org.springframework.beans.factory.annotation.Qualifier;7import org.springframework.context.annotation.Bean;8import org.springframework.context.annotation.Configuration;9import org.springframework.core.io.FileSystemResource;10import org.springframework.mail.javamail.JavaMailSender;11import org.springframework.mail.javamail.JavaMailSenderImpl;12import org.springframework.util.ResourceUtils;13import com.consol.citrus.dsl.builder.BuilderSupport;14import com.consol.citrus.dsl.builder.SendMailActionBuilder;15import com.consol.citrus.dsl.builder.SendSoapMessageActionBuilder;16import com.consol.citrus.dsl.runner.TestRunner;17import com.consol.citrus.dsl.runner.TestRunnerSupport;18import com.consol.citrus.mail.message.MailMessage;19public class MailClientConfig {20    public JavaMailSender mailSender() {21        JavaMailSenderImpl mailSender = new JavaMailSenderImpl();22        mailSender.setHost("localhost");23        mailSender.setPort(3025);24        return mailSender;25    }26    public BuilderSupport<SendMailActionBuilder> sendMail() {27                .message(mailMessage());28    }29    public MailMessage mailMessage() {30        MailMessage mailMessage = new MailMessage();31        mailMessage.setFrom("test@localhost");32        mailMessage.setTo("test@localhost");33        mailMessage.setCc("test@localhost");34        mailMessage.setBcc("test@localhost");35        mailMessage.setSubject("Test subject");36        mailMessage.setText("Test text");37        mailMessage.setAttachments(Arrays.asList(new FileSystemResource("classpath:com/consol/citrus/samples/citrus-logo.png")));38        return mailMessage;39    }40}41package com.consol.citrus.samples;42import java.io.File;43import java.io.IOException;44import java.util.Arrays;45import org.springframework.beans.factory.annotation.Autowired;46import org.springframework.beans.factory.annotation.Qualifier;47import org.springframework.context.annotation.Bean;48import org.springframework.context.annotation.Configuration;49import org.springframework.core.io.FileSystemResource;50import org.springframework.mail.javamail.JavaMailSender;51import

marshaller

Using AI Code Generation

1package com.consol.citrus.mail.message;2import org.springframework.mail.SimpleMailMessage;3import org.springframework.mail.javamail.JavaMailSenderImpl;4import com.consol.citrus.mail.message.MailMessage;5import com.consol.citrus.mail.message.MailMessageMarshaller;6public class MailMessageMarshallerTest {7public static void main(String[] args) {8MailMessageMarshaller mar = new MailMessageMarshaller();9JavaMailSenderImpl mailSender = new JavaMailSenderImpl();10mailSender.setHost("localhost");11mailSender.setPort(25);12mailSender.setUsername("test");13mailSender.setPassword("test");14SimpleMailMessage simpleMessage = new SimpleMailMessage();15simpleMessage.setFrom("

marshaller

Using AI Code Generation

1package com.consol.citrus.mail;2import java.io.IOException;3import java.io.InputStream;4import java.io.OutputStream;5import java.io.Reader;6import java.io.Writer;7import java.util.Properties;8import javax.mail.Message;9import javax.mail.MessagingException;10import javax.mail.Session;11import javax.mail.internet.MimeMessage;12import org.springframework.core.io.Resource;13import org.springframework.mail.javamail.MimeMessageHelper;14import org.springframework.util.Assert;15import org.springframework.util.FileCopyUtils;16import org.springframework.util.StringUtils;17import com.consol.citrus.exceptions.CitrusRuntimeException;18import com.consol.citrus.message.DefaultMessage;19import com.consol.citrus.message.MessageHeaders;20import com.consol.citrus.message.MessageType;21public class MailMessage extends DefaultMessage {22    private MimeMessage mimeMessage;23    public MailMessage() {24        super();25    }26    public MailMessage(MimeMessage mimeMessage) {27        super(mimeMessage);28        this.mimeMessage = mimeMessage;29    }30    public MailMessage(String payload) {31        super(payload);32    }33    public MailMessage(String payload, MessageHeaders headers) {34        super(payload, headers);35    }36    public void setPayload(Object payload) {37        if (payload instanceof MimeMessage) {38            this.mimeMessage = (MimeMessage) payload;39        } else if (payload instanceof Resource) {40            try {41                this.mimeMessage = new MimeMessage(Session.getInstance(new Properties()), ((Resource) payload).getInputStream());42            } catch (IOException e) {43                throw new CitrusRuntimeException("Failed to read mail message payload from resource", e);44            } catch (MessagingException e) {45                throw new CitrusRuntimeException("Failed to read mail message payload from resource", e);46            }47        } else {48            throw new CitrusRuntimeException("Unsupported payload type for mail message. Only MimeMessage and Resource payload types are supported

marshaller

Using AI Code Generation

1package com.consol.citrus.mail;2import java.util.Properties;3import com.consol.citrus.mail.message.MailMessage;4import com.consol.citrus.message.Message;5import com.consol.citrus.message.MessageType;6import com.consol.citrus.message.MessageTypeResolver;7import com.consol.citrus.message.MessageTypeResolverChain;8public class MailMessageTypeResolver implements MessageTypeResolver {9    private MailMessageMarshaller marshaller = new MailMessageMarshaller();10    private MailMessageUnmarshaller unmarshaller = new MailMessageUnmarshaller();11    public MessageType resolveMessageType(Message message) {12        if (message.getPayload() instanceof String) {13            return MessageType.PLAINTEXT;14        } else if (message.getPayload() instanceof MailMessage) {15            return MessageType.XML;16        } else {17            return MessageType.UNKNOWN;18        }19    }20    public Message marshallMessage(Message message, Properties headers) {21        if (message.getPayload() instanceof String) {22            return message;23        } else if (message.getPayload() instanceof MailMessage) {24            return marshaller.marshall(message, headers);25        } else {26            return message;27        }28    }29    public Message unmarshallMessage(Message message, Properties headers) {30        if (message.getPayload() instanceof String) {31            return unmarshaller.unmarshall(message, headers);32        } else if (message.getPayload() instanceof MailMessage) {33            return message;34        } else {35            return message;36        }37    }38    public void setNext(MessageTypeResolver resolver) {39        if (resolver instanceof MessageTypeResolverChain) {40            ((MessageTypeResolver

marshaller

Using AI Code Generation

1package com.consol.citrus.mail.actions;2import com.consol.citrus.context.TestContext;3import com.consol.citrus.mail.message.MailMessage;4import com.consol.citrus.message.Message;5import com.consol.citrus.message.MessageType;6import com.consol.citrus.message.builder.ObjectPayloadMessageBuilder;7import com.consol.citrus.message.builder.PayloadTemplateMessageBuilder;8import com.consol.citrus.message.builder.ScriptMessageBuilder;9import com.consol.citrus.message.builder.StaticMessageContentBuilder;10import com.consol.citrus.message.builder.TemplateMessageBuilder;11import com.consol.citrus.message.builder.TextMessageBuilder;12import com.consol.citrus.message.builder.VariableMessageBuilder;13import com.consol.citrus.message.builder.XMLMessageBuilder;14import com.consol.citrus.message.builder.XpathMessageBuilder;15import com.consol.citrus.message.builder.XpathPayloadTemplateMessageBuilder;16import com.consol.citrus.message.builder.XpathResultMessageBuilder;17import com.consol.citrus.message.builder.XpathResultPayloadTemplateMessageBuilder;18import com.consol.citrus.message.builder.XpathSelectMessageBuilder;19import com.consol.citrus.message.builder.XpathSelectPayloadTemplateMessageBuilder;20import com.consol.citrus.message.builder.XpathVariableMessageBuilder;21import com.consol.citrus.message.builder.XpathVariablePayloadTemplateMessageBuilder;22import com.consol.citrus.message.builder.XsdMessageBuilder;23import com.consol.citrus.message.builder.XsdPayloadTemplateMessageBuilder;24import com.consol.citrus.message.builder.XsdSchemaRepository;25import com.consol.citrus.message.builder.XsltMessageBuilder;26import com.consol.citrus.message.builder.XsltPayloadTemplateMessageBuilder;27import com.consol.citrus.message.builder.XsltSchemaRepository;28import com.consol.citrus.message.builder.XsltTransformationRepository;29import com.consol.citrus.message.builder.XsltVariables;30import com.consol.citrus.message.builder.XsltVariables.XsltVariable;31import com.consol.citrus.message.builder.XsltVariables.XsltVariable.Builder;32import com.consol.citrus.message.builder.XsltVariables.XsltVariable.Type;33import com.consol.citrus.message.builder.XsltVariables.XsltVariable.Value;34import com.consol.cit

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.