Move album parsing to new configuration parser.
[Sone.git] / src / test / java / net / pterodactylus / sone / core / ConfigurationSoneParserTest.java
index daf2cf9..b5e57f7 100644 (file)
@@ -4,30 +4,48 @@ import static com.google.common.base.Optional.fromNullable;
 import static com.google.common.base.Optional.of;
 import static java.lang.System.currentTimeMillis;
 import static java.util.UUID.randomUUID;
+import static net.pterodactylus.sone.Matchers.isAlbum;
 import static net.pterodactylus.sone.Matchers.isPost;
+import static net.pterodactylus.sone.Matchers.isPostReply;
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.contains;
+import static org.hamcrest.Matchers.containsInAnyOrder;
 import static org.hamcrest.Matchers.emptyIterable;
 import static org.hamcrest.Matchers.hasSize;
 import static org.hamcrest.Matchers.is;
 import static org.hamcrest.Matchers.notNullValue;
 import static org.hamcrest.Matchers.nullValue;
+import static org.mockito.Matchers.any;
 import static org.mockito.Matchers.anyString;
 import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
+import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
+import java.util.Set;
 import java.util.concurrent.atomic.AtomicReference;
 
+import net.pterodactylus.sone.core.ConfigurationSoneParser.InvalidAlbumFound;
+import net.pterodactylus.sone.core.ConfigurationSoneParser.InvalidParentAlbumFound;
 import net.pterodactylus.sone.core.ConfigurationSoneParser.InvalidPostFound;
+import net.pterodactylus.sone.core.ConfigurationSoneParser.InvalidPostReplyFound;
+import net.pterodactylus.sone.data.Album;
+import net.pterodactylus.sone.data.Album.Modifier;
+import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.PostReply;
 import net.pterodactylus.sone.data.Profile;
 import net.pterodactylus.sone.data.Profile.Field;
 import net.pterodactylus.sone.data.Sone;
+import net.pterodactylus.sone.database.AlbumBuilder;
+import net.pterodactylus.sone.database.AlbumBuilderFactory;
 import net.pterodactylus.sone.database.PostBuilder;
 import net.pterodactylus.sone.database.PostBuilderFactory;
+import net.pterodactylus.sone.database.PostReplyBuilder;
+import net.pterodactylus.sone.database.PostReplyBuilderFactory;
 import net.pterodactylus.util.config.Configuration;
 import net.pterodactylus.util.config.ConfigurationException;
 import net.pterodactylus.util.config.Value;
@@ -200,6 +218,186 @@ public class ConfigurationSoneParserTest {
                setupPost("1", null, 0L, null, null);
        }
 
+       @Test
+       public void postRepliesAreParsedCorrectly() {
+               setupPostReplies();
+               PostReplyBuilderFactory postReplyBuilderFactory =
+                               new PostReplyBuilderFactory() {
+                                       @Override
+                                       public PostReplyBuilder newPostReplyBuilder() {
+                                               return new TestPostReplyBuilder();
+                                       }
+                               };
+               Collection<PostReply> postReplies =
+                               configurationSoneParser.parsePostReplies(
+                                               postReplyBuilderFactory);
+               assertThat(postReplies, hasSize(2));
+               assertThat(postReplies,
+                               containsInAnyOrder(isPostReply("R0", "P0", 1000L, "T0"),
+                                               isPostReply("R1", "P1", 1001L, "T1")));
+       }
+
+       private void setupPostReplies() {
+               setupPostReply("0", "R0", "P0", 1000L, "T0");
+               setupPostReply("1", "R1", "P1", 1001L, "T1");
+               setupPostReply("2", null, null, 0L, null);
+       }
+
+       private void setupPostReply(String postReplyNumber, String postReplyId,
+                       String postId, long time, String text) {
+               setupString("Sone/1/Replies/" + postReplyNumber + "/ID", postReplyId);
+               setupString("Sone/1/Replies/" + postReplyNumber + "/Post/ID", postId);
+               setupLong("Sone/1/Replies/" + postReplyNumber + "/Time", time);
+               setupString("Sone/1/Replies/" + postReplyNumber + "/Text", text);
+       }
+
+       @Test(expected = InvalidPostReplyFound.class)
+       public void missingPostIdIsRecognized() {
+               setupPostReplyWithMissingPostId();
+               configurationSoneParser.parsePostReplies(null);
+       }
+
+       private void setupPostReplyWithMissingPostId() {
+               setupPostReply("0", "R0", null, 1000L, "T0");
+       }
+
+       @Test(expected = InvalidPostReplyFound.class)
+       public void missingPostReplyTimeIsRecognized() {
+               setupPostReplyWithMissingPostReplyTime();
+               configurationSoneParser.parsePostReplies(null);
+       }
+
+       private void setupPostReplyWithMissingPostReplyTime() {
+               setupPostReply("0", "R0", "P0", 0L, "T0");
+       }
+
+       @Test(expected = InvalidPostReplyFound.class)
+       public void missingPostReplyTextIsRecognized() {
+               setupPostReplyWithMissingPostReplyText();
+               configurationSoneParser.parsePostReplies(null);
+       }
+
+       private void setupPostReplyWithMissingPostReplyText() {
+               setupPostReply("0", "R0", "P0", 1000L, null);
+       }
+
+       @Test
+       public void likedPostIdsParsedCorrectly() {
+               setupLikedPostIds();
+               Set<String> likedPostIds =
+                               configurationSoneParser.parseLikedPostIds();
+               assertThat(likedPostIds, containsInAnyOrder("P1", "P2", "P3"));
+       }
+
+       private void setupLikedPostIds() {
+               setupString("Sone/1/Likes/Post/0/ID", "P1");
+               setupString("Sone/1/Likes/Post/1/ID", "P2");
+               setupString("Sone/1/Likes/Post/2/ID", "P3");
+               setupString("Sone/1/Likes/Post/3/ID", null);
+       }
+
+       @Test
+       public void likedPostReplyIdsAreParsedCorrectly() {
+               setupLikedPostReplyIds();
+               Set<String> likedPostReplyIds =
+                               configurationSoneParser.parseLikedPostReplyIds();
+               assertThat(likedPostReplyIds, containsInAnyOrder("R1", "R2", "R3"));
+       }
+
+       private void setupLikedPostReplyIds() {
+               setupString("Sone/1/Likes/Reply/0/ID", "R1");
+               setupString("Sone/1/Likes/Reply/1/ID", "R2");
+               setupString("Sone/1/Likes/Reply/2/ID", "R3");
+               setupString("Sone/1/Likes/Reply/3/ID", null);
+       }
+
+       @Test
+       public void friendsAreParsedCorrectly() {
+               setupFriends();
+               Set<String> friends = configurationSoneParser.parseFriends();
+               assertThat(friends, containsInAnyOrder("F1", "F2", "F3"));
+       }
+
+       private void setupFriends() {
+               setupString("Sone/1/Friends/0/ID", "F1");
+               setupString("Sone/1/Friends/1/ID", "F2");
+               setupString("Sone/1/Friends/2/ID", "F3");
+               setupString("Sone/1/Friends/3/ID", null);
+       }
+
+       @Test
+       public void topLevelAlbumsAreParsedCorrectly() {
+               setupTopLevelAlbums();
+               AlbumBuilderFactory albumBuilderFactory = createAlbumBuilderFactory();
+               List<Album> topLevelAlbums =
+                               configurationSoneParser.parseTopLevelAlbums(
+                                               albumBuilderFactory);
+               assertThat(topLevelAlbums, hasSize(2));
+               Album firstAlbum = topLevelAlbums.get(0);
+               assertThat(firstAlbum, isAlbum("A1", null, "T1", "D1", "I1"));
+               assertThat(firstAlbum.getAlbums(), emptyIterable());
+               assertThat(firstAlbum.getImages(), emptyIterable());
+               Album secondAlbum = topLevelAlbums.get(1);
+               assertThat(secondAlbum, isAlbum("A2", null, "T2", "D2", null));
+               assertThat(secondAlbum.getAlbums(), hasSize(1));
+               assertThat(secondAlbum.getImages(), emptyIterable());
+               Album thirdAlbum = secondAlbum.getAlbums().get(0);
+               assertThat(thirdAlbum, isAlbum("A3", "A2", "T3", "D3", "I3"));
+               assertThat(thirdAlbum.getAlbums(), emptyIterable());
+               assertThat(thirdAlbum.getImages(), emptyIterable());
+       }
+
+       private void setupTopLevelAlbums() {
+               setupAlbum(0, "A1", null, "T1", "D1", "I1");
+               setupAlbum(1, "A2", null, "T2", "D2", null);
+               setupAlbum(2, "A3", "A2", "T3", "D3", "I3");
+               setupAlbum(3, null, null, null, null, null);
+       }
+
+       private void setupAlbum(int albumNumber, String albumId,
+                       String parentAlbumId,
+                       String title, String description, String imageId) {
+               final String albumPrefix = "Sone/1/Albums/" + albumNumber;
+               setupString(albumPrefix + "/ID", albumId);
+               setupString(albumPrefix + "/Title", title);
+               setupString(albumPrefix + "/Description", description);
+               setupString(albumPrefix + "/Parent", parentAlbumId);
+               setupString(albumPrefix + "/AlbumImage", imageId);
+       }
+
+       private AlbumBuilderFactory createAlbumBuilderFactory() {
+               AlbumBuilderFactory albumBuilderFactory =
+                               mock(AlbumBuilderFactory.class);
+               when(albumBuilderFactory.newAlbumBuilder()).thenAnswer(
+                               new Answer<AlbumBuilder>() {
+                                       @Override
+                                       public AlbumBuilder answer(InvocationOnMock invocation) {
+                                               return new TestAlbumBuilder();
+                                       }
+                               });
+               return albumBuilderFactory;
+       }
+
+       @Test(expected = InvalidAlbumFound.class)
+       public void albumWithInvalidTitleIsRecognized() {
+               setupAlbum(0, "A1", null, null, "D1", "I1");
+               configurationSoneParser.parseTopLevelAlbums(
+                               createAlbumBuilderFactory());
+       }
+
+       @Test(expected = InvalidAlbumFound.class)
+       public void albumWithInvalidDescriptionIsRecognized() {
+               setupAlbum(0, "A1", null, "T1", null, "I1");
+               configurationSoneParser.parseTopLevelAlbums(
+                               createAlbumBuilderFactory());
+       }
+
+       @Test(expected = InvalidParentAlbumFound.class)
+       public void albumWithInvalidParentIsRecognized() {
+               setupAlbum(0, "A1", "A0", "T1", "D1", "I1");
+               configurationSoneParser.parseTopLevelAlbums(
+                               createAlbumBuilderFactory());
+       }
 
        private static class TestValue<T> implements Value<T> {
 
@@ -289,4 +487,174 @@ public class ConfigurationSoneParserTest {
 
        }
 
+       private static class TestPostReplyBuilder implements PostReplyBuilder {
+
+               private final PostReply postReply = mock(PostReply.class);
+
+               @Override
+               public PostReplyBuilder to(String postId) {
+                       when(postReply.getPostId()).thenReturn(postId);
+                       return this;
+               }
+
+               @Override
+               public PostReply build() throws IllegalStateException {
+                       return postReply;
+               }
+
+               @Override
+               public PostReplyBuilder randomId() {
+                       when(postReply.getId()).thenReturn(randomUUID().toString());
+                       return this;
+               }
+
+               @Override
+               public PostReplyBuilder withId(String id) {
+                       when(postReply.getId()).thenReturn(id);
+                       return this;
+               }
+
+               @Override
+               public PostReplyBuilder from(String senderId) {
+                       Sone sone = mock(Sone.class);
+                       when(sone.getId()).thenReturn(senderId);
+                       when(postReply.getSone()).thenReturn(sone);
+                       return this;
+               }
+
+               @Override
+               public PostReplyBuilder currentTime() {
+                       when(postReply.getTime()).thenReturn(currentTimeMillis());
+                       return this;
+               }
+
+               @Override
+               public PostReplyBuilder withTime(long time) {
+                       when(postReply.getTime()).thenReturn(time);
+                       return this;
+               }
+
+               @Override
+               public PostReplyBuilder withText(String text) {
+                       when(postReply.getText()).thenReturn(text);
+                       return this;
+               }
+
+       }
+
+       private static class TestAlbumBuilder implements AlbumBuilder {
+
+               private final Album album = mock(Album.class);
+               private final List<Album> albums = new ArrayList<Album>();
+               private final List<Image> images = new ArrayList<Image>();
+               private Album parentAlbum;
+               private String title;
+               private String description;
+               private String imageId;
+
+               public TestAlbumBuilder() {
+                       when(album.getTitle()).thenAnswer(new Answer<String>() {
+                               @Override
+                               public String answer(InvocationOnMock invocation) {
+                                       return title;
+                               }
+                       });
+                       when(album.getDescription()).thenAnswer(new Answer<String>() {
+                               @Override
+                               public String answer(InvocationOnMock invocation) {
+                                       return description;
+                               }
+                       });
+                       when(album.getAlbumImage()).thenAnswer(new Answer<Image>() {
+                               @Override
+                               public Image answer(InvocationOnMock invocation) {
+                                       if (imageId == null) {
+                                               return null;
+                                       }
+                                       Image image = mock(Image.class);
+                                       when(image.getId()).thenReturn(imageId);
+                                       return image;
+                               }
+                       });
+                       when(album.getAlbums()).thenReturn(albums);
+                       when(album.getImages()).thenReturn(images);
+                       doAnswer(new Answer<Void>() {
+                               @Override
+                               public Void answer(InvocationOnMock invocation) {
+                                       albums.add((Album) invocation.getArguments()[0]);
+                                       ((Album) invocation.getArguments()[0]).setParent(album);
+                                       return null;
+                               }
+                       }).when(album).addAlbum(any(Album.class));
+                       doAnswer(new Answer<Void>() {
+                               @Override
+                               public Void answer(InvocationOnMock invocation) {
+                                       images.add((Image) invocation.getArguments()[0]);
+                                       return null;
+                               }
+                       }).when(album).addImage(any(Image.class));
+                       doAnswer(new Answer<Void>() {
+                               @Override
+                               public Void answer(InvocationOnMock invocation) {
+                                       parentAlbum = (Album) invocation.getArguments()[0];
+                                       return null;
+                               }
+                       }).when(album).setParent(any(Album.class));
+                       when(album.getParent()).thenAnswer(new Answer<Album>() {
+                               @Override
+                               public Album answer(InvocationOnMock invocation) {
+                                       return parentAlbum;
+                               }
+                       });
+                       when(album.modify()).thenReturn(new Modifier() {
+                               @Override
+                               public Modifier setTitle(String title) {
+                                       TestAlbumBuilder.this.title = title;
+                                       return this;
+                               }
+
+                               @Override
+                               public Modifier setDescription(String description) {
+                                       TestAlbumBuilder.this.description = description;
+                                       return this;
+                               }
+
+                               @Override
+                               public Modifier setAlbumImage(String imageId) {
+                                       TestAlbumBuilder.this.imageId = imageId;
+                                       return this;
+                               }
+
+                               @Override
+                               public Album update() throws IllegalStateException {
+                                       return album;
+                               }
+                       });
+               }
+
+               @Override
+               public AlbumBuilder randomId() {
+                       when(album.getId()).thenReturn(randomUUID().toString());
+                       return this;
+               }
+
+               @Override
+               public AlbumBuilder withId(String id) {
+                       when(album.getId()).thenReturn(id);
+                       return this;
+               }
+
+               @Override
+               public AlbumBuilder by(Sone sone) {
+                       when(album.getSone()).thenReturn(sone);
+                       return this;
+               }
+
+               @Override
+               public Album build() throws IllegalStateException {
+                       return album;
+               }
+
+       }
+
 }