How To Find Broken Links Using Selenium WebDriver

// Import Packages
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.Test;

public class BrokenLinksHomepage {
	WebDriver driver;

	@BeforeTest
	public void beforeMethod() throws Exception {
		System.setProperty("webdriver.chrome.driver", "<ChromeDriver Path>");

		driver = new ChromeDriver();

		// Enter url.
		driver.get("http://www.homepage.com/");

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

	@Test
	public void checkBrokenLinks() {
		List<WebElement> allLinks = findAllLinks(driver);

		System.out.println("Total number of elements found " + allLinks.size());

		// Identifying and Validating URL
		for (WebElement element : allLinks) {
			try {
				System.out.println("URL: " + element.getAttribute("href") + " returned " + isLinkBroken(new URL(element.getAttribute("href"))));
			}

			catch (Exception exp) {
				System.out.println("At " + element.getAttribute("innerHTML") + "Exception occured -&gt; " + exp.getMessage());
			}
		}
	}

	public static List<WebElement> findAllLinks(WebDriver driver) {

		// Finding all the links on homepage
		List<WebElement> elementList = new ArrayList<WebElement>();

		elementList = driver.findElements(By.tagName("a"));

		elementList.addAll(driver.findElements(By.tagName("img")));

		List<WebElement> finalList = new ArrayList<WebElement>();

		for (WebElement element : elementList) {
			if (element.getAttribute("href") != null) {
				finalList.add(element);
			}
		}
		return finalList;
	}

	public static int isLinkBroken(URL url) throws Exception {
		// url = new URL("http://homepage.com");

		int response;

		// Send HTTP Request
		HttpURLConnection connection = (HttpURLConnection) url.openConnection();

		// Validating Links
		try {
			connection.connect();

			response = connection.getResponseCode();

			connection.disconnect();

			return response;
		}

		catch (Exception exp) {
			System.out.println("Exception occured" + exp.getMessage());
			return 404;
		}
	}
}