ru.yandex.qatools.allure.events.MakeAttachEvent Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of allure-java-commons Show documentation

There is a newer version: 1.4.0.RC9

package ru.yandex.qatools.allure.events;

import ru.yandex.qatools.allure.config.AllureConfig;
import ru.yandex.qatools.allure.exceptions.AllureException;
import ru.yandex.qatools.allure.model.Attachment;
import ru.yandex.qatools.allure.model.AttachmentType;
import ru.yandex.qatools.allure.model.Step;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;

import static ru.yandex.qatools.allure.utils.AllureResultsUtils.writeAttachment;
import static ru.yandex.qatools.allure.utils.AllureResultsUtils.writeAttachmentWithErrorMessage;

/**
 * @author Dmitry Baev [email protected]
 *         Date: 11.11.13
 * @deprecated since 1.4
 */
@Deprecated
public class MakeAttachEvent extends AbstractMakeAttachEvent {

    public MakeAttachEvent(String title, AttachmentType attachmentType, Object attach) {
        setTitle(title);
        setAttachmentType(attachmentType);
        setAttach(attach);
    }

    @Override
    public void process(Step step) {
        Attachment attachment = writeSafely(getAttach(), getTitle());

        step.getAttachments().add(attachment);
    }

    private static Attachment write(Object attachment, String title) throws IOException {
        if (attachment instanceof String) {
            return writeAttachment(
                    ((String) attachment).getBytes(AllureConfig.newInstance().getAttachmentsEncoding()),
                    title
            );
        } else if (attachment instanceof File) {
            byte[] bytes = Files.readAllBytes(Paths.get(((File) attachment).toURI()));
            return writeAttachment(bytes, title);
        }
        throw new AllureException("Attach-method should be return 'java.lang.String' or 'java.io.File'.");
    }

    private static Attachment writeSafely(Object attachment, String title) {
        try {
            return write(attachment, title);
        } catch (Exception e) {
            return writeAttachmentWithErrorMessage(e, title);
        }
    }
}