summaryrefslogtreecommitdiff
path: root/www/wiki/extensions/SemanticMediaWiki/tests/phpunit/Unit/MediaWiki/Connection/SequenceTest.php
blob: 3ec0ab44d3ff5f1096de14d3358b2ce08d68e880 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
<?php

namespace SMW\Tests\MediaWiki\Connection;

use SMW\MediaWiki\Connection\Sequence;
use SMW\Tests\PHPUnitCompat;

/**
 * @covers \SMW\MediaWiki\Connection\Sequence
 * @group semantic-mediawiki
 *
 * @license GNU GPL v2+
 * @since 3.0
 *
 * @author mwjames
 */
class SequenceTest extends \PHPUnit_Framework_TestCase {

	use PHPUnitCompat;

	private $connection;

	protected function setUp() {

		$this->connection = $this->getMockBuilder( '\SMW\MediaWiki\Database' )
			->disableOriginalConstructor()
			->getMock();
	}

	public function testCanConstruct() {
		$this->assertInstanceOf(
			Sequence::class,
			new Sequence( $this->connection )
		);
	}

	public function testConstructWithInvalidConnectionThrowsException() {
		$this->setExpectedException( '\RuntimeException' );
		new Sequence( 'Foo' );
	}

	public function testMakeSequence() {
		$this->assertEquals(
			'Foo_bar_seq',
			Sequence::makeSequence( 'Foo', 'bar' )
		);
	}

	public function testNonPostgres() {

		$this->connection->expects( $this->once() )
			->method( 'getType' )
			->will( $this->returnValue( 'foo' ) );

		$instance = new Sequence(
			$this->connection
		);

		$this->assertEquals(
			null,
			$instance->restart( 'Foo', 'bar')
		);
	}

	public function testPostgres() {

		$this->connection->expects( $this->once() )
			->method( 'getType' )
			->will( $this->returnValue( 'postgres' ) );

		$this->connection->expects( $this->once() )
			->method( 'onTransactionIdle' )
			->will( $this->returnCallback( function( $callback ) { return $callback(); } ) );

		$this->connection->expects( $this->once() )
			->method( 'query' )
			->with( $this->equalTo( 'ALTER SEQUENCE Foo_bar_seq RESTART WITH 43' ) );

		$this->connection->expects( $this->once() )
			->method( 'selectField' )
			->will( $this->returnValue( 42 ) );

		$instance = new Sequence(
			$this->connection
		);

		$this->assertEquals(
			43,
			$instance->restart( 'Foo', 'bar' )
		);
	}

}