optimize opml export (fix #687)

src/test/java/com/commafeed/backend/opml/OPMLExporterTest.java

@@ -0,0 +1,141 @@
+package com.commafeed.backend.opml;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.when;
+
+import java.util.ArrayList;
+import java.util.HashSet;
+import java.util.List;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+
+import com.commafeed.backend.dao.FeedCategoryDAO;
+import com.commafeed.backend.dao.FeedSubscriptionDAO;
+import com.commafeed.backend.model.Feed;
+import com.commafeed.backend.model.FeedCategory;
+import com.commafeed.backend.model.FeedSubscription;
+import com.commafeed.backend.model.User;
+import com.rometools.opml.feed.opml.Opml;
+import com.rometools.opml.feed.opml.Outline;
+
+public class OPMLExporterTest {
+
+	@Mock
+	private FeedCategoryDAO feedCategoryDAO;
+	@Mock
+	private FeedSubscriptionDAO feedSubscriptionDAO;
+
+	private User user = new User();
+
+	private FeedCategory cat1 = new FeedCategory();
+	private FeedCategory cat2 = new FeedCategory();
+
+	private FeedSubscription rootFeed = newFeedSubscription("rootFeed", "rootFeed.com");
+	private FeedSubscription cat1Feed = newFeedSubscription("cat1Feed", "cat1Feed.com");
+	private FeedSubscription cat2Feed = newFeedSubscription("cat2Feed", "cat2Feed.com");
+
+	private List<FeedCategory> categories = new ArrayList<>();
+	private List<FeedSubscription> subscriptions = new ArrayList<>();
+
+	@Before
+	public void before_each_test() {
+		MockitoAnnotations.initMocks(this);
+
+		user.setName("John Doe");
+
+		cat1.setId(1l);
+		cat1.setName("cat1");
+		cat1.setParent(null);
+		cat1.setChildren(new HashSet<FeedCategory>());
+		cat1.setSubscriptions(new HashSet<FeedSubscription>());
+
+		cat2.setId(2l);
+		cat2.setName("cat2");
+		cat2.setParent(cat1);
+		cat2.setChildren(new HashSet<FeedCategory>());
+		cat2.setSubscriptions(new HashSet<FeedSubscription>());
+
+		cat1.getChildren().add(cat2);
+
+		rootFeed.setCategory(null);
+		cat1Feed.setCategory(cat1);
+		cat2Feed.setCategory(cat2);
+
+		cat1.getSubscriptions().add(cat1Feed);
+		cat2.getSubscriptions().add(cat2Feed);
+
+		categories.add(cat1);
+		categories.add(cat2);
+
+		subscriptions.add(rootFeed);
+		subscriptions.add(cat1Feed);
+		subscriptions.add(cat2Feed);
+	}
+
+	private Feed newFeed(String url) {
+		Feed feed = new Feed();
+		feed.setUrl(url);
+		return feed;
+	}
+
+	private FeedSubscription newFeedSubscription(String title, String url) {
+		FeedSubscription feedSubscription = new FeedSubscription();
+		feedSubscription.setTitle(title);
+		feedSubscription.setFeed(newFeed(url));
+		return feedSubscription;
+	}
+
+	@Test
+	public void generates_OPML_correctly() {
+		when(feedCategoryDAO.findAll(user)).thenReturn(categories);
+		when(feedSubscriptionDAO.findAll(user)).thenReturn(subscriptions);
+
+		Opml opml = new OPMLExporter(feedCategoryDAO, feedSubscriptionDAO).export(user);
+
+		List<Outline> rootOutlines = opml.getOutlines();
+		assertEquals(2, rootOutlines.size());
+		assertTrue(containsCategory(rootOutlines, "cat1"));
+		assertTrue(containsFeed(rootOutlines, "rootFeed", "rootFeed.com"));
+
+		Outline cat1Outline = getCategoryOutline(rootOutlines, "cat1");
+		List<Outline> cat1Children = cat1Outline.getChildren();
+		assertEquals(2, cat1Children.size());
+		assertTrue(containsCategory(cat1Children, "cat2"));
+		assertTrue(containsFeed(cat1Children, "cat1Feed", "cat1Feed.com"));
+
+		Outline cat2Outline = getCategoryOutline(cat1Children, "cat2");
+		List<Outline> cat2Children = cat2Outline.getChildren();
+		assertEquals(1, cat2Children.size());
+		assertTrue(containsFeed(cat2Children, "cat2Feed", "cat2Feed.com"));
+	}
+
+	private boolean containsCategory(List<Outline> outlines, String category) {
+		for (Outline o : outlines)
+			if (!"rss".equals(o.getType()))
+				if (category.equals(o.getTitle()))
+					return true;
+
+		return false;
+	}
+
+	private boolean containsFeed(List<Outline> outlines, String title, String url) {
+		for (Outline o : outlines)
+			if ("rss".equals(o.getType()))
+				if (title.equals(o.getTitle()) && o.getAttributeValue("xmlUrl").equals(url))
+					return true;
+
+		return false;
+	}
+
+	private Outline getCategoryOutline(List<Outline> outlines, String title) {
+		for (Outline o : outlines)
+			if (o.getTitle().equals(title))
+				return o;
+
+		return null;
+	}
+}