1 files changed, 120 insertions, 0 deletions
diff --git a/www/wiki/extensions/Translate/tests/phpunit/TPSectionTest.php b/www/wiki/extensions/Translate/tests/phpunit/TPSectionTest.php
new file mode 100644
index 00000000..ccd9f300
--- /dev/null
+++ b/www/wiki/extensions/Translate/tests/phpunit/TPSectionTest.php
@@ -0,0 +1,120 @@
+<?php
+/**
+ * @author Niklas Laxström
+ * @license GPL-2.0-or-later
+ * @file
+ */
+
+/**
+ * @ingroup PageTranslation
+ */
+class TPSectionTest extends PHPUnit\Framework\TestCase {
+	/**
+	 * @dataProvider providerTestGetMarkedText
+	 */
+	public function testGetMarkedText( $name, $text, $inline, $expected ) {
+		$section = new TPSection();
+		$section->name = $name;
+		$section->text = $text;
+		$section->setIsInline( $inline );
+
+		$output = $section->getMarkedText();
+
+		$this->assertEquals( $expected, $output );
+	}
+
+	/**
+	 * @dataProvider providerTestGetTextWithVariables
+	 */
+	public function testGetTextWithVariables( $text, $expected ) {
+		$section = new TPSection();
+		$section->text = $text;
+
+		$output = $section->getTextWithVariables();
+
+		$this->assertEquals( $expected, $output );
+	}
+
+	/**
+	 * @dataProvider providerTestGetTextForTrans
+	 */
+	public function testGetTextForTrans( $text, $expected ) {
+		$section = new TPSection();
+		$section->text = $text;
+
+		$output = $section->getTextForTrans();
+
+		$this->assertEquals( $expected, $output );
+	}
+
+	public static function providerTestGetMarkedText() {
+		$cases = [];
+
+		// Inline syntax
+		$cases[] = [
+			'name',
+			'Hello',
+			true,
+			'<!--T:name--> Hello',
+		];
+
+		// Normal syntax
+		$cases[] = [
+			'name',
+			'Hello',
+			false,
+			"<!--T:name-->\nHello",
+		];
+
+		// Inline should not matter for headings, which have special syntax, but test both values
+		$cases[] = [
+			'name',
+			'== Hello ==',
+			true,
+			'== Hello == <!--T:name-->',
+		];
+
+		$cases[] = [
+			'name',
+			'====== Hello ======',
+			false,
+			'====== Hello ====== <!--T:name-->',
+		];
+
+		return $cases;
+	}
+
+	public static function providerTestGetTextWithVariables() {
+		$cases = [];
+
+		// syntax
+		$cases[] = [
+			"<tvar|abc>Peter\n cat!</>",
+			'$abc',
+		];
+
+		$cases[] = [
+			"<tvar|1>Hello</>\n<tvar|2>Hello</>",
+			"$1\n$2",
+		];
+
+		return $cases;
+	}
+
+	public static function providerTestGetTextForTrans() {
+		$cases = [];
+
+		// syntax
+		$cases[] = [
+			"<tvar|abc>Peter\n cat!</>",
+			"Peter\n cat!",
+		];
+
+		$cases[] = [
+			"<tvar|1>Hello</>\n<tvar|2>Hello</>",
+			"Hello\nHello",
+		];
+
+		return $cases;
+	}
+}