silverstripe-framework/tests/api/RSSFeedTest.php

<?php
/**
 * @package framework
 * @subpackage tests
 */
class RSSFeedTest extends SapphireTest {

	protected static $original_host;

	public function testRSSFeed() {
		$list = new ArrayList();
		$list->push(new RSSFeedTest_ItemA());
		$list->push(new RSSFeedTest_ItemB());
		$list->push(new RSSFeedTest_ItemC());

		$rssFeed = new RSSFeed($list, "http://www.example.com", "Test RSS Feed", "Test RSS Feed Description");
		$content = $rssFeed->outputToBrowser();

		$this->assertContains('<link>http://www.example.org/item-a/</link>', $content);
		$this->assertContains('<link>http://www.example.com/item-b.html</link>', $content);
		$this->assertContains('<link>http://www.example.com/item-c.html</link>', $content);

		$this->assertContains('<title>ItemA</title>', $content);
		$this->assertContains('<title>ItemB</title>', $content);
		$this->assertContains('<title>ItemC</title>', $content);

		$this->assertContains('<description>ItemA Content</description>', $content);
		$this->assertContains('<description>ItemB Content</description>', $content);
		$this->assertContains('<description>ItemC Content</description>', $content);


		// Feed #2 - put Content() into <title> and AltContent() into <description>
		$rssFeed = new RSSFeed($list, "http://www.example.com", "Test RSS Feed", "Test RSS Feed Description",
			"Content", "AltContent");
		$content = $rssFeed->outputToBrowser();

		$this->assertContains('<title>ItemA Content</title>', $content);
		$this->assertContains('<title>ItemB Content</title>', $content);
		$this->assertContains('<title>ItemC Content</title>', $content);

		$this->assertContains('<description>ItemA AltContent</description>', $content);
		$this->assertContains('<description>ItemB AltContent</description>', $content);
		$this->assertContains('<description>ItemC AltContent</description>', $content);
	}

	public function testLinkEncoding() {
		$list = new ArrayList();
		$rssFeed = new RSSFeed($list, "http://www.example.com/?param1=true&param2=true", "Test RSS Feed");
		$content = $rssFeed->outputToBrowser();
		$this->assertContains('<link>http://www.example.com/?param1=true&amp;param2=true', $content);
	}

	public function testRSSFeedWithShortcode() {
		$list = new ArrayList();
		$list->push(new RSSFeedTest_ItemD());

		$rssFeed = new RSSFeed($list, "http://www.example.com", "Test RSS Feed", "Test RSS Feed Description");
		$content = $rssFeed->outputToBrowser();

		$this->assertContains('<link>http://www.example.org/item-d.html</link>', $content);

		$this->assertContains('<title>ItemD</title>', $content);

		$this->assertContains(
			'<description>&lt;p&gt;ItemD Content test shortcode output&lt;/p&gt;</description>',
			$content
		);
	}

	public function testRenderWithTemplate() {
		$rssFeed = new RSSFeed(new ArrayList(), "", "", "");
		$rssFeed->setTemplate('RSSFeedTest');

		$content = $rssFeed->outputToBrowser();
		$this->assertContains('<title>Test Custom Template</title>', $content);

		$rssFeed->setTemplate('RSSFeed');
		$content = $rssFeed->outputToBrowser();
		$this->assertNotContains('<title>Test Custom Template</title>', $content);
	}

	public function setUp() {
		parent::setUp();
		Config::inst()->update('Director', 'alternate_base_url', '/');
		if(!self::$original_host) self::$original_host = $_SERVER['HTTP_HOST'];
		$_SERVER['HTTP_HOST'] = 'www.example.org';

		ShortcodeParser::get('default')->register('test_shortcode', function() {
			return 'test shortcode output';
		});
	}

	public function tearDown() {
		parent::tearDown();
		Config::inst()->update('Director', 'alternate_base_url', null);
		$_SERVER['HTTP_HOST'] = self::$original_host;
	}
}

class RSSFeedTest_ItemA extends ViewableData {
	// RSS-feed items must have $casting/$db information.
	private static $casting = array(
		'Title' => 'Varchar',
		'Content' => 'Text',
		'AltContent' => 'Text',
	);

	public function getTitle() {
		return "ItemA";
	}

	public function getContent() {
		return "ItemA Content";
	}

	public function getAltContent() {
		return "ItemA AltContent";
	}

	public function Link($action = null) {
		return Controller::join_links("item-a/", $action);
	}
}

class RSSFeedTest_ItemB extends ViewableData {
	// ItemB tests without $casting

	public function Title() {
		return "ItemB";
	}

	public function AbsoluteLink() {
		return "http://www.example.com/item-b.html";
	}

	public function Content() {
		return "ItemB Content";
	}

	public function AltContent() {
		return "ItemB AltContent";
	}
}

class RSSFeedTest_ItemC extends ViewableData {
	// ItemC tests fields - Title has casting, Content doesn't.
	private static $casting = array(
		'Title' => 'Varchar',
		'AltContent' => 'Text',
	);

	public $Title = "ItemC";
	public $Content = "ItemC Content";
	public $AltContent = "ItemC AltContent";

	public function Link() {
		return "item-c.html";
	}

	public function AbsoluteLink() {
		return "http://www.example.com/item-c.html";
	}
}

class RSSFeedTest_ItemD extends ViewableData {
	// ItemD test fields - all fields use casting but Content & AltContent cast as HTMLText
	private static $casting = array(
		'Title' => 'Varchar',
		'Content' => 'HTMLText'
	);

	public $Title = 'ItemD';
	public $Content = '<p>ItemD Content [test_shortcode]</p>';

	public function Link() {
		return 'item-d.html';
	}

	public function AbsoluteLink() {
		return 'http://www.example.org/item-d.html';
	}
}