X-Git-Url: https://scripts.mit.edu/gitweb/autoinstallsdev/mediawiki.git/blobdiff_plain/d57edfddd6c01f0ed6b1a84019649cdf6cddd5f8..83d871ca0d985c6d586b323bf96161afb510ebf6:/tests/phpunit/includes/utils/ClassCollectorTest.php

diff --git a/tests/phpunit/includes/utils/ClassCollectorTest.php b/tests/phpunit/includes/utils/ClassCollectorTest.php
new file mode 100644
index 00000000..e8a228e3
--- /dev/null
+++ b/tests/phpunit/includes/utils/ClassCollectorTest.php
@@ -0,0 +1,48 @@
+<?php
+
+/**
+ * @covers ClassCollector
+ */
+class ClassCollectorTest extends PHPUnit_Framework_TestCase {
+
+	public static function provideCases() {
+		return [
+			[
+				"class Foo {}",
+				[ 'Foo' ],
+			],
+			[
+				"namespace Example;\nclass Foo {}\nclass Bar {}",
+				[ 'Example\Foo', 'Example\Bar' ],
+			],
+			[
+				"class_alias( 'Foo', 'Bar' );",
+				[ 'Bar' ],
+			],
+			[
+				"namespace Example;\nclass Foo {}\nclass_alias( 'Example\Foo', 'Foo' );",
+				[ 'Example\Foo', 'Foo' ],
+			],
+			[
+				"namespace Example;\nclass Foo {}\nclass_alias( 'Example\Foo', 'Bar' );",
+				[ 'Example\Foo', 'Bar' ],
+			],
+			[
+				"class_alias( Foo::class, 'Bar' );",
+				[ 'Bar' ],
+			],
+			[
+				"namespace Example;\nclass Foo {}\nclass_alias( Foo::class, 'Bar' );",
+				[ 'Example\Foo', 'Bar' ],
+			],
+		];
+	}
+
+	/**
+	 * @dataProvider provideCases
+	 */
+	public function testGetClasses( $code, array $classes, $message = null ) {
+		$cc = new ClassCollector();
+		$this->assertEquals( $classes, $cc->getClasses( "<?php\n$code" ), $message );
+	}
+}