1 files changed, 99 insertions, 0 deletions
diff --git a/www/wiki/extensions/Translate/tests/phpunit/TTMServerTest.php b/www/wiki/extensions/Translate/tests/phpunit/TTMServerTest.php
new file mode 100644
index 00000000..ed434ddc
--- /dev/null
+++ b/www/wiki/extensions/Translate/tests/phpunit/TTMServerTest.php
@@ -0,0 +1,99 @@
+<?php
+/**
+ * @file
+ * @author Niklas Laxström
+ * @copyright Copyright © 2012-2013, Niklas Laxström
+ * @license GPL-2.0-or-later
+ */
+
+class TTMServerTest extends MediaWikiTestCase {
+	protected $config;
+
+	protected function setUp() {
+		global $wgTranslateTranslationServices;
+		$this->config = $wgTranslateTranslationServices;
+		parent::setUp();
+
+		$wgTranslateTranslationServices = [];
+		$wgTranslateTranslationServices['localtm'] = [
+			'url' => 'http://example.com/sandwiki/api.php',
+			'displayname' => 'example.com',
+			'cutoff' => 0.75,
+			'type' => 'ttmserver',
+		];
+
+		$wgTranslateTranslationServices['apitm'] = [
+			'url' => 'http://example.com/w/api.php',
+			'displayname' => 'example.com',
+			'cutoff' => 0.75,
+			'timeout-sync' => 4,
+			'timeout-async' => 4,
+			'type' => 'ttmserver',
+			'class' => 'RemoteTTMServer',
+		];
+	}
+
+	protected function tearDown() {
+		global $wgTranslateTranslationServices;
+		$wgTranslateTranslationServices = $this->config;
+		parent::tearDown();
+	}
+
+	public function testConstruct() {
+		$server = TTMServer::primary();
+		$this->assertEquals(
+			'FakeTTMServer',
+			get_class( $server ),
+			'Fake server given when default server is disabled'
+		);
+		global $wgTranslateTranslationServices,
+			$wgTranslateTranslationDefaultService;
+		$wgTranslateTranslationServices[$wgTranslateTranslationDefaultService] = [
+			'database' => false, // Passed to wfGetDB
+			'cutoff' => 0.75,
+			'type' => 'ttmserver',
+			'public' => false,
+		];
+		$server = TTMServer::primary();
+		$this->assertEquals(
+			'DatabaseTTMServer',
+			get_class( $server ),
+			'Real server given when default server is enabled'
+		);
+		unset( $wgTranslateTranslationServices[$wgTranslateTranslationDefaultService] );
+	}
+
+	public function testFakeTTMServer() {
+		$server = new FakeTTMServer();
+		$this->assertEquals(
+			[],
+			$server->query( 'en', 'fi', 'daa' ),
+			'FakeTTMServer returns no suggestions for all queries'
+		);
+
+		$title = new Title();
+		$handle = new MessageHandle( $title );
+
+		$this->assertNull(
+			$server->update( $handle, 'text' ),
+			'FakeTTMServer returns null on update'
+		);
+	}
+
+	public function testMirrorsConfig() {
+		global $wgTranslateTranslationServices;
+		$wgTranslateTranslationServices['primary'] = [
+			'class' => 'ElasticSearchTTMServer',
+			'mirrors' => [ 'secondary' ]
+		];
+		$wgTranslateTranslationServices['secondary'] = [
+			'class' => 'ElasticSearchTTMServer',
+			'mirrors' => [ 'primary', 'unknown' ]
+		];
+		$primary = TTMServer::factory( $wgTranslateTranslationServices['primary'] );
+		$this->assertEquals( [ 'secondary' ], $primary->getMirrors() );
+		$secondary = TTMServer::factory( $wgTranslateTranslationServices['secondary'] );
+		$this->setExpectedException( TTMServerException::class );
+		$secondary->getMirrors();
+	}
+}