131 lines
3.7 KiB
C++
131 lines
3.7 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 "mock_networkaccessmanager.h"
|
|
|
|
#include <QtDebug>
|
|
#include <QUrlQuery>
|
|
|
|
#include <algorithm>
|
|
using std::min;
|
|
|
|
using ::testing::MakeMatcher;
|
|
using ::testing::Matcher;
|
|
using ::testing::MatcherInterface;
|
|
using ::testing::MatchResultListener;
|
|
using ::testing::Return;
|
|
|
|
class RequestForUrlMatcher : public MatcherInterface<const QNetworkRequest&> {
|
|
public:
|
|
RequestForUrlMatcher(const QString& contains,
|
|
const QMap<QString, QString>& expected_params)
|
|
: contains_(contains),
|
|
expected_params_(expected_params) {
|
|
}
|
|
virtual ~RequestForUrlMatcher() {}
|
|
|
|
virtual bool Matches(const QNetworkRequest& req) const {
|
|
const QUrl& url = req.url();
|
|
|
|
if (!url.toString().contains(contains_)) {
|
|
return false;
|
|
}
|
|
|
|
QUrlQuery url_query(url);
|
|
for (QMap<QString, QString>::const_iterator it = expected_params_.constBegin();
|
|
it != expected_params_.constEnd(); ++it) {
|
|
if (!url_query.hasQueryItem(it.key()) ||
|
|
url_query.queryItemValue(it.key()) != it.value()) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
virtual bool MatchAndExplain(const QNetworkRequest& req, MatchResultListener* listener) const {
|
|
*listener << "which is " << req.url().toString().toUtf8().constData();
|
|
return Matches(req);
|
|
}
|
|
|
|
virtual void DescribeTo(::std::ostream* os) const {
|
|
*os << "matches url";
|
|
}
|
|
|
|
private:
|
|
QString contains_;
|
|
QMap<QString, QString> expected_params_;
|
|
};
|
|
|
|
inline Matcher<const QNetworkRequest&> RequestForUrl(
|
|
const QString& contains,
|
|
const QMap<QString, QString>& params) {
|
|
return MakeMatcher(new RequestForUrlMatcher(contains, params));
|
|
}
|
|
|
|
MockNetworkReply* MockNetworkAccessManager::ExpectGet(
|
|
const QString& contains,
|
|
const QMap<QString, QString>& expected_params,
|
|
int status,
|
|
const QByteArray& data) {
|
|
MockNetworkReply* reply = new MockNetworkReply(data);
|
|
reply->setAttribute(QNetworkRequest::HttpStatusCodeAttribute, status);
|
|
|
|
EXPECT_CALL(*this, createRequest(
|
|
GetOperation, RequestForUrl(contains, expected_params), nullptr)).
|
|
WillOnce(Return(reply));
|
|
|
|
return reply;
|
|
}
|
|
|
|
MockNetworkReply::MockNetworkReply()
|
|
: data_(nullptr) {
|
|
}
|
|
|
|
MockNetworkReply::MockNetworkReply(const QByteArray& data)
|
|
: data_(data),
|
|
pos_(0) {
|
|
}
|
|
|
|
void MockNetworkReply::SetData(const QByteArray& data) {
|
|
data_ = data;
|
|
pos_ = 0;
|
|
}
|
|
|
|
qint64 MockNetworkReply::readData(char* data, qint64 size) {
|
|
if (data_.size() == pos_) {
|
|
return -1;
|
|
}
|
|
qint64 bytes_to_read = min(data_.size() - pos_, size);
|
|
memcpy(data, data_.constData() + pos_, bytes_to_read);
|
|
pos_ += bytes_to_read;
|
|
return bytes_to_read;
|
|
}
|
|
|
|
qint64 MockNetworkReply::writeData(const char* data, qint64) {
|
|
ADD_FAILURE() << "Something tried to write to a QNetworkReply";
|
|
return -1;
|
|
}
|
|
|
|
void MockNetworkReply::Done() {
|
|
setOpenMode(QIODevice::ReadOnly);
|
|
emit finished();
|
|
}
|
|
|
|
void MockNetworkReply::setAttribute(QNetworkRequest::Attribute code, const QVariant& value) {
|
|
QNetworkReply::setAttribute(code, value);
|
|
}
|