planetmanagertest.java

这个weblogging 设计得比较精巧

            assertNotNull(sub);
            
            PlanetEntryData entry1 = new PlanetEntryData();
            entry1.setPermalink("test_entry1");
            entry1.setCategoriesString("test,test2");
            entry1.setSubscription(sub);
            entry1.setPublished(new Date());
            entry1.save();
            sub.addEntry(entry1);
            
            PlanetEntryData entry2 = new PlanetEntryData();
            entry2.setPermalink("test_entry2");
            entry2.setCategoriesString("test_cat1,test_cat2,test_cat3");
            entry2.setSubscription(sub);
            entry2.setPublished(new Date());
            entry2.save();
            sub.addEntry(entry2);
            
            roller.commit();
        }
        {   // get sub and check it's entries, remove it
            roller.begin();
            PlanetSubscriptionData sub = planet.getSubscription("test_url");
            assertEquals(2, sub.getEntries().size());
            planet.deleteSubscription(sub);
            roller.commit();
        }
        {   // make sure sub is gone
            roller.begin();
            PlanetSubscriptionData sub = planet.getSubscription("test_url");
            assertNull(sub);
            roller.rollback();
        }
    }
    public void testRefreshEntries()
    {
        try 
        {
            Roller roller = getRoller();
            assertNotNull(roller);       
            PlanetManager planet = roller.getPlanetManager();
            assertNotNull(planet);     
            String feed_url1 = "http://rollerweblogger.org/rss/roller";
            {   
                roller.begin();
                PlanetConfigData config = new PlanetConfigData();
                config.setCacheDir("/tmp"); // TODO use real temp dir
                config.setTitle("test_title");
                config.setAdminEmail("test_admin_email");
                planet.saveConfiguration(config);
                
                PlanetGroupData group = new PlanetGroupData();
                group.setDescription("test_group_desc");
                group.setHandle("test_handle");
                group.setTitle("test_title");
                planet.saveGroup(group);
                
                PlanetSubscriptionData sub = new PlanetSubscriptionData();
                sub.setFeedUrl(feed_url1);
                planet.saveSubscription(sub);
                
                group.addSubscription(sub); 
                planet.saveGroup(group);
                
                roller.commit();
            }        
            {   
                roller.begin();
                planet.refreshEntries();
                roller.commit();
                
                roller.begin();
                PlanetSubscriptionData sub = planet.getSubscription(feed_url1);
                assertTrue(sub.getEntries().size() > 0);
          
                PlanetGroupData group = planet.getGroup("test_handle");
                assertNotNull(group);
                
                planet.deleteSubscription(sub);
                planet.deleteGroup(group);
                
                PlanetConfigData config = planet.getConfiguration();
                config.remove();
                roller.commit();
            }
        }
        catch (Exception e) 
        {
            e.printStackTrace();
        }
    }
    public void testAggregations() throws Exception
    {
        try 
        {
            Roller roller = getRoller();
            assertNotNull(roller);       
            PlanetManager planet = roller.getPlanetManager();
            assertNotNull(planet);     
            String feed_url1 = "http://rollerweblogger.org/rss/roller";
            String feed_url2 = "http://linuxintegrators.com/acoliver/?flavor=rss2";
            {   
                roller.begin();
                PlanetConfigData config = new PlanetConfigData();
                config.setCacheDir("/tmp"); // TODO use real temp dir
                config.setTitle("test_title");
                config.setAdminEmail("test_admin_email");
                planet.saveConfiguration(config);
                
                PlanetGroupData group = new PlanetGroupData();
                group.setDescription("test_group_desc");
                group.setHandle("test_handle");
                group.setTitle("test_title");
                planet.saveGroup(group);
                
                PlanetSubscriptionData sub1 = new PlanetSubscriptionData();
                sub1.setFeedUrl(feed_url1);
                planet.saveSubscription(sub1);
                
                PlanetSubscriptionData sub2 = new PlanetSubscriptionData();
                sub2.setFeedUrl(feed_url2);
                planet.saveSubscription(sub2);
                
                group.addSubscription(sub1); 
                group.addSubscription(sub2); 
                planet.saveGroup(group);
                
                roller.commit();
            }        
            {   
                roller.begin();
                planet.refreshEntries();
                roller.commit();
                
                roller.begin();
                int count = 0;
                Iterator subs = planet.getAllSubscriptions();
                while  (subs.hasNext()) 
                {
                    PlanetSubscriptionData sub= (PlanetSubscriptionData)subs.next();
                    count += sub.getEntries().size();    
                }
                PlanetSubscriptionData sub1 = planet.getSubscription(feed_url1);
                assertTrue(sub1.getEntries().size() > 0);
                PlanetSubscriptionData sub2 = planet.getSubscription(feed_url2);
                assertTrue(sub2.getEntries().size() > 0);
                assertEquals(count, sub1.getEntries().size() + sub2.getEntries().size());
                
                PlanetGroupData group = planet.getGroup("test_handle");
                assertNotNull(group);
    
                List bigag = planet.getAggregation(group, 1000);
                assertEquals(bigag.size(), count);
                      
                List littleag = planet.getAggregation(group, 10);
                assertEquals(littleag.size(), count);
                                  
                planet.deleteSubscription(sub1);
                planet.deleteSubscription(sub2);
                planet.deleteGroup(group);
                
                PlanetConfigData config = planet.getConfiguration();
                config.remove();
                roller.commit(); 
            }
        }
        catch (Exception e)
        {
            e.printStackTrace();
        }
    }
    public void testSubscriptionCount() throws Exception
    {
        try 
        {
            Roller roller = getRoller();
            assertNotNull(roller);       
            PlanetManager planet = roller.getPlanetManager();
            assertNotNull(planet);     
            String feed_url1 = "http://rollerweblogger.org/rss/roller";
            String feed_url2 = "http://linuxintegrators.com/acoliver/?flavor=rss2";
            {   
                roller.begin();            
                PlanetSubscriptionData sub1 = new PlanetSubscriptionData();
                sub1.setFeedUrl(feed_url1);
                planet.saveSubscription(sub1);            
                PlanetSubscriptionData sub2 = new PlanetSubscriptionData();
                sub2.setFeedUrl(feed_url2);
                planet.saveSubscription(sub2);
                roller.commit();
    
                roller.begin();
                assertEquals(2, planet.getSubscriptionCount());
                roller.rollback();
            }     
        }
        catch (Exception e) 
        {
            e.printStackTrace();
        }
    }
    /*
    public void testRankingRange()
    {
        int limit = 1000;
        int count = 5500;
        int mod = (count / limit) + 1;        
        for (int i=0; i<20; i++)
        {
            int start = (i % mod) * limit;
            int end = start + 999;
            end = end > count ? count : end; 
            System.out.println("start="+start+" end="+end);
        }
    }
    */
    public Roller getRoller() throws RollerException
    {
        return RollerFactory.getRoller();
    }

    public static Test suite()
    {
        return new TestSuite(PlanetManagerTest.class);
    }
}