Verificando pdf usando Selenium + Java

Como verificar o pdf usando Selenium + Java?

Até recentemente, eu verificava o pdf usando o Firefox (certificando-se de que o arquivo pdf não estava vazio). Isso foi bom e funcionou bem, mas não é multiplataforma.

Agora preciso verificar o pdf usando chrome, edge, safari, etc ... Portanto, fazer algo específico de plataforma não é mais uma opção.

Aqui está o que eu estava fazendo até agora usando o Firefox:

• Clique no link do pdf

• Espere a nova janela aparecer

• Mudar para a nova janela

• Verifique a presença da classe "page" e do id "viewer"

• Feche a nova janela e volte para a antiga

No entanto, a classe "page" e o id "viewer" são específicos de como o Firefox exibe o pdf e este mesmo código não funciona com o Chrome.

Então, como posso verificar um pdf usando Selenium + Java?

Depois de muito pensar e pesquisar, aqui está a solução que finalmente encontrei:

1. Baixe o pdf

2. Verifique o pdf usando Java

Verificar o pdf usando Java foi a parte fácil (já que era, por padrão, multiplataforma). Aqui está o site que me ajudou a fazer isso:https://www.mkyong.com/java/pdfbox-how-to-read-pdf-file-in-java/

Baixar o pdf em uma forma multiplataforma era mais complicado. Principalmente porque não pude usar a classe "Robot" porque meu driver estava executando remotamente (eu não tentei essa solução, mas li que não funcionaria no meu caso específico, então não me arrisquei e pesquisar outra maneira de fazer o que eu queria).

Finalmente encontrei esta resposta que me conduziu à resolução do meu problema.

Então, depois de tudo isso, aqui está (parte do) meu código:

pom.xml:

<!-- https://mvnrepository.com/artifact/org.apache.pdfbox/pdfbox -->
<dependency>
    <groupId>org.apache.pdfbox</groupId>
    <artifactId>pdfbox</artifactId>
    <version>2.0.17</version>
</dependency>

Java:

import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.text.PDFTextStripper;
import org.apache.pdfbox.text.PDFTextStripperByArea;
import java.io.InputStream;
import java.net.URI;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.nio.file.StandardCopyOption;


public class MyClass {
    private WebDriver driver;
    private String parentHandle;


    //My entry point
    public void iVerifyThatPdfLinksAreProperlyDisplayed() {
        List<WebElement> pdfLinks = driver.findElements(By.xpath("//a[contains(@href,'pdf')]"));

        for (WebElement pdfLink : pdfLinks) {
            pdfLink.click();
            switchToNewWindow();
            verifyPdf();
            goBackToOldWindow();
        }
    }

    private void switchToNewWindow() {
        parentHandle = driver.getWindowHandle(); // get the current window handle

        WebDriverWait wait = new WebDriverWait(driver, 5);
        wait.until(ExpectedConditions.numberOfWindowsToBe(2));

        for (String winHandle : driver.getWindowHandles()) {
            if (!winHandle.equals(parentHandle)){
                driver.switchTo().window(winHandle); // switch focus of WebDriver to the next found window handle (that's your newly opened window)
            }
        }
    }

    private void goBackToOldWindow(){
        if (!driver.getWindowHandle().equals(parentHandle)){
            driver.close(); // close newly opened window when done with it
        }
        driver.switchTo().window(parentHandle); // switch back to the original window
    }

    private void verifyPdf(){
        WebDriverWait wait = new WebDriverWait(driver, 10);
        wait.until(ExpectedConditions.not(ExpectedConditions.urlContains("about:blank")));

        System.out.println(driver.getCurrentUrl());

        try {
            ClassWhereIPutStaticStuff.download(driver.getCurrentUrl(), "target"+ File.separator +"my_pdf.pdf");
            ClassWhereIPutStaticStuff.readPDF("target"+ File.separator +"my_pdf.pdf", "");
        } catch (Exception e) {
            e.printStackTrace();
            fail();
        }
    }
}

////////////////////////////////////////

public class ClassWhereIPutStaticStuff {

    public static void download(String url, String fileName) throws Exception {
        try (InputStream in = URI.create(url).toURL().openStream()) {
            Files.copy(in, Paths.get(fileName), StandardCopyOption.REPLACE_EXISTING);
        }
    }

    public static void readPDF(String fileName, String textWeWant) throws IOException {
        try (PDDocument document = PDDocument.load(new File(fileName))) {

            if (!document.isEncrypted()) {
                PDFTextStripperByArea stripper = new PDFTextStripperByArea();
                stripper.setSortByPosition(true);

                PDFTextStripper tStripper = new PDFTextStripper();

                String pdfFileInText = tStripper.getText(document);

                // split by whitespace
                String lines[] = pdfFileInText.split("\\r?\\n");
                for (String line : lines) {
                    System.out.println(line);
                }

                pdfFileInText = pdfFileInText.toLowerCase();

                assertNotNull(pdfFileInText);
                assertFalse(pdfFileInText.isEmpty());
                assertTrue(pdfFileInText.length() > 10);
                assertTrue(pdfFileInText.contains(textWeWant.toLowerCase()));
            } else {
                fail("file is encrypted!");
            }
        }
    }
}