84 lines
2.8 KiB
C++
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);
|
|
}
|
|
};
|