1 /**
2 * UGENE - Integrated Bioinformatics Tools.
3 * Copyright (C) 2008-2021 UniPro <ugene@unipro.ru>
4 * http://ugene.net
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License
8 * as published by the Free Software Foundation; either version 2
9 * of the License, or (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
19 * MA 02110-1301, USA.
20 */
21
22 #include <primitives/GTWidget.h>
23
24 #include <U2Core/AppContext.h>
25 #include <U2Core/Task.h>
26
27 #include "GTUtilsTask.h"
28 #include "GTUtilsTaskTreeView.h"
29
30 namespace U2 {
31
32 #define GT_CLASS_NAME "GTUtilsTask"
33
34 #define GT_METHOD_NAME "getTopLevelTasks"
getTopLevelTasks(HI::GUITestOpStatus & os)35 QList<Task *> GTUtilsTask::getTopLevelTasks(HI::GUITestOpStatus &os) {
36 Q_UNUSED(os);
37 TaskScheduler *scheduller = AppContext::getTaskScheduler();
38 GT_CHECK_RESULT(scheduller != nullptr, "task scheduler is NULL", QList<Task *>());
39 return scheduller->getTopLevelTasks();
40 }
41 #undef GT_METHOD_NAME
42
43 #define GT_METHOD_NAME "getSubTaskByName"
getSubTaskByName(HI::GUITestOpStatus & os,QString taskName,GTGlobals::FindOptions options)44 Task *GTUtilsTask::getSubTaskByName(HI::GUITestOpStatus &os, QString taskName, GTGlobals::FindOptions options) {
45 Task *result = nullptr;
46 QList<Task *> tasks = getTopLevelTasks(os);
47 foreach (Task *t, tasks) {
48 result = getSubTaskByName(os, t, taskName, false);
49 }
50 if (options.failIfNotFound) {
51 GT_CHECK_RESULT(result != nullptr, "no subtask with name " + taskName, nullptr);
52 }
53 return result;
54 }
55 #undef GT_METHOD_NAME
56
57 #define GT_METHOD_NAME "getSubTaskByName"
getSubTaskByName(HI::GUITestOpStatus & os,Task * parent,QString taskName,GTGlobals::FindOptions options)58 Task *GTUtilsTask::getSubTaskByName(HI::GUITestOpStatus &os, Task *parent, QString taskName, GTGlobals::FindOptions options) {
59 Task *result = nullptr;
60 foreach (const QPointer<Task> &t, parent->getSubtasks()) {
61 if (t->getTaskName() == taskName) {
62 return t.data();
63 } else {
64 result = getSubTaskByName(os, t.data(), taskName, false);
65 }
66 }
67
68 if (options.failIfNotFound) {
69 GT_CHECK_RESULT(result != nullptr, "no subtask with name " + taskName, nullptr);
70 }
71 return result;
72 }
73 #undef GT_METHOD_NAME
74
75 #define GT_METHOD_NAME "getTaskByName"
getTaskByName(HI::GUITestOpStatus & os,QString taskName,GTGlobals::FindOptions options)76 Task *GTUtilsTask::getTaskByName(HI::GUITestOpStatus &os, QString taskName, GTGlobals::FindOptions options) {
77 QList<Task *> tasks = getTopLevelTasks(os);
78 foreach (Task *t, tasks) {
79 QString name = t->getTaskName();
80 if (name == taskName) {
81 return t;
82 }
83 }
84 if (options.failIfNotFound == true) {
85 GT_CHECK_RESULT(false, "task " + taskName + " not found", nullptr);
86 }
87 return nullptr;
88 }
89 #undef GT_METHOD_NAME
90
91 #define GT_METHOD_NAME "checkTask"
checkTask(HI::GUITestOpStatus & os,QString taskName)92 void GTUtilsTask::checkTask(HI::GUITestOpStatus &os, QString taskName) {
93 getTaskByName(os, taskName);
94 }
95 #undef GT_METHOD_NAME
96
97 #define GT_METHOD_NAME "checkNoTask"
checkNoTask(HI::GUITestOpStatus & os,QString taskName)98 void GTUtilsTask::checkNoTask(HI::GUITestOpStatus &os, QString taskName) {
99 Task *t = getTaskByName(os, taskName, GTGlobals::FindOptions(false));
100 GT_CHECK(t == nullptr, "tast " + taskName + " unexpectidly found");
101 }
102 #undef GT_METHOD_NAME
103
104 #define GT_METHOD_NAME "cancelTask"
cancelTask(HI::GUITestOpStatus & os,QString taskName)105 void GTUtilsTask::cancelTask(HI::GUITestOpStatus &os, QString taskName) {
106 Task *t = getTaskByName(os, taskName);
107 t->cancel();
108 }
109 #undef GT_METHOD_NAME
110
111 #define GT_METHOD_NAME "cancelSubTask"
cancelSubTask(HI::GUITestOpStatus & os,QString taskName)112 void GTUtilsTask::cancelSubTask(HI::GUITestOpStatus &os, QString taskName) {
113 Task *t = getSubTaskByName(os, taskName);
114 t->cancel();
115 }
116 #undef GT_METHOD_NAME
117
118 #define GT_METHOD_NAME "waitTaskStart"
waitTaskStart(HI::GUITestOpStatus & os,QString taskName,int timeOut)119 void GTUtilsTask::waitTaskStart(HI::GUITestOpStatus &os, QString taskName, int timeOut) {
120 int i = 0;
121 while (getTaskByName(os, taskName, GTGlobals::FindOptions(false)) == nullptr) {
122 GTGlobals::sleep(100);
123 i++;
124 if (i > (timeOut / 100)) {
125 GT_CHECK(false, "task " + taskName + " not launched");
126 break;
127 }
128 }
129 }
130 #undef GT_METHOD_NAME
131
132 #undef GT_CLASS_NAME
133 } // namespace U2
134