1
0
mirror of https://github.com/clementine-player/Clementine synced 2024-12-25 17:12:35 +01:00
Clementine-audio-player-Mac.../tests/mock_sqldriver.h

84 lines
2.8 KiB
C++

/* This file is part of Clementine.
Copyright 2010, David Sansome <me@davidsansome.com>
Clementine 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.
Clementine 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 Clementine. If not, see <http://www.gnu.org/licenses/>.
*/
#include <gmock/gmock.h>
#include <QSqlDriver>
#include <QSqlResult>
#include <QSqlIndex>
#include <QSqlField>
class MockSqlDriver : public QSqlDriver {
public:
virtual ~MockSqlDriver() {}
MOCK_METHOD0(beginTransaction, bool());
MOCK_METHOD0(close, void());
MOCK_METHOD0(commitTransaction, bool());
MOCK_CONST_METHOD0(createResult, QSqlResult*());
MOCK_CONST_METHOD2(escapeIdentifier, QString(const QString&, IdentifierType));
MOCK_CONST_METHOD2(formatValue, QString(const QSqlField&, bool));
MOCK_CONST_METHOD0(handle, QVariant());
MOCK_CONST_METHOD1(hasFeature, bool(DriverFeature));
MOCK_CONST_METHOD0(isOpen, bool());
MOCK_METHOD6(open, bool(const QString&, const QString&, const QString&, const QString&, int, const QString&));
MOCK_CONST_METHOD1(primaryIndex, QSqlIndex(const QString&));
MOCK_CONST_METHOD1(record, QSqlRecord(const QString&));
MOCK_METHOD0(rollbackTransaction, bool());
MOCK_CONST_METHOD4(sqlStatement, QString(StatementType, const QString&, const QSqlRecord&, bool));
MOCK_CONST_METHOD1(tables, QStringList(QSql::TableType));
MOCK_METHOD1(setLastError, void(const QSqlError&));
MOCK_METHOD1(setOpen, void(bool));
MOCK_METHOD1(setOpenError, void(bool));
};
class MockSqlResult : public QSqlResult {
public:
MockSqlResult(QSqlDriver* driver)
: QSqlResult(driver) {
}
virtual ~MockSqlResult() {}
MOCK_METHOD3(bindValue, void(int, const QVariant&, QSql::ParamType));
MOCK_METHOD3(bindValue, void(const QString&, const QVariant&, QSql::ParamType));
MOCK_METHOD1(data, QVariant(int));
MOCK_METHOD0(exec, bool());
MOCK_METHOD1(fetch, bool(int));
MOCK_METHOD0(fetchFirst, bool());
MOCK_METHOD0(fetchLast, bool());
MOCK_METHOD1(isNull, bool(int));
MOCK_CONST_METHOD0(lastInsertId, QVariant());
MOCK_METHOD0(numRowsAffected, int());
MOCK_METHOD1(prepare, bool(const QString&));
MOCK_METHOD1(reset, bool(const QString&));
MOCK_METHOD0(size, int());
void hackSetActive(const QString&) {
setActive(true);
setAt(0);
setSelect(true);
}
void hackSetActiveVoid() {
hackSetActive(QString());
}
void setAt(int i) {
QSqlResult::setAt(i);
}
};