From f667781c9cdd0acbcc7704363718fce265afe2f1 Mon Sep 17 00:00:00 2001
From: silverqx <silver.zachara@gmail.com>
Date: Mon, 8 Aug 2022 16:29:58 +0200
Subject: [PATCH] tests added chunkMap test for btm on Relation &

---
 .../model_relations/tst_model_relations.cpp   | 30 +++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/tests/auto/functional/orm/tiny/model_relations/tst_model_relations.cpp b/tests/auto/functional/orm/tiny/model_relations/tst_model_relations.cpp
index 623209234..ef0405e17 100644
--- a/tests/auto/functional/orm/tiny/model_relations/tst_model_relations.cpp
+++ b/tests/auto/functional/orm/tiny/model_relations/tst_model_relations.cpp
@@ -191,6 +191,7 @@ private slots:
     void chunkMap_EmptyResult() const;
 
     void chunkMap_TemplatedReturnValue() const;
+    void chunkMap_TemplatedReturnValue_OnRelationRef() const;
     void chunkMap_EnforceOrderBy_TemplatedReturnValue() const;
     void chunkMap_EmptyResult_TemplatedReturnValue() const;
 
@@ -3246,6 +3247,35 @@ void tst_Model_Relations::chunkMap_TemplatedReturnValue() const
     QCOMPARE(result, expectedResult);
 }
 
+void tst_Model_Relations::chunkMap_TemplatedReturnValue_OnRelationRef() const
+{
+    QFETCH_GLOBAL(QString, connection);
+
+    ConnectionOverride::connection = connection;
+
+    /* Even if the chunkMap<> is called on the Relation & it can't crash or fail, it
+       should normally works but the pivot table will not be hydrated. */
+    auto result = Torrent::find(2)->tags()->orderBy(ID)
+                  .chunkMap<QString>([](Tag &&model)
+    {
+        // Pivot table is not hydrated
+        Q_ASSERT(!model.relationLoaded("tagged"));
+
+        // Return the modify name directly
+        return QStringLiteral("%1_mapped").arg(model[NAME]->template value<QString>());
+    });
+
+    QVector<QString> expectedResult {
+        {"tag1_mapped"},
+        {"tag2_mapped"},
+        {"tag3_mapped"},
+        {"tag4_mapped"},
+    };
+
+    QVERIFY(expectedResult.size() == result.size());
+    QCOMPARE(result, expectedResult);
+}
+
 void
 tst_Model_Relations::chunkMap_EnforceOrderBy_TemplatedReturnValue() const
 {