@Attachment(value = "{name}", type = "image/png")
public static byte[] attachScreenshot(final String name, final byte[] data) {
    return data;
}

public static boolean hasDiff(final Screenshot actual, final Screenshot expected) {
    final ImageDiff diff = new ImageDiffer().makeDiff(expected, actual);
    final boolean hasDiff = diff.hasDiff();

    if (hasDiff) {
        addLabels(new Label().withName("testType").withValue("screenshotDiff"));
        attachScreenshot("actual", toByteArray(actual.getImage()));
        attachScreenshot("expected", toByteArray(expected.getImage()));
        attachScreenshot("diff", toByteArray(diff.getDiffImage()));
    }

    return hasDiff;
}

private static byte[] toByteArray(final BufferedImage image) {
    try (final ByteArrayOutputStream out = new ByteArrayOutputStream()) {
        ImageIO.write(image, "png", out);
        return out.toByteArray();
    } catch (IOException ignored) {
        return new byte[0];
    }
}

@Test
public void checkDiff() {
    assertThat(isSimilar("feature1", scr1, scr2)).isTrue();
    assertThat(isSimilar("feature2", scr3, scr4)).isTrue();
}

@Step("Compare screenshots for {name}.")
public boolean isSimilar(final String name, final Screenshot scr1, final Screenshot scr2) {
    return !hasDiff(scr1, scr2);
}