From b3ce575e63a3e068da98322857890c7988ef8445 Mon Sep 17 00:00:00 2001
From: Mark Nelson <markn@moodle.com>
Date: Fri, 9 Nov 2018 13:20:37 +0800
Subject: [PATCH] MDL-63864 core_message: added unit tests

---
 message/tests/api_test.php         | 31 +++++++++++++++++++++++++
 message/tests/externallib_test.php | 37 ++++++++++++++++++++++++++++++
 2 files changed, 68 insertions(+)

diff --git a/message/tests/api_test.php b/message/tests/api_test.php
index db5bf421a60..3e10f614a8b 100644
--- a/message/tests/api_test.php
+++ b/message/tests/api_test.php
@@ -3672,6 +3672,37 @@ class core_message_api_testcase extends core_message_messagelib_testcase {
         $this->assertObjectHasAttribute('iscontact', $request);
     }
 
+    /**
+     * Test getting contact requests when there are none.
+     */
+    public function test_get_contact_requests_no_requests() {
+        $this->resetAfterTest();
+
+        $user1 = self::getDataGenerator()->create_user();
+
+        $requests = \core_message\api::get_contact_requests($user1->id);
+
+        $this->assertEmpty($requests);
+    }
+
+    /**
+     * Test getting contact requests with limits.
+     */
+    public function test_get_contact_requests_with_limits() {
+        $this->resetAfterTest();
+
+        $user1 = self::getDataGenerator()->create_user();
+        $user2 = self::getDataGenerator()->create_user();
+        $user3 = self::getDataGenerator()->create_user();
+
+        \core_message\api::create_contact_request($user2->id, $user1->id);
+        \core_message\api::create_contact_request($user3->id, $user1->id);
+
+        $requests = \core_message\api::get_contact_requests($user1->id, 0, 1);
+
+        $this->assertCount(1, $requests);
+    }
+
     /**
      * Test adding contacts.
      */
diff --git a/message/tests/externallib_test.php b/message/tests/externallib_test.php
index 0dc3e3c56d7..6ca9e11a81a 100644
--- a/message/tests/externallib_test.php
+++ b/message/tests/externallib_test.php
@@ -583,6 +583,43 @@ class core_message_externallib_testcase extends externallib_advanced_testcase {
         $this->assertArrayHasKey('iscontact', $request);
     }
 
+    /**
+     * Test getting contact requests when there are none.
+     */
+    public function test_get_contact_requests_no_requests() {
+        $this->resetAfterTest();
+
+        $user1 = self::getDataGenerator()->create_user();
+
+        $this->setUser($user1);
+
+        $requests = core_message_external::get_contact_requests($user1->id);
+        $requests = external_api::clean_returnvalue(core_message_external::get_contact_requests_returns(), $requests);
+
+        $this->assertEmpty($requests);
+    }
+
+    /**
+     * Test getting contact requests with limits.
+     */
+    public function test_get_contact_requests_with_limits() {
+        $this->resetAfterTest();
+
+        $user1 = self::getDataGenerator()->create_user();
+        $user2 = self::getDataGenerator()->create_user();
+        $user3 = self::getDataGenerator()->create_user();
+
+        $this->setUser($user1);
+
+        \core_message\api::create_contact_request($user2->id, $user1->id);
+        \core_message\api::create_contact_request($user3->id, $user1->id);
+
+        $requests = core_message_external::get_contact_requests($user1->id, 0, 1);
+        $requests = external_api::clean_returnvalue(core_message_external::get_contact_requests_returns(), $requests);
+
+        $this->assertCount(1, $requests);
+    }
+
     /**
      * Test getting contact requests with messaging disabled.
      */