com.automation.remarks.junit.VideoRule Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of video-recorder-junit Show documentation

Video Recorder is a library that allows to record selenium tests

There is a newer version: 2.0

package com.automation.remarks.junit;

import com.automation.remarks.video.annotations.Video;
import com.automation.remarks.video.recorder.VideoRecorder;
import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;

import java.io.File;
import java.util.LinkedList;

import static com.automation.remarks.video.RecordingMode.ALL;
import static com.automation.remarks.video.RecordingUtils.doVideoProcessing;
import static com.automation.remarks.video.VideoConfiguration.MODE;


/**
 * Created by sergey on 4/21/16.
 */
public class VideoRule implements TestRule {

    @Override
    public Statement apply(final Statement base, final Description description) {
        return new Statement() {
            @Override
            public void evaluate() throws Throwable {
                Video video = description.getAnnotation(Video.class);
                String name = getFileName(video, description);
                if (MODE.equals(ALL)) {
                    recordVideo(name, base);
                } else if (video != null && video.enabled()) {
                    recordVideo(name, base);
                } else {
                    base.evaluate();
                }
            }

            private void recordVideo(String name, final Statement base) throws Throwable {
                boolean successful = false;
                VideoRecorder recorder = new VideoRecorder(name);
                recorder.start();
                try {
                    base.evaluate();
                    successful = true;
                } finally {
                    LinkedList files = recorder.stop();
                    if (description.isTest()) {
                        doVideoProcessing(successful, files);
                    }
                }
            }

            private String getFileName(Video video, Description description) {
                String methodName = description.getMethodName();
                if (video == null) {
                    return methodName;
                }
                String videoName = video.name();
                return videoName.length() > 1 ? videoName : methodName;
            }
        };
    }
}