Added Country functionality to Lookup

pom.xml

@@ -3,7 +3,7 @@
 	<modelVersion>4.0.0</modelVersion>
 	<groupId>be.ceau</groupId>
 	<artifactId>itunes-api</artifactId>
-	<version>4.1</version>
+	<version>4.1.1-SNAPSHOT</version>
 	<packaging>jar</packaging>
 	<name>iTunes API</name>
 	<description>Java client for the Apple iTunes APIs</description>

src/main/java/be/ceau/itunesapi/Lookup.java

@@ -27,6 +27,7 @@
 import java.util.Map.Entry;
 import java.util.Set;
 
+import be.ceau.itunesapi.request.Country;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -60,6 +61,7 @@ public class Lookup implements Serializable {
 	private Entity entity;
 	private int limit;
 	private Sort sort;
+	private Country country;
 
 	/**
 	 * No-arg constructor.
@@ -383,6 +385,23 @@ public Lookup setSort(Sort sort) {
 		return this;
 	}
 
+	/**
+	 * @return {@link Country} instance, or {@code null}
+	 */
+	public Country getCountry() {
+		return country;
+	}
+
+	/**
+	 * @param country
+	 *            {@link Country}, can be {@code null}
+	 * @return {@code this} instance for method chaining
+	 */
+	public Lookup setCountry(Country country) {
+		this.country = country;
+		return this;
+	}
+
 	/**
 	 * Create the request url for this {@link Lookup}
 	 * 
@@ -421,6 +440,12 @@ public String build() {
 			}
 			sb.append("sort").append("=").append(sort.toString());
 		}
+		if (country != null) {
+			if (sb.length() > 0) {
+				sb.append("&");
+			}
+			sb.append("country").append("=").append(country.getIso());
+		}
 		return API_ENDPOINT + sb.toString();
 	}
 

src/test/java/be/ceau/itunesapi/LookupTest.java

@@ -15,6 +15,7 @@
 */
 package be.ceau.itunesapi;
 
+import be.ceau.itunesapi.request.Country;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
@@ -45,4 +46,15 @@ public void albums() {
 		Assert.assertEquals(response.getResults().size(), response.getResultCount());
 	}
 
+	@Test
+	public void albumsFromUSStore() {
+		Response response = new Lookup()
+				.addId("178834")
+				.setEntity(Entity.ALBUM)
+				.setCountry(Country.UNITED_STATES)
+				.execute();
+		Assert.assertTrue(response.getResultCount() > 0);
+		Assert.assertEquals(response.getResults().size(), response.getResultCount());
+	}
+
 }