ITunesGeneratorTest.java

/*
 * ITunesGeneratorTest.java
 * JUnit based test
 *
 * Created on August 2, 2005, 2:31 PM
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package com.rometools.modules.itunes;

import com.rometools.modules.AbstractTestCase;
import com.rometools.modules.itunes.types.Category;
import com.rometools.rome.feed.synd.SyndEntry;
import com.rometools.rome.feed.synd.SyndFeed;
import com.rometools.rome.feed.synd.SyndFeedImpl;
import com.rometools.rome.io.SyndFeedInput;
import com.rometools.rome.io.SyndFeedOutput;
import com.rometools.rome.io.XmlReader;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.StringWriter;
import java.net.URL;
import java.util.List;
import junit.framework.Test;
import junit.framework.TestSuite;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class ITunesGeneratorTest extends AbstractTestCase {

    private static final Logger LOG = LoggerFactory.getLogger(ITunesGeneratorTest.class);

    static final String URI = AbstractITunesObject.URI;

    public ITunesGeneratorTest(final String testName) {
        super(testName);
    }

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

    /**
     * Test of generate method, of class com.totsp.xml.syndication.itunes.ITunesGenerator.
     */
    public void testEndToEnd() throws Exception {
        LOG.debug("testEndToEnd");
        testFile("xml/leshow.xml");
        // testFile( "/test/xml/apple.xml" );
        testFile("xml/lr.xml");
    }

    private void testFile(final String filename) throws Exception {
        final File feed = new File(getTestFile(filename));
        final SyndFeedInput input = new SyndFeedInput();
        final SyndFeed syndfeed = input.build(new XmlReader(feed));

        final SyndFeedOutput output = new SyndFeedOutput();
        final File outfeed = new File(feed.getAbsolutePath() + ".output");
        output.output(syndfeed, outfeed);

        final SyndFeed syndCheck = input.build(new XmlReader(outfeed));
        LOG.debug(syndCheck.getModule(AbstractITunesObject.URI).toString());
        assertEquals("Feed Level: ", syndfeed.getModule(AbstractITunesObject.URI).toString(), syndCheck.getModule(AbstractITunesObject.URI).toString());

        final List<SyndEntry> syndEntries = syndfeed.getEntries();
        final List<SyndEntry> syndChecks = syndCheck.getEntries();

        for (int i = 0; i < syndEntries.size(); i++) {
            final SyndEntry entry = syndEntries.get(i);
            final SyndEntry check = syndChecks.get(i);
            LOG.debug("Original: " + entry.getModule(AbstractITunesObject.URI));
            LOG.debug("Check:    " + check.getModule(AbstractITunesObject.URI));
            LOG.debug(entry.getModule(AbstractITunesObject.URI).toString());
            LOG.debug("-----------------------------------------");
            LOG.debug(check.getModule(AbstractITunesObject.URI).toString());
            assertEquals("Entry Level: ", entry.getModule(AbstractITunesObject.URI).toString(), check.getModule(AbstractITunesObject.URI).toString());
        }
    }

    public void testCreate() throws Exception {

        final SyndFeed feed = new SyndFeedImpl();
        final String feedType = "rss_2.0";
        feed.setFeedType(feedType);
        feed.setLanguage("en-us");
        feed.setTitle("sales.com on the Radio!");
        feed.setDescription("sales.com radio shows in MP3 format");
        feed.setLink("http://foo/rss/podcasts.rss");

        final FeedInformation fi = new FeedInformationImpl();
        fi.setOwnerName("sales.com");
        fi.getCategories().add(new Category("Shopping"));
        fi.setOwnerEmailAddress("patti@sales.com");
        fi.setType("serial");
        feed.getModules().add(fi);

        final SyndFeedOutput output = new SyndFeedOutput();
        final StringWriter writer = new StringWriter();
        output.output(feed, writer);
        LOG.debug("{}", writer);
    }

    public void testImage() throws Exception {
        SyndFeed feed = new SyndFeedImpl();
        feed.setFeedType("rss_2.0");
        feed.setTitle("title");
        feed.setDescription("description");
        feed.setLink("https://example.org");

        FeedInformation itunesFeed = new FeedInformationImpl();
        itunesFeed.setImage(new URL("https://example.org/test.png"));
        feed.getModules().add(itunesFeed);

        String xml = new SyndFeedOutput().outputString(feed);

        AbstractITunesObject parsedItunesFeed =
                (AbstractITunesObject) new SyndFeedInput()
                        .build(new XmlReader(new ByteArrayInputStream(xml.getBytes("UTF-8"))))
                        .getModule(AbstractITunesObject.URI);
        assertEquals(new URL("https://example.org/test.png"), parsedItunesFeed.getImage());
        assertEquals("https://example.org/test.png", parsedItunesFeed.getImageUri());
    }

    public void testImageUri() throws Exception {
        SyndFeed feed = new SyndFeedImpl();
        feed.setFeedType("rss_2.0");
        feed.setTitle("title");
        feed.setDescription("description");
        feed.setLink("https://example.org");

        FeedInformation itunesFeed = new FeedInformationImpl();
        itunesFeed.setImageUri("https://example.org/test.png");
        feed.getModules().add(itunesFeed);

        String xml = new SyndFeedOutput().outputString(feed);

        AbstractITunesObject parsedItunesFeed =
                (AbstractITunesObject) new SyndFeedInput()
                        .build(new XmlReader(new ByteArrayInputStream(xml.getBytes("UTF-8"))))
                        .getModule(AbstractITunesObject.URI);
        assertEquals("https://example.org/test.png",
                parsedItunesFeed.getImageUri());
        assertEquals(new URL("https://example.org/test.png"),
                parsedItunesFeed.getImage());
    }

    public void testImageTakesPrecedenceOverImageUri() throws Exception {
        SyndFeed feed = new SyndFeedImpl();
        feed.setFeedType("rss_2.0");
        feed.setTitle("title");
        feed.setDescription("description");
        feed.setLink("https://example.org");

        FeedInformation itunesFeed = new FeedInformationImpl();
        itunesFeed.setImage(new URL("https://example.org/test1.png"));
        itunesFeed.setImageUri("https://example.org/test2.png");
        feed.getModules().add(itunesFeed);

        String xml = new SyndFeedOutput().outputString(feed);

        AbstractITunesObject parsedItunesFeed =
                (AbstractITunesObject) new SyndFeedInput()
                        .build(new XmlReader(new ByteArrayInputStream(xml.getBytes("UTF-8"))))
                        .getModule(AbstractITunesObject.URI);
        assertEquals(new URL("https://example.org/test1.png"), parsedItunesFeed.getImage());
        assertEquals("https://example.org/test1.png", parsedItunesFeed.getImageUri());
    }
}