Wake up before UI test

BaristaUtils.java

public final class BaristaUtils {

    public static void wakeUp() {

        InstrumentationRegistry.getInstrumentation().runOnMainSync(new Runnable() {
            @Override
            public void run() {
                Window window = new ComponentFinder().findCurrentActivity().getWindow();
                window.addFlags(WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
            }
        });
    }
}