diff --git a/src/libdanbooru/tests/test_service.h b/src/libdanbooru/tests/test_service.h
index 53a3e4e..60dd692 100644
--- a/src/libdanbooru/tests/test_service.h
+++ b/src/libdanbooru/tests/test_service.h
@@ -17,41 +17,23 @@
* along with Danbooru Client. If not, see .
*/
-#ifndef DANBOORU_TEST_CONNECT_H_
-#define DANBOORU_TEST_CONNECT_H_
+#ifndef DANBOORU_TEST_SERVICE_H_
+#define DANBOORU_TEST_SERVICE_H_
-#include "../konachan.h"
+#include "../moebooru.h"
+#include "../danbooru_donmai.h"
#include "../servicebase.h"
#include "../danboorupost.h"
#include "../danboorupool.h"
#include
-/*
- * Copyright 2015 Luca Beltrame
- *
- * This file is part of Danbooru Client.
- *
- * Danbooru Client is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * Danbooru Client is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with Danbooru Client. If not, see .
- */
-
#include
class QLabel;
namespace Danbooru {
-class KonachanServiceTest: public QObject
+class DanbooruServiceTest: public QObject
{
Q_OBJECT
@@ -59,8 +41,8 @@ private:
DanbooruServiceBase* m_service;
QLabel* m_label;
public:
- KonachanServiceTest(QUrl boardUrl);
- ~KonachanServiceTest();
+ DanbooruServiceTest(QUrl boardUrl, Danbooru::ApiType type = Danbooru::MoeBooru);
+ ~DanbooruServiceTest();
void testPostService();
void testPoolService();
@@ -78,4 +60,4 @@ private Q_SLOTS:
-#endif
\ No newline at end of file
+#endif