1 files changed, 51 insertions, 0 deletions
diff --git a/www/wiki/tests/phpunit/includes/api/ApiSetNotificationTimestampIntegrationTest.php b/www/wiki/tests/phpunit/includes/api/ApiSetNotificationTimestampIntegrationTest.php
new file mode 100644
index 00000000..dacd48f6
--- /dev/null
+++ b/www/wiki/tests/phpunit/includes/api/ApiSetNotificationTimestampIntegrationTest.php
@@ -0,0 +1,51 @@
+<?php
+use MediaWiki\MediaWikiServices;
+
+/**
+ * @author Addshore
+ * @covers ApiSetNotificationTimestamp
+ * @group API
+ * @group medium
+ * @group Database
+ */
+class ApiSetNotificationTimestampIntegrationTest extends ApiTestCase {
+
+	protected function setUp() {
+		parent::setUp();
+		self::$users[__CLASS__] = new TestUser( __CLASS__ );
+	}
+
+	public function testStuff() {
+		$user = self::$users[__CLASS__]->getUser();
+		$page = WikiPage::factory( Title::newFromText( 'UTPage' ) );
+
+		$user->addWatch( $page->getTitle() );
+
+		$result = $this->doApiRequestWithToken(
+			[
+				'action' => 'setnotificationtimestamp',
+				'timestamp' => '20160101020202',
+				'pageids' => $page->getId(),
+			],
+			null,
+			$user
+		);
+
+		$this->assertEquals(
+			[
+				'batchcomplete' => true,
+				'setnotificationtimestamp' => [
+					[ 'ns' => 0, 'title' => 'UTPage', 'notificationtimestamp' => '2016-01-01T02:02:02Z' ]
+				],
+			],
+			$result[0]
+		);
+
+		$watchedItemStore = MediaWikiServices::getInstance()->getWatchedItemStore();
+		$this->assertEquals(
+			$watchedItemStore->getNotificationTimestampsBatch( $user, [ $page->getTitle() ] ),
+			[ [ 'UTPage' => '20160101020202' ] ]
+		);
+	}
+
+}