summaryrefslogtreecommitdiff
path: root/tests/phpunit/includes/poolcounter/PoolCounterTest.php
blob: 019e532c930f9f88af31744a28ba690e1abafffa (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
<?php

// We will use this class with getMockForAbstractClass to create a concrete mock class.
// That call will die if the contructor is not public, unless we use disableOriginalConstructor(),
// in which case we could not test the constructor.
abstract class PoolCounterAbstractMock extends PoolCounter {
	public function __construct() {
		call_user_func_array( 'parent::__construct', func_get_args() );
	}
}

class PoolCounterTest extends MediaWikiTestCase {
	public function testConstruct() {
		$poolCounterConfig = array(
			'class' => 'PoolCounterMock',
			'timeout' => 10,
			'workers' => 10,
			'maxqueue' => 100,
		);

		$poolCounter = $this->getMockBuilder( 'PoolCounterAbstractMock' )
			->setConstructorArgs( array( $poolCounterConfig, 'testCounter', 'someKey' ) )
			// don't mock anything - the proper syntax would be setMethods(null), but due
			// to a PHPUnit bug that does not work with getMockForAbstractClass()
			->setMethods( array( 'idontexist' ) )
			->getMockForAbstractClass();
		$this->assertInstanceOf( 'PoolCounter', $poolCounter );
	}

	public function testConstructWithSlots() {
		$poolCounterConfig = array(
			'class' => 'PoolCounterMock',
			'timeout' => 10,
			'workers' => 10,
			'slots' => 2,
			'maxqueue' => 100,
		);

		$poolCounter = $this->getMockBuilder( 'PoolCounterAbstractMock' )
			->setConstructorArgs( array( $poolCounterConfig, 'testCounter', 'key' ) )
			->setMethods( array( 'idontexist' ) ) // don't mock anything
			->getMockForAbstractClass();
		$this->assertInstanceOf( 'PoolCounter', $poolCounter );
	}

	public function testHashKeyIntoSlots() {
		$poolCounter = $this->getMockBuilder( 'PoolCounterAbstractMock' )
			// don't mock anything - the proper syntax would be setMethods(null), but due
			// to a PHPUnit bug that does not work with getMockForAbstractClass()
			->setMethods( array( 'idontexist' ) )
			->disableOriginalConstructor()
			->getMockForAbstractClass();

		$hashKeyIntoSlots = new ReflectionMethod( $poolCounter, 'hashKeyIntoSlots' );
		$hashKeyIntoSlots->setAccessible( true );

		$keysWithTwoSlots = $keysWithFiveSlots = array();
		foreach ( range( 1, 100 ) as $i ) {
			$keysWithTwoSlots[] = $hashKeyIntoSlots->invoke( $poolCounter, 'key ' . $i, 2 );
			$keysWithFiveSlots[] = $hashKeyIntoSlots->invoke( $poolCounter, 'key ' . $i, 5 );
		}

		$this->assertArrayEquals( range( 0, 1 ), array_unique( $keysWithTwoSlots ) );
		$this->assertArrayEquals( range( 0, 4 ), array_unique( $keysWithFiveSlots ) );

		// make sure it is deterministic
		$this->assertEquals(
			$hashKeyIntoSlots->invoke( $poolCounter, 'asdfgh', 1000 ),
			$hashKeyIntoSlots->invoke( $poolCounter, 'asdfgh', 1000 )
		);
	}
}