PDFReportsTest.php

<?php
/**
 * Piwik - Open source web analytics
 *
 * @link http://piwik.org
 * @license http://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
 */
require_once 'PDFReports/PDFReports.php';

class PDFReportsTest extends DatabaseTestCase
{
	private $idSite = 1;

	public function setUp()
	{
		parent::setUp();

		// setup the access layer
		self::setSuperUser();
		Piwik_PluginsManager::getInstance()->loadPlugins(array('API', 'UserCountry', 'PDFReports', 'MobileMessaging'));
		Piwik_PluginsManager::getInstance()->installLoadedPlugins();

		Piwik_SitesManager_API::getInstance()->addSite("Test", array("http://piwik.net"));

		Piwik_SitesManager_API::getInstance()->addSite("Test", array("http://piwik.net"));
		FakeAccess::setIdSitesView(array($this->idSite, 2));
		Piwik_PDFReports_API::$cache = array();
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testAddReportGetReports()
	{
		$data = array(
			'idsite' => $this->idSite,
			'description' => 'test description"',
			'type' => 'email',
			'period' => Piwik_ScheduledTime::PERIOD_DAY,
			'hour' => '4',
			'format' => 'pdf',
			'reports' => array('UserCountry_getCountry'),
			'parameters' => array(
				'displayFormat' => '1',
				'emailMe' => true,
				'additionalEmails' => array('test@test.com', 't2@test.com'),
				'evolutionGraph' => true
			)
		);

		$dataWebsiteTwo = $data;
		$dataWebsiteTwo['idsite'] = 2;
		$dataWebsiteTwo['period'] = Piwik_ScheduledTime::PERIOD_MONTH;

		self::addReport($dataWebsiteTwo);

		// Testing getReports without parameters
		$tmp = Piwik_PDFReports_API::getInstance()->getReports();
		$report = reset($tmp);
		$this->assertReportsEqual($report, $dataWebsiteTwo);

		$idReport = self::addReport($data);

		// Passing 3 parameters
		$tmp = Piwik_PDFReports_API::getInstance()->getReports($this->idSite, $data['period'], $idReport);
		$report = reset($tmp);
		$this->assertReportsEqual($report, $data);

		// Passing only idsite
		$tmp = Piwik_PDFReports_API::getInstance()->getReports($this->idSite);
		$report = reset($tmp);
		$this->assertReportsEqual($report, $data);

		// Passing only period
		$tmp = Piwik_PDFReports_API::getInstance()->getReports($idSite = false, $data['period']);
		$report = reset($tmp);
		$this->assertReportsEqual($report, $data);

		// Passing only idreport
		$tmp = Piwik_PDFReports_API::getInstance()->getReports($idSite = false, $period = false, $idReport);
		$report = reset($tmp);
		$this->assertReportsEqual($report, $data);
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetReportsIdReportNotFound()
	{
		try
		{
			Piwik_PDFReports_API::getInstance()->getReports($idSite = false, $period = false, $idReport = 1);
		} catch (Exception $e) {
			return;
		}
		$this->fail('Expected exception not raised');
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetReportsInvalidPermission()
	{
		try
		{
			Piwik_PDFReports_API::getInstance()->getReports(
				$idSite = 44,
				$period = false,
				self::addReport(self::getDailyPDFReportData($this->idSite))
			);

		} catch (Exception $e) {
			return;
		}
		$this->fail('Expected exception not raised');
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testAddReportInvalidWebsite()
	{
		try
		{
			self::addReport(self::getDailyPDFReportData(33));
		} catch (Exception $e) {
			return;
		}
		$this->fail('Expected exception not raised');
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testAddReportInvalidPeriod()
	{
		try
		{
			$data = self::getDailyPDFReportData($this->idSite);
			$data['period'] = 'dx';
			self::addReport($data);
		} catch (Exception $e) {
			return;
		}
		$this->fail('Expected exception not raised');
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testUpdateReport()
	{
		$idReport = self::addReport(self::getDailyPDFReportData($this->idSite));
		$dataAfter = self::getMonthlyEmailReportData($this->idSite);

		self::updateReport($idReport, $dataAfter);

		$reports = Piwik_PDFReports_API::getInstance()->getReports($idSite = false, $period = false, $idReport);

		$this->assertReportsEqual(
			reset($reports),
			$dataAfter
		);
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testDeleteReport()
	{
		// Deletes non existing report throws exception
		try {
			Piwik_PDFReports_API::getInstance()->deleteReport($idReport = 1);
			$this->fail('Exception not raised');
		} catch (Exception $e) {
		}

		$idReport = self::addReport(self::getMonthlyEmailReportData($this->idSite));
		$this->assertEquals(1, count(Piwik_PDFReports_API::getInstance()->getReports()));
		Piwik_PDFReports_API::getInstance()->deleteReport($idReport);
		$this->assertEquals(0, count(Piwik_PDFReports_API::getInstance()->getReports()));
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetTopMenuTranslationKeyMobileMessagingInactive()
	{
		// unload MobileMessaging plugin
		Piwik_PluginsManager::getInstance()->loadPlugins(array('PDFReports'));

		$pdfReportPlugin = new Piwik_PDFReports();
		$this->assertEquals(
			Piwik_PDFReports::PDF_REPORTS_TOP_MENU_TRANSLATION_KEY,
			$pdfReportPlugin->getTopMenuTranslationKey()
		);
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetTopMenuTranslationKeyUserIsAnonymous()
	{
		$anonymousAccess = new FakeAccess;
		FakeAccess::$identity = 'anonymous';
		Zend_Registry::set('access', $anonymousAccess);

		$pdfReportPlugin = new Piwik_PDFReports();
		$this->assertEquals(
			Piwik_PDFReports::MOBILE_MESSAGING_TOP_MENU_TRANSLATION_KEY,
			$pdfReportPlugin->getTopMenuTranslationKey()
		);
	}

	/**
	 * top menu should display 'Email & SMS reports' when the user has set-up a valid mobile provider account
	 * even though there is no sms reports configured
	 *
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetTopMenuTranslationKeyNoReportMobileAccountOK()
	{
		// set mobile provider account
		self::setSuperUser();
		Piwik_MobileMessaging_API::getInstance()->setSMSAPICredential('StubbedProvider', '');

		$pdfReportPlugin = new Piwik_PDFReports();
		$this->assertEquals(
			Piwik_PDFReports::MOBILE_MESSAGING_TOP_MENU_TRANSLATION_KEY,
			$pdfReportPlugin->getTopMenuTranslationKey()
		);
	}

	/**
	 * top menu should display 'Email reports' when the user has not set-up a valid mobile provider account
	 * and no reports at all have been configured
	 *
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetTopMenuTranslationKeyNoReportMobileAccountKO()
	{
		$pdfReportPlugin = new Piwik_PDFReports();
		$this->assertEquals(
			Piwik_PDFReports::PDF_REPORTS_TOP_MENU_TRANSLATION_KEY,
			$pdfReportPlugin->getTopMenuTranslationKey()
		);
	}

	/**
	 * top menu should display 'Email & SMS reports' if there is at least one sms report
	 * whatever the status of the mobile provider account
	 *
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetTopMenuTranslationKeyOneSMSReportMobileAccountKO()
	{
		Piwik_PDFReports_API::getInstance()->addReport(
			1,
			'',
			Piwik_ScheduledTime::PERIOD_DAY,
			0,
			Piwik_MobileMessaging::MOBILE_TYPE,
			Piwik_MobileMessaging::SMS_FORMAT,
			array(),
			array(
				Piwik_MobileMessaging::PHONE_NUMBERS_PARAMETER => array()
			)
		);

		$pdfReportPlugin = new Piwik_PDFReports();
		$this->assertEquals(
			Piwik_PDFReports::MOBILE_MESSAGING_TOP_MENU_TRANSLATION_KEY,
			$pdfReportPlugin->getTopMenuTranslationKey()
		);
	}

	/**
	 * top menu should display 'Email reports' if there are no SMS reports and at least one email report
	 * whatever the status of the mobile provider account
	 *
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetTopMenuTranslationKeyNoSMSReportAccountOK()
	{
		// set mobile provider account
		self::setSuperUser();
		Piwik_MobileMessaging_API::getInstance()->setSMSAPICredential('StubbedProvider', '');

		self::addReport(self::getMonthlyEmailReportData($this->idSite));

		$pdfReportPlugin = new Piwik_PDFReports();
		$this->assertEquals(
			Piwik_PDFReports::PDF_REPORTS_TOP_MENU_TRANSLATION_KEY,
			$pdfReportPlugin->getTopMenuTranslationKey()
		);
	}

	/**
	 * @group Plugins
	 * @group PDFReports
	 */
	public function testGetScheduledTasks()
	{
		// stub Piwik_PDFReports_API to control getReports() return values
		$report1 = self::getDailyPDFReportData($this->idSite);
		$report1['idreport'] = 1;
		$report1['hour'] = 0;
		$report1['deleted'] = 0;

		$report2 = self::getMonthlyEmailReportData($this->idSite);
		$report2['idreport'] = 2;
		$report2['idsite'] = 2;
		$report2['hour'] = 0;
		$report2['deleted'] = 0;

		$report3 = self::getMonthlyEmailReportData($this->idSite);
		$report3['idreport'] = 3;
		$report3['deleted'] = 1; // should not be scheduled

		$report4 = self::getMonthlyEmailReportData($this->idSite);
		$report4['idreport'] = 4;
		$report4['idsite'] = 1;
		$report4['hour'] = 8;
		$report4['deleted'] = 0;

		$report5 = self::getMonthlyEmailReportData($this->idSite);
		$report5['idreport'] = 5;
		$report5['idsite'] = 2;
		$report5['hour'] = 8;
		$report5['deleted'] = 0;

		// test no exception is raised when a scheduled report is set to never send
		$report6 = self::getMonthlyEmailReportData($this->idSite);
		$report6['idreport'] = 6;
		$report6['period'] = Piwik_ScheduledTime::PERIOD_NEVER;
		$report6['deleted'] = 0;

		$stubbedPDFReportsAPI = $this->getMock('Piwik_PDFReports_API');
		$stubbedPDFReportsAPI->expects($this->any())->method('getReports')->will($this->returnValue(
			array($report1, $report2, $report3, $report4, $report5, $report6))
		);

		$stubbedPDFReportsAPIClass = new ReflectionProperty('Piwik_PDFReports_API', 'instance');
		$stubbedPDFReportsAPIClass->setAccessible(true);
		$stubbedPDFReportsAPIClass->setValue($stubbedPDFReportsAPI);

		// initialize sites 1 and 2
		Piwik_Site::$infoSites = array(
			1 => array('timezone' => 'Europe/Paris'),
			2 => array('timezone' => 'UTC-6.5'),
		);

		// expected tasks
		$scheduleTask1 = new Piwik_ScheduledTime_Daily();
		$scheduleTask1->setHour(23); // paris is UTC-1, period ends at 23h UTC

		$scheduleTask2 = new Piwik_ScheduledTime_Monthly();
		$scheduleTask2->setHour(7); // site is UTC-6.5, period ends at 6h30 UTC, smallest resolution is hour

		$scheduleTask3 = new Piwik_ScheduledTime_Monthly();
		$scheduleTask3->setHour(7); // paris is UTC-1, configured to be sent at 8h

		$scheduleTask4 = new Piwik_ScheduledTime_Monthly();
		$scheduleTask4->setHour(15); // site is UTC-6.5, configured to be sent at 8h

		$expectedTasks = array(
			new Piwik_ScheduledTask (Piwik_PDFReports_API::getInstance(), 'sendReport', 1, $scheduleTask1),
			new Piwik_ScheduledTask (Piwik_PDFReports_API::getInstance(), 'sendReport', 2, $scheduleTask2),
			new Piwik_ScheduledTask (Piwik_PDFReports_API::getInstance(), 'sendReport', 4, $scheduleTask3),
			new Piwik_ScheduledTask (Piwik_PDFReports_API::getInstance(), 'sendReport', 5, $scheduleTask4),
		);

		$pdfReportPlugin = new Piwik_PDFReports();
		$tasks = array();
		$pdfReportPlugin->getScheduledTasks(new Piwik_Event_Notification($tasks, 'fakeEvent'));
		$this->assertEquals($expectedTasks, $tasks);

		// restore Piwik_PDFReports_API
		$stubbedPDFReportsAPIClass->setValue(null);
	}

	private function assertReportsEqual($report, $data)
	{
		foreach ($data as $key => $value)
		{
			if ($key == 'description') $value = substr($value, 0, 250);
			$this->assertEquals($value, $report[$key], "Error for $key for report " . var_export($report, true) . " and data " . var_export($data, true));
		}
	}

	private static function addReport($data)
	{
		$idReport = Piwik_PDFReports_API::getInstance()->addReport(
			$data['idsite'],
			$data['description'],
			$data['period'],
			$data['hour'],
			$data['type'],
			$data['format'],
			$data['reports'],
			$data['parameters']
		);
		return $idReport;
	}

	private static function getDailyPDFReportData($idSite)
	{
		return array(
			'idsite' => $idSite,
			'description' => 'test description"',
			'period' => Piwik_ScheduledTime::PERIOD_DAY,
			'hour' => '7',
			'type' => 'email',
			'format' => 'pdf',
			'reports' => array('UserCountry_getCountry'),
			'parameters' => array(
				'displayFormat' => '1',
				'emailMe' => true,
				'additionalEmails' => array('test@test.com', 't2@test.com'),
				'evolutionGraph' => false
			)
		);
	}

	private static function getMonthlyEmailReportData($idSite)
	{
		return array(
			'idsite' => $idSite,
			'description' => 'very very long and possibly truncated description. very very long and possibly truncated description. very very long and possibly truncated description. very very long and possibly truncated description. very very long and possibly truncated description. ',
			'period' => Piwik_ScheduledTime::PERIOD_MONTH,
			'hour' => '0',
			'type' => 'email',
			'format' => 'pdf',
			'reports' => array('UserCountry_getContinent'),
			'parameters' => array(
				'displayFormat' => '1',
				'emailMe' => false,
				'additionalEmails' => array('blabla@ec.fr'),
				'evolutionGraph' => false
			)
		);
	}

	private static function updateReport($idReport, $data)
	{
		$idReport = Piwik_PDFReports_API::getInstance()->updateReport(
			$idReport,
			$data['idsite'],
			$data['description'],
			$data['period'],
			$data['hour'],
			$data['type'],
			$data['format'],
			$data['reports'],
			$data['parameters']);
		return $idReport;
	}

	private static function setSuperUser()
	{
		$pseudoMockAccess = new FakeAccess;
		FakeAccess::$superUser = true;
		Zend_Registry::set('access', $pseudoMockAccess);
	}
}