SiteParser.parse(): add unit tests.

php-coder · php-coder · commit ecfdd3c17d3e · 2017-12-12T23:51:30.000+01:00
Addressed to #685 No functional changes.
diff --git a/src/main/java/ru/mystamps/web/util/extractor/SeriesInfo.java b/src/main/java/ru/mystamps/web/util/extractor/SeriesInfo.java
@@ -17,14 +17,18 @@
  */
 package ru.mystamps.web.util.extractor;
 
+import lombok.EqualsAndHashCode;
 import lombok.Getter;
 import lombok.Setter;
+import lombok.ToString;
 
 /**
  * Representation of a series info.
  */
 @Getter
 @Setter
+@ToString
+@EqualsAndHashCode
 public class SeriesInfo {
 	private String categoryName;
 	private String countryName;
diff --git a/src/test/java/ru/mystamps/web/util/extractor/SiteParserTest.java b/src/test/java/ru/mystamps/web/util/extractor/SiteParserTest.java
@@ -292,6 +292,101 @@ public void parseShouldRequireNonBlankPageContent() {
 		parser.parse(nullOrBlank());
 	}
 	
+	@Test
+	public void parseShouldExtractSeriesInfo() {
+		String baseUri = "http://base.uri";
+		String expectedCategory = Random.categoryName();
+		String expectedCountry = Random.countryName();
+		String expectedIssueDate = Random.issueYear().toString();
+		String imageUrl = String.format(
+			"/%s-%s-%s.png",
+			expectedCountry.toLowerCase(),
+			expectedCategory.toLowerCase(),
+			expectedIssueDate
+		);
+		String expectedImageUrl = baseUri + imageUrl;
+		
+		parser.setMatchedUrl(baseUri);
+		parser.setCategoryLocator("#category-name");
+		parser.setCountryLocator("#country-name");
+		parser.setIssueDateLocator("#issue-date");
+		parser.setImageUrlLocator("#image-url");
+		
+		SeriesInfo expectedInfo = new SeriesInfo();
+		expectedInfo.setCategoryName(expectedCategory);
+		expectedInfo.setCountryName(expectedCountry);
+		expectedInfo.setIssueDate(expectedIssueDate);
+		expectedInfo.setImageUrl(expectedImageUrl);
+		
+		String html = String.format(
+			"<html>"
+				+ "<body>"
+					+ "<p id='category-name'>%s</p>"
+					+ "<p id='country-name'>%s</p>"
+					+ "<p id='issue-date'>%s</p>"
+					+ "<a id='image-url' href='%s'>look at image</a>"
+				+ "</body>"
+			+ "</html",
+			expectedCategory,
+			expectedCountry,
+			expectedIssueDate,
+			imageUrl
+		);
+		
+		SeriesInfo info = parser.parse(html);
+		
+		assertThat(info, is(equalTo(expectedInfo)));
+	}
+	@Test
+	public void parseShouldExtractSeriesInfoFromFirstMatchedElements() {
+		String baseUri = "http://base.uri";
+		String expectedCategory = Random.categoryName();
+		String expectedCountry = Random.countryName();
+		String expectedIssueDate = Random.issueYear().toString();
+		String imageUrl = String.format(
+			"/%s-%s-%s.png",
+			expectedCountry.toLowerCase(),
+			expectedCategory.toLowerCase(),
+			expectedIssueDate
+		);
+		String expectedImageUrl = baseUri + imageUrl;
+		
+		parser.setMatchedUrl(baseUri);
+		parser.setCategoryLocator("h1");
+		parser.setCountryLocator("p");
+		parser.setIssueDateLocator("span");
+		parser.setImageUrlLocator("a");
+		
+		SeriesInfo expectedInfo = new SeriesInfo();
+		expectedInfo.setCategoryName(expectedCategory);
+		expectedInfo.setCountryName(expectedCountry);
+		expectedInfo.setIssueDate(expectedIssueDate);
+		expectedInfo.setImageUrl(expectedImageUrl);
+		
+		String html = String.format(
+			"<html>"
+				+ "<body>"
+					+ "<h1>%s</h1>"
+					+ "<p>%s</p>"
+					+ "<span>%s</span>"
+					+ "<a href='%s'>look at image</a>"
+					+ "<h1>ignored</h1>"
+					+ "<p>ignored</p>"
+					+ "<span>ignored</span>"
+					+ "<a href='none'>look at image</a>"
+				+ "</body>"
+			+ "</html",
+			expectedCategory,
+			expectedCountry,
+			expectedIssueDate,
+			expectedImageUrl
+		);
+		
+		SeriesInfo info = parser.parse(html);
+		
+		assertThat(info, is(equalTo(expectedInfo)));
+	}
+	
 	//
 	// Tests for toString()
 	//
