From 63cf67a2ec8b23ea7e8b3bf35f646be7b1480273 Mon Sep 17 00:00:00 2001
From: Marc Morera <yuhu@mmoreram.com>
Date: Thu, 13 Apr 2023 11:28:26 +0200
Subject: [PATCH] Addd setSubresults method in Result

---
 Result/Result.php           | 10 ++++++++++
 Tests/Result/ResultTest.php | 13 +++++++++++++
 2 files changed, 23 insertions(+)

diff --git a/Result/Result.php b/Result/Result.php
index 24da9e7..7dbf433 100644
--- a/Result/Result.php
+++ b/Result/Result.php
@@ -402,6 +402,16 @@ public function getSubresults(): array
         return $this->subresults;
     }
 
+    /**
+     * @param array $subResults
+     *
+     * @return array
+     */
+    public function setSubResults(array $subResults)
+    {
+        $this->subresults = $subResults;
+    }
+
     /**
      * Set metadata.
      *
diff --git a/Tests/Result/ResultTest.php b/Tests/Result/ResultTest.php
index 4473f4d..2b48447 100644
--- a/Tests/Result/ResultTest.php
+++ b/Tests/Result/ResultTest.php
@@ -246,6 +246,19 @@ public function testMultiResult()
         $this->assertEquals(3, $subqueries['res1']->getTotalHits());
         $this->assertEquals(4, $subqueries['res2']->getTotalHits());
         $this->assertEquals(5, $subqueries['res3']->getTotalHits());
+
+        $result = Result::createFromArray([]);
+        $result->setSubResults([
+            'res1' => Result::create(Query::createMatchAll()->identifyWith('1'), 10, 3, null, [], []),
+            'res2' => Result::create(Query::createMatchAll()->identifyWith('2'), 10, 4, null, [], []),
+            'res3' => Result::create(Query::createMatchAll()->identifyWith('3'), 10, 5, null, [], []),
+        ]);
+
+        $this->assertCount(3, $result->getSubresults());
+        $subqueries = HttpHelper::emulateHttpTransport($result)->getSubresults();
+        $this->assertEquals(3, $subqueries['res1']->getTotalHits());
+        $this->assertEquals(4, $subqueries['res2']->getTotalHits());
+        $this->assertEquals(5, $subqueries['res3']->getTotalHits());
     }
 
     /**