Open Gmail inbox and get the unread mails count using Selenium Webdriver

Question

Anyone please suggest how to automate opening gmail inbox and getting a count of unread mails using Selenium Webdriver?

Abha · Answer 1 · May 21, 2019

Hi Naitik, you can follow this piece of code to achieve this type of automation using Selenium Webdriver:

System.setProperty("webdriver.chrome.driver", C:\\Users\\Abha_Rathour\\Downloads\\ExtractedFiles\\chromedriver_win32\\chromedriver.exe");

WebDriver driver = new ChromeDriver();

driver.manage().window().maximize();

driver.manage().timeouts().implicitlyWait(5, TimeUnit.SECONDS);

driver.get("https://gmail.com/");

driver.findElement(By.id("identifierId")).sendKeys("email_id");

driver.findElement(By.className("CwaK9")).click();   // Identifier for Next button

driver.findElement(By.xpath("//*[@id=\"password\"]/div[1]/div/div[1]/input")).sendKeys("password");

driver.findElement(By.className("CwaK9")).click();   // Identifier for Next button

// Get title of the current page
System.out.println("Title of Page: " + driver.getTitle());

// List of all Unread Emails
List<WebElement> unreademail = driver.findElements(By.className("zE"));

// Display count of unread emails
System.out.println("Total No. of Unread Mails: " + unreademail.size());

answered May 21, 2019 by Abha
• 28,140 points

public static void ReadGMailMessageBody(String username,String password) throws MessagingException, IOException{

try{

Properties properties = new Properties();

properties.setProperty("mail.store.protocol", "imaps");

Session emailSession = Session.getDefaultInstance(properties);

Store emailstore = emailSession.getStore("imaps");

emailstore.connect("smtp.gmail.com","xyz@gmail.com","password");

//getting Inbox Folder

System.out.println("Email connection done");

Folder emailFolder = emailstore.getFolder("INBOX");

emailFolder.open(Folder.READ_ONLY);

Message messages[] = emailFolder.getMessages();

String result = "";

for(int i = messages.length;i>=messages.length-4;i--){

System.out.println("Inbox box email count: "+ messages.length);

Message message = messages[i];

System.out.println(message.getSubject());

if( message.getSubject().contains("example:specific character from the email body")){

MimeMultipart mimeMultipart = (MimeMultipart) message.getContent();

result = getTextFromMimeMultipart(mimeMultipart);

System.out.println(result );

break;

}

}

emailFolder.close(false);

emailstore.close();

}

catch(NoSuchProviderException nspe){

}catch(MessagingException me){

me.printStackTrace();

}

}

private static String getTextFromMimeMultipart(

        MimeMultipart mimeMultipart) throws MessagingException, IOException{

    String result = "";

    int count = mimeMultipart.getCount();

    for (int i = 0; i < count; i++) {

        BodyPart bodyPart = mimeMultipart.getBodyPart(i);

        if (bodyPart.isMimeType("text/plain")) {

            result = result + "\n" + bodyPart.getContent();

            break; // without break same text appears twice in my tests

        } else if (bodyPart.isMimeType("text/html")) {

            String html = (String) bodyPart.getContent();

            result = result + "\n" + org.jsoup.Jsoup.parse(html).text();

        } else if (bodyPart.getContent() instanceof MimeMultipart){

            result = result + getTextFromMimeMultipart((MimeMultipart)bodyPart.getContent());

        }

    }

    return result;

}

public List<String> getUrlsFromMessage(Message message, String linkText) throws Exception{

    String html = getMessageContent(message);

    List<String> allMatches = new ArrayList<String>();

    Matcher matcher = Pattern.compile("(<a [^>]+>)"+linkText+"</a>").matcher(html);

    while (matcher.find()) {

      String aTag = matcher.group(1);

      allMatches.add(aTag.substring(aTag.indexOf("http"), aTag.indexOf("\">")));

    }

    return allMatches;

  }

public String getMessageContent(Message message) throws Exception {

    StringBuilder builder = new StringBuilder();

    BufferedReader reader = new BufferedReader(new InputStreamReader(message.getInputStream()));

    String line;

    while ((line = reader.readLine()) != null) {

      builder.append(line);

    }

    return builder.toString();

  }