X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=tests%2Fphpunit%2Fincludes%2Flibs%2FSamplingStatsdClientTest.php;h=3439df8b0e63e067a1a639f1e02a9be49de29756;hb=823a96e2ceab3a552075bde676c03c2cb163ce8f;hp=1ebe55110f99bc13c630c9462244462a7fe21ff1;hpb=a85d1b9d0cd7f02111d3a647d5b91e3b1f334563;p=lhc%2Fweb%2Fwiklou.git

diff --git a/tests/phpunit/includes/libs/SamplingStatsdClientTest.php b/tests/phpunit/includes/libs/SamplingStatsdClientTest.php
index 1ebe55110f..3439df8b0e 100644
--- a/tests/phpunit/includes/libs/SamplingStatsdClientTest.php
+++ b/tests/phpunit/includes/libs/SamplingStatsdClientTest.php
@@ -1,13 +1,17 @@
 <?php
 
 use Liuggio\StatsdClient\Entity\StatsdData;
+use Liuggio\StatsdClient\Sender\SenderInterface;
 
+/**
+ * @covers SamplingStatsdClient
+ */
 class SamplingStatsdClientTest extends PHPUnit_Framework_TestCase {
 	/**
 	 * @dataProvider samplingDataProvider
 	 */
 	public function testSampling( $data, $sampleRate, $seed, $expectWrite ) {
-		$sender = $this->getMock( 'Liuggio\StatsdClient\Sender\SenderInterface' );
+		$sender = $this->getMockBuilder( SenderInterface::class )->getMock();
 		$sender->expects( $this->any() )->method( 'open' )->will( $this->returnValue( true ) );
 		if ( $expectWrite ) {
 			$sender->expects( $this->once() )->method( 'write' )
@@ -32,12 +36,35 @@ class SamplingStatsdClientTest extends PHPUnit_Framework_TestCase {
 
 		return [
 			// $data, $sampleRate, $seed, $expectWrite
-			[ $unsampled, 1, 0 /*0.44*/, $unsampled ],
-			[ $sampled, 1, 0 /*0.44*/, null ],
-			[ $sampled, 1, 4 /*0.03*/, $sampled ],
-			[ $unsampled, 0.1, 4 /*0.03*/, $sampled ],
-			[ $sampled, 0.5, 0 /*0.44*/, null ],
-			[ $sampled, 0.5, 4 /*0.03*/, $sampled ],
+			[ $unsampled, 1, 0 /*0.44*/, true ],
+			[ $sampled, 1, 0 /*0.44*/, false ],
+			[ $sampled, 1, 4 /*0.03*/, true ],
+			[ $unsampled, 0.1, 0 /*0.44*/, false ],
+			[ $sampled, 0.5, 0 /*0.44*/, false ],
+			[ $sampled, 0.5, 4 /*0.03*/, false ],
 		];
 	}
+
+	public function testSetSamplingRates() {
+		$matching = new StatsdData();
+		$matching->setKey( 'foo.bar' );
+		$matching->setValue( 1 );
+
+		$nonMatching = new StatsdData();
+		$nonMatching->setKey( 'oof.bar' );
+		$nonMatching->setValue( 1 );
+
+		$sender = $this->getMockBuilder( SenderInterface::class )->getMock();
+		$sender->expects( $this->any() )->method( 'open' )->will( $this->returnValue( true ) );
+		$sender->expects( $this->once() )->method( 'write' )->with( $this->anything(),
+			$this->equalTo( $nonMatching ) );
+
+		$client = new SamplingStatsdClient( $sender );
+		$client->setSamplingRates( [ 'foo.*' => 0.2 ] );
+
+		mt_srand( 0 ); // next random is 0.44
+		$client->send( $matching );
+		mt_srand( 0 );
+		$client->send( $nonMatching );
+	}
 }