1 files changed, 32 insertions, 4 deletions

diff --git a/sources/shiboken6/ApiExtractor/tests/testmodifydocumentation.cpp b/sources/shiboken6/ApiExtractor/tests/testmodifydocumentation.cpp
index 05fd3bf6b..9cf2e0cc7 100644
--- a/sources/shiboken6/ApiExtractor/tests/testmodifydocumentation.cpp
+++ b/sources/shiboken6/ApiExtractor/tests/testmodifydocumentation.cpp
@@ -4,6 +4,7 @@
 #include "testmodifydocumentation.h"
 #include "testutil.h"
 #include <abstractmetalang.h>
+#include <abstractmetafunction.h>
 #include <documentation.h>
 #include <modifications.h>
 #include <complextypeentry.h>
@@ -19,7 +20,7 @@ using namespace Qt::StringLiterals;
 
 void TestModifyDocumentation::testModifyDocumentation()
 {
-    const char* cppCode ="struct B { void b(); }; class A {};\n";
+    const char cppCode[] = "struct B { void b(); }; class A {};\n";
     const char xmlCode[] =
 R"(<typesystem package="Foo">
     <value-type name='B'>
@@ -32,8 +33,8 @@ R"(<typesystem package="Foo">
 </typesystem>
 )";
     QScopedPointer<AbstractMetaBuilder> builder(TestUtil::parse(cppCode, xmlCode));
-    QVERIFY(!builder.isNull());
-    AbstractMetaClass *classA = AbstractMetaClass::findClass(builder->classes(), u"A");
+    QVERIFY(builder);
+    const auto classA = AbstractMetaClass::findClass(builder->classes(), "A");
     QVERIFY(classA);
     DocModificationList docMods = classA->typeEntry()->docModifications();
     QCOMPARE(docMods.size(), 2);
@@ -46,7 +47,7 @@ R"(<typesystem package="Foo">
     // cannot handle Qt resources.
     QTemporaryDir tempDir(QDir::tempPath() + u"/shiboken_testmodifydocXXXXXX"_s);
     QVERIFY2(tempDir.isValid(), qPrintable(tempDir.errorString()));
-    const QString docFileName = u"a.xml"_s;
+    constexpr auto docFileName = "a.xml"_L1;
     QVERIFY(QFile::copy(u":/"_s + docFileName, tempDir.filePath(docFileName)));
 
     QtDocParser docParser;
@@ -78,6 +79,33 @@ R"(<?xml version="1.0"?>
     QCOMPARE(actualDocSimplified, expectedDocSimplified);
 }
 
+void TestModifyDocumentation::testInjectAddedFunctionDocumentation()
+{
+    const char cppCode[] ="class A {};\n";
+    const char xmlCode[] = R"XML(
+<typesystem package="Foo">
+    <value-type name='A'>
+        <add-function signature="foo(int@parameter_name@)">
+              <inject-documentation format="target" mode="append">
+                  Injected documentation of added function foo.
+              </inject-documentation>
+        </add-function>
+    </value-type>
+</typesystem>
+)XML";
+    QScopedPointer<AbstractMetaBuilder> builder(TestUtil::parse(cppCode, xmlCode));
+    QVERIFY(builder);
+    const auto classA = AbstractMetaClass::findClass(builder->classes(), "A");
+    QVERIFY(classA);
+    const auto f = classA->findFunction("foo");
+    QVERIFY(f);
+    QVERIFY(f->isUserAdded());
+    auto docMods = f->addedFunctionDocModifications();
+    QCOMPARE(docMods.size(), 1);
+    const QString code = docMods.constFirst().code();
+    QVERIFY(code.contains(u"Injected documentation of added function foo."));
+}
+
 // We expand QTEST_MAIN macro but using QCoreApplication instead of QApplication
 // because this test needs an event loop but can't use QApplication to avoid a crash
 // on our ARMEL/FRAMANTLE buildbot