67 lines
2.1 KiB
C++
67 lines
2.1 KiB
C++
#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);
|
|
}
|
|
};
|