1 files changed, 14 insertions, 2 deletions
diff --git a/tests/baseline/shared/qbaselinetest.cpp b/tests/baseline/shared/qbaselinetest.cpp
index 004d8013f1c..493be4bd7a4 100644
--- a/tests/baseline/shared/qbaselinetest.cpp
+++ b/tests/baseline/shared/qbaselinetest.cpp
@@ -238,8 +238,10 @@ bool connectToBaselineServer(QByteArray *msg, const QString &testProject, const
     bool dummy;
     QByteArray dummyMsg;
 
-    definedTestProject = testProject;
-    definedTestCase = testCase;
+    if (!testProject.isEmpty())
+        definedTestProject = testProject;
+    if (!testCase.isEmpty())
+        definedTestCase = testCase;
 
     return connect(msg ? msg : &dummyMsg, &dummy);
 }
@@ -255,6 +257,16 @@ void setSimFail(bool fail)
     simfail = fail;
 }
 
+void setProject(const QString &projectName)
+{
+    definedTestProject = projectName;
+}
+
+void setProjectImageKeys(const QStringList &keys)
+{
+    QString keyList = keys.join(QLC(','));
+    addClientProperty(PI_ProjectImageKeys, keyList);
+}
 
 void modifyImage(QImage *img)
 {