Engine.onCompile.phpt

<?php

/**
 * Test: Latte\Engine: {onCompile}
 */

use Tester\Assert;


require __DIR__ . '/../bootstrap.php';


test(function () {
	$latte = new Latte\Engine;
	$latte->setLoader(new Latte\Loaders\StringLoader);

	$latte->onCompile[] = function (\Latte\Engine $engine) {
		Notes::add('adding macros 1');
	};

	$latte->onCompile[] = function (\Latte\Engine $engine) {
		Notes::add('adding macros 2');
	};

	Assert::match('%A%$var%A%', $latte->compile('{$var}'));
	Assert::match('%A%$var%A%', $latte->compile('{$var}'));

	Assert::same(array(
		'adding macros 1',
		'adding macros 2',
	), Notes::fetch());
});


test(function () {
	$callbacks = array();
	$callbacks[] = function (\Latte\Engine $engine) {
		Notes::add('adding macros 3');
	};

	$callbacks[] = function (\Latte\Engine $engine) {
		Notes::add('adding macros 4');
	};

	$latte = new Latte\Engine;
	$latte->setLoader(new Latte\Loaders\StringLoader);
	$latte->onCompile = new ArrayIterator($callbacks);

	Assert::match('%A%$var%A%', $latte->compile('{$var}'));
	Assert::match('%A%$var%A%', $latte->compile('{$var}'));

	Assert::same(array(
		'adding macros 3',
		'adding macros 4',
	), Notes::fetch());
});


test(function () {
	class Event implements IteratorAggregate
	{
		public $events;

		public function __construct($events)
		{
			$this->events = $events;
		}

		public function getIterator()
		{
			return new ArrayIterator($this->events);
		}
	}

	$callbacks = array();
	$callbacks[] = function (\Latte\Engine $engine) {
		Notes::add('adding macros 5');
	};

	$callbacks[] = function (\Latte\Engine $engine) {
		Notes::add('adding macros 6');
	};


	$latte = new Latte\Engine;
	$latte->setLoader(new Latte\Loaders\StringLoader);
	$latte->onCompile = new Event($callbacks);

	Assert::match('%A%$var%A%', $latte->compile('{$var}'));
	Assert::match('%A%$var%A%', $latte->compile('{$var}'));

	Assert::same(array(
		'adding macros 5',
		'adding macros 6',
	), Notes::fetch());
});