diff --git a/partial-response/pom.xml b/partial-response/pom.xml
index e4f4db4ae..7ce36ecec 100644
--- a/partial-response/pom.xml
+++ b/partial-response/pom.xml
@@ -15,6 +15,10 @@
junit
junit
+
+ org.mockito
+ mockito-core
+
diff --git a/partial-response/src/main/java/com/iluwatar/partialresponse/FieldJsonMapper.java b/partial-response/src/main/java/com/iluwatar/partialresponse/FieldJsonMapper.java
new file mode 100644
index 000000000..32e9ae83d
--- /dev/null
+++ b/partial-response/src/main/java/com/iluwatar/partialresponse/FieldJsonMapper.java
@@ -0,0 +1,31 @@
+/*
+ * The MIT License (MIT)
+ *
+ * Copyright (c) 2017 Gopinath Langote
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+package com.iluwatar.partialresponse;
+
+public class FieldJsonMapper {
+ public String toJson(Video video, String[] fields) {
+ return null;
+ }
+}
diff --git a/partial-response/src/main/java/com/iluwatar/partialresponse/VideoResource.java b/partial-response/src/main/java/com/iluwatar/partialresponse/VideoResource.java
index b233c42d3..8c571e6d6 100644
--- a/partial-response/src/main/java/com/iluwatar/partialresponse/VideoResource.java
+++ b/partial-response/src/main/java/com/iluwatar/partialresponse/VideoResource.java
@@ -27,13 +27,18 @@ package com.iluwatar.partialresponse;
import java.util.Map;
public class VideoResource {
+ private FieldJsonMapper fieldJsonMapper;
private Map videos;
- public VideoResource(Map videos) {
+ public VideoResource(FieldJsonMapper fieldJsonMapper, Map videos) {
+ this.fieldJsonMapper = fieldJsonMapper;
this.videos = videos;
}
- public String getDetails(Integer id) {
- return videos.get(id).toString();
+ public String getDetails(Integer id, String... fields) {
+ if (fields.length == 0) {
+ return videos.get(id).toString();
+ }
+ return fieldJsonMapper.toJson(videos.get(id), fields);
}
}
diff --git a/partial-response/src/test/java/com/iluwatar/partialresponse/VideoResourceTest.java b/partial-response/src/test/java/com/iluwatar/partialresponse/VideoResourceTest.java
index 329b10087..df25c2dea 100644
--- a/partial-response/src/test/java/com/iluwatar/partialresponse/VideoResourceTest.java
+++ b/partial-response/src/test/java/com/iluwatar/partialresponse/VideoResourceTest.java
@@ -26,13 +26,23 @@ package com.iluwatar.partialresponse;
import org.junit.Before;
import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.runners.MockitoJUnitRunner;
import java.util.HashMap;
import java.util.Map;
import static org.junit.Assert.assertEquals;
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.when;
+@RunWith(MockitoJUnitRunner.class)
public class VideoResourceTest {
+ @Mock
+ private FieldJsonMapper fieldJsonMapper;
+
private VideoResource resource;
@Before
@@ -41,14 +51,26 @@ public class VideoResourceTest {
videos.put(1, new Video(1, "Avatar", 178, "epic science fiction film", "James Cameron", "English"));
videos.put(2, new Video(2, "Godzilla Resurgence", 120, "Action & drama movie|", "Hideaki Anno", "Japanese"));
videos.put(3, new Video(3, "Interstellar", 169, "Adventure & Sci-Fi", "Christopher Nolan", "English"));
- resource = new VideoResource(videos);
+ resource = new VideoResource(fieldJsonMapper, videos);
}
@Test
public void shouldGiveVideoDetailsById() {
- String details = resource.getDetails(1);
+ String actualDetails = resource.getDetails(1);
String expectedDetails = "{\"id\": \"1\",\"title\": \"Avatar\",\"length\": 178,\"description\": \"epic science fiction film\",\"director\": \"James Cameron\",\"language\": \"English\",}";
- assertEquals(expectedDetails, details);
+ assertEquals(expectedDetails, actualDetails);
+ }
+
+ @Test
+ public void shouldGiveSpecifiedFieldsInformationOfVideo() {
+ String[] fields = new String[]{"title", "length"};
+
+ String expectedDetails = "{\"id\": \"1\",\"title\": \"Avatar\",\"length\": 178}";
+ when(fieldJsonMapper.toJson(any(Video.class), eq(fields))).thenReturn(expectedDetails);
+
+ String actualFieldsDetails = resource.getDetails(2, fields);
+
+ assertEquals(expectedDetails, actualFieldsDetails);
}
}
\ No newline at end of file