/**
* Copyright (C) 2012-2014 Gist Labs, LLC. (http://gistlabs.com)
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*/
package com.gistlabs.mechanize.document.html;
import static com.gistlabs.mechanize.util.css.CSSHelper.byIdOrClass;
import static com.gistlabs.mechanize.util.css.CSSHelper.contains;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import org.junit.Test;
import com.gistlabs.mechanize.MechanizeTestCase;
import com.gistlabs.mechanize.PageRequest;
import com.gistlabs.mechanize.Resource;
import com.gistlabs.mechanize.document.AbstractDocument;
import com.gistlabs.mechanize.document.link.Link;
import com.gistlabs.mechanize.util.apache.ContentType;
/**
* @author Martin Kersten<Martin.Kersten.mk@gmail.com>
*/
public class LinksTest extends MechanizeTestCase {
protected String contentType() {
return ContentType.TEXT_HTML.getMimeType();
}
@Test
public void testFollowingAnAbsoluteLink() {
addPageRequest("http://test.com",
newHtml("Test Page", "<a id=\"t\" href=\"http://test.com/myPage.html\">myPage</a>"));
addPageRequest("http://test.com/myPage.html", newHtml("My Page", ""));
AbstractDocument page = agent().get("http://test.com");
assertEquals(1, page.links().size());
Link link = page.links().find(byIdOrClass("t"));
assertNotNull(link);
Resource myPage = link.click();
assertEquals("My Page", myPage.getTitle());
}
@Test
public void testDontFind() {
addPageRequest("http://test.com",
newHtml("Test Page", "<a id=\"t\" href=\"http://test.com/myPage.html\">myPage</a>"));
AbstractDocument page = agent().get("http://test.com");
assertEquals(1, page.links().size());
Link link = page.links().find("#nothere");
assertNull(link);
}
@Test
public void testFollowingAnRelativeLink() {
addPageRequest("http://test.com",
newHtml("Test Page", "<a href=\"myPage.html\">myPage</a>"));
addPageRequest("http://test.com/myPage.html", newHtml("My Page", ""));
AbstractDocument page = agent().get("http://test.com");
Link link = page.links().find(contains("myPage"));
assertNotNull(link);
Resource myPage = link.click();
assertEquals("My Page", myPage.getTitle());
}
@Test
public void testFollowingAnRelativeLinkWithBaseSetInHtmlPage() {
addPageRequest("http://test.com",
"<html><head><base href=\"http://www1.test.com\"/></head><body><a href=\"myPage.html\">myPage</a></body></html>");
addPageRequest("http://www1.test.com/myPage.html", newHtml("My Page", ""));
AbstractDocument page = agent().get("http://test.com");
Link link = page.links().get(0);
assertNotNull(link);
Resource myPage = link.click();
assertEquals("My Page", myPage.getTitle());
}
@Test
public void testFollowingAnRelativeLinkWithContentLocationSetInHtmlResponseHeader() {
PageRequest pageRequest = addPageRequest("http://test.com",
newHtml("Test Page", "<a href=\"myPage.html\">myPage</a>"));
pageRequest.setContentLocation("http://www1.test.com");
addPageRequest("http://www1.test.com/myPage.html", newHtml("My Page", ""));
AbstractDocument page = agent().get("http://test.com");
Link link = page.links().find(contains("myPage"));
assertNotNull(link);
Resource myPage = link.click();
assertEquals("My Page", myPage.getTitle());
}
}