mediawiki-extensions-Echo/tests/phpunit/model/TargetPageTest.php

<?php

class EchoTargetPageTest extends MediaWikiTestCase {

	public function testCreate() {
		$this->assertNull(
			EchoTargetPage::create(
				$this->mockTitle( 0 ),
				$this->mockEchoEvent()
			)
		);

		$this->assertInstanceOf(
			'EchoTargetPage',
			EchoTargetPage::create(
				$this->mockTitle( 1 ),
				$this->mockEchoEvent()
			)
		);
	}

	public function testNewFromRow() {
		$row = (object)[
			'etp_page' => 2,
			'etp_event' => 3
		];
		$obj = EchoTargetPage::newFromRow( $row );
		$this->assertInstanceOf( 'EchoTargetPage', $obj );

		return $obj;
	}

	/**
	 * @expectedException MWException
	 */
	public function testNewFromRowWithException() {
		$row = (object)[
			'etp_event' => 3
		];
		$this->assertInstanceOf( 'EchoTargetPage', EchoTargetPage::newFromRow( $row ) );
	}

	/**
	 * @depends testNewFromRow
	 */
	public function testToDbArray( $obj ) {
		$row = $obj->toDbArray();
		$this->assertTrue( is_array( $row ) );

		// Not very common to assert that a field does _not_ exist
		// but since we are explicitly removing it, it seems to make sense.
		$this->assertArrayNotHasKey( 'etp_user', $row );

		$this->assertArrayHasKey( 'etp_page', $row );
		$this->assertArrayHasKey( 'etp_event', $row );
	}

	/**
	 * Mock object of Title
	 */
	protected function mockTitle( $pageId ) {
		$event = $this->getMockBuilder( 'Title' )
			->disableOriginalConstructor()
			->getMock();
		$event->expects( $this->any() )
			->method( 'getArticleID' )
			->will( $this->returnValue( $pageId ) );

		return $event;
	}

	/**
	 * Mock object of EchoEvent
	 */
	protected function mockEchoEvent( $eventId = 1 ) {
		$event = $this->getMockBuilder( 'EchoEvent' )
			->disableOriginalConstructor()
			->getMock();
		$event->expects( $this->any() )
			->method( 'getId' )
			->will( $this->returnValue( $eventId ) );

		return $event;
	}

}