qiurui
/
qtgrpc
mirror of https://github.com/qt/qtgrpc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
qtgrpc/tests/auto/grpc/client/tst_grpc_client.cpp

703 lines
24 KiB
C++
Raw Blame History

// Copyright (C) 2022 The Qt Company Ltd.
// Copyright (C) 2019 Alexey Edelev <semlanik@gmail.com>
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only
#include <QGrpcCallReply>
#if QT_CONFIG(native_grpc)
# include <QGrpcChannel>
#endif
#include <QGrpcCredentials>
#include <QGrpcHttp2Channel>
#include <QGrpcInsecureCallCredentials>
#include <QGrpcInsecureChannelCredentials>
#include <QCoreApplication>
#include <QCryptographicHash>
#include <QFile>
#include <QProcess>
#include <QSignalSpy>
#include <QTest>
#include <QtCore/QThread>
#include <QtCore/QTimer>
#if QT_CONFIG(native_grpc)
# include <grpcpp/security/credentials.h>
#endif
#include "testservice_client.grpc.qpb.h"
namespace {
constexpr int MessageLatency = QT_GRPC_TEST_MESSAGE_LATENCY;
constexpr int MessageLatencyThreshold = MessageLatency / 5;
constexpr int MessageLatencyWithThreshold = MessageLatency + MessageLatencyThreshold;
constexpr int FailTimeout = QT_GRPC_TEST_MESSAGE_LATENCY * 5;
using namespace qtgrpc::tests;
static QSharedPointer<TestService::Client> createHttp2Client()
{
auto client = QSharedPointer<TestService::Client>::create();
client->attachChannel(QSharedPointer<QGrpcHttp2Channel>::create(
QUrl("http://localhost:50051", QUrl::StrictMode),
QGrpcInsecureChannelCredentials() | QGrpcInsecureCallCredentials()));
return client;
}
#if QT_CONFIG(native_grpc)
# ifndef Q_OS_WINDOWS
static QSharedPointer<TestService::Client> createGrpcSocketClient()
{
auto client = QSharedPointer<TestService::Client>::create();
client->attachChannel(QSharedPointer<QGrpcChannel>::create(QString("unix:///tmp/test.sock"),
QGrpcChannel::InsecureChannelCredentials));
return client;
}
# endif
static QSharedPointer<TestService::Client> createGrpcHttpClient()
{
auto client = QSharedPointer<TestService::Client>::create();
client->attachChannel(QSharedPointer<QGrpcChannel>::create(QString("localhost:50051"),
QGrpcChannel::InsecureChannelCredentials));
return client;
}
#endif
} // namespace
class QtGrpcClientTest : public QObject
{
Q_OBJECT
public:
private slots:
void initTestCase_data()
{
QTest::addColumn<QSharedPointer<TestService::Client>>("client");
QTest::newRow("Http2Client") << createHttp2Client();
#if QT_CONFIG(native_grpc)
# ifndef Q_OS_WINDOWS
QTest::newRow("GrpcSocket") << createGrpcSocketClient();
# endif
QTest::newRow("GrpcHttp") << createGrpcHttpClient();
#endif
}
void initTestCase()
{
startServer();
qRegisterProtobufTypes();
}
void cleanupTestCase()
{
if (serverProc && serverProc->state() == QProcess::ProcessState::Running) {
serverProc->kill();
serverProc->waitForFinished();
}
}
void init()
{
QFETCH_GLOBAL(QSharedPointer<TestService::Client>, client);
_client = std::move(client);
if (serverProc->state() != QProcess::ProcessState::Running) {
qInfo() << "Restarting server";
startServer();
QVERIFY2(serverProc->state() == QProcess::ProcessState::Running,
"Precondition failed - Server cannot be started.");
}
}
void CallAndRecieveStringTest();
void CallAndAsyncRecieveWithSubscribeStringTest();
void CallAndAsyncRecieveWithLambdaStringTest();
void CallAndAsyncImmediateAbortStringTest();
void CallAndAsyncDeferredAbortStringTest();
void StreamStringTest();
void StreamStringAndAbortTest();
void StreamStringAndDeferredAbortTest();
// void StreamStringAndGetChangedSignalsTests();
void StreamBlobTest();
void CallMessageAsyncTest();
void CallMessageClientAsyncTest();
void CallMessageClientTest();
void StatusMessageClientTest();
void StreamAndGetAsyncReplyTest();
void MultipleStreamsTest();
void MultipleStreamsCancelTest();
void CallNonCompatibleArgRetTest();
void CallStringThreadTest();
void StringAsyncThreadTest();
void StreamStringThreadTest();
void StreamCancelWhileErrorTimeoutTest();
private:
void startServer()
{
serverProc = std::make_unique<QProcess>();
QString serverPath = QFINDTESTDATA("../shared/test_server/testserver");
QVERIFY2(!serverPath.isEmpty(), "testserver binary is missing");
serverProc->start(serverPath);
serverProc->waitForStarted();
// Extra time for the server to setup
QTest::qSleep(1000);
}
std::unique_ptr<QProcess> serverProc;
QSharedPointer<TestService::Client> _client;
};
void QtGrpcClientTest::CallAndRecieveStringTest()
{
SimpleStringMessage request;
auto result = QSharedPointer<SimpleStringMessage>::create();
request.setTestFieldString("Hello Qt!");
QCOMPARE_EQ(_client->testMethod(request, result.get()), QGrpcStatus::Ok);
QCOMPARE_EQ(result->testFieldString(), "Hello Qt!");
}
void QtGrpcClientTest::CallAndAsyncRecieveWithSubscribeStringTest()
{
SimpleStringMessage request;
SimpleStringMessage result;
request.setTestFieldString("Hello Qt!");
QEventLoop waiter;
QSharedPointer<QGrpcCallReply> reply = _client->testMethod(request);
reply->subscribe(this, [reply, &result, &waiter]() {
result = reply->read<SimpleStringMessage>();
reply->deleteLater();
waiter.quit();
});
waiter.exec();
QCOMPARE_EQ(result.testFieldString(), "Hello Qt!");
}
void QtGrpcClientTest::CallAndAsyncRecieveWithLambdaStringTest()
{
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Hello Qt!");
QEventLoop waiter;
_client->testMethod(request, this, [&result, &waiter](QSharedPointer<QGrpcCallReply> reply) {
result = reply->read<SimpleStringMessage>();
waiter.quit();
});
waiter.exec();
QCOMPARE_EQ(result.testFieldString(), "Hello Qt!");
}
void QtGrpcClientTest::CallAndAsyncImmediateAbortStringTest()
{
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("sleep");
QSharedPointer<QGrpcCallReply> reply = _client->testMethod(request);
result.setTestFieldString("Result not changed by echo");
QObject::connect(reply.get(), &QGrpcCallReply::finished, this, [&result, reply]() {
result = reply->read<SimpleStringMessage>();
reply->deleteLater();
});
QSignalSpy replyErrorSpy(reply.get(), &QGrpcCallReply::errorOccurred);
QSignalSpy clientErrorSpy(_client.get(), &TestService::Client::errorOccurred);
QVERIFY(replyErrorSpy.isValid());
QVERIFY(clientErrorSpy.isValid());
reply->abort();
QTRY_COMPARE_EQ_WITH_TIMEOUT(replyErrorSpy.count(), 1, FailTimeout);
QTRY_COMPARE_EQ_WITH_TIMEOUT(clientErrorSpy.count(), 1, FailTimeout);
QCOMPARE(qvariant_cast<QGrpcStatus>(replyErrorSpy.at(0).first()).code(),
QGrpcStatus::StatusCode::Aborted);
QCOMPARE(qvariant_cast<QGrpcStatus>(clientErrorSpy.at(0).first()).code(),
QGrpcStatus::StatusCode::Aborted);
QCOMPARE_EQ(result.testFieldString(), "Result not changed by echo");
}
void QtGrpcClientTest::CallAndAsyncDeferredAbortStringTest()
{
SimpleStringMessage request;
request.setTestFieldString("sleep");
SimpleStringMessage result;
result.setTestFieldString("Result not changed by echo");
QSharedPointer<QGrpcCallReply> reply = _client->testMethod(request);
QObject::connect(reply.get(), &QGrpcCallReply::finished, this,
[reply, &result]() { result = reply->read<SimpleStringMessage>(); });
QSignalSpy replyErrorSpy(reply.get(), &QGrpcCallReply::errorOccurred);
QVERIFY(replyErrorSpy.isValid());
QTimer::singleShot(MessageLatencyThreshold, reply.get(), &QGrpcCallReply::abort);
QTRY_COMPARE_EQ_WITH_TIMEOUT(replyErrorSpy.count(), 1, FailTimeout);
QCOMPARE_EQ(result.testFieldString(), "Result not changed by echo");
}
void QtGrpcClientTest::StreamStringTest()
{
const int ExpectedMessageCount = 4;
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Stream");
auto stream = _client->streamTestMethodServerStream(request);
QSignalSpy messageReceivedSpy(stream.get(), &QGrpcStream::messageReceived);
QVERIFY(messageReceivedSpy.isValid());
QSignalSpy streamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamErrorSpy.isValid());
QSignalSpy streamFinishedSpy(stream.get(), &QGrpcStream::finished);
QObject::connect(stream.get(), &QGrpcStream::messageReceived, this, [&result, stream]() {
SimpleStringMessage ret = stream->read<SimpleStringMessage>();
result.setTestFieldString(result.testFieldString() + ret.testFieldString());
});
QTRY_COMPARE_EQ_WITH_TIMEOUT(streamFinishedSpy.count(), 1,
MessageLatencyWithThreshold * ExpectedMessageCount);
QCOMPARE(streamErrorSpy.count(), 0);
QCOMPARE(messageReceivedSpy.count(), ExpectedMessageCount);
QCOMPARE_EQ(result.testFieldString(), "Stream1Stream2Stream3Stream4");
}
void QtGrpcClientTest::StreamStringAndAbortTest()
{
const int ExpectedMessageCount = 2;
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Stream");
auto stream = _client->streamTestMethodServerStream(request);
QSignalSpy streamFinishedSpy(stream.get(), &QGrpcStream::finished);
QVERIFY(streamFinishedSpy.isValid());
QSignalSpy streamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamErrorSpy.isValid());
int i = 0;
QObject::connect(stream.get(), &QGrpcStream::messageReceived, this, [&result, &i, stream]() {
SimpleStringMessage ret = stream->read<SimpleStringMessage>();
result.setTestFieldString(result.testFieldString() + ret.testFieldString());
if (++i == ExpectedMessageCount)
stream->abort();
});
QTRY_COMPARE_EQ_WITH_TIMEOUT(streamFinishedSpy.count(), 1,
MessageLatencyWithThreshold * ExpectedMessageCount);
QCOMPARE(streamErrorSpy.count(), 0);
QCOMPARE(i, 2);
QCOMPARE_EQ(result.testFieldString(), "Stream1Stream2");
}
void QtGrpcClientTest::StreamStringAndDeferredAbortTest()
{
const int ExpectedMessageCount = 3;
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Stream");
auto stream = _client->streamTestMethodServerStream(request);
QSignalSpy streamFinishedSpy(stream.get(), &QGrpcStream::finished);
QVERIFY(streamFinishedSpy.isValid());
QSignalSpy streamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamErrorSpy.isValid());
QSignalSpy messageReceivedSpy(stream.get(), &QGrpcStream::messageReceived);
QVERIFY(messageReceivedSpy.isValid());
int i = 0;
QObject::connect(stream.get(), &QGrpcStream::messageReceived, this, [&result, stream, &i]() {
SimpleStringMessage ret = stream->read<SimpleStringMessage>();
result.setTestFieldString(result.testFieldString() + ret.testFieldString());
if (++i == ExpectedMessageCount)
QTimer::singleShot(MessageLatencyThreshold, stream.get(), &QGrpcStream::abort);
});
QTRY_COMPARE_EQ_WITH_TIMEOUT(streamFinishedSpy.count(), 1,
MessageLatencyWithThreshold * ExpectedMessageCount);
QCOMPARE(streamErrorSpy.count(), 0);
QCOMPARE(messageReceivedSpy.count(), ExpectedMessageCount);
QCOMPARE_EQ(result.testFieldString(), "Stream1Stream2Stream3");
}
// TODO: Protobuf property doesn't have changed signal
/*
void QtGrpcClientTest::StreamStringAndGetChangedSignalsTests
{
SimpleStringMessage request;
request.setTestFieldString("Stream");
auto result = QSharedPointer<SimpleStringMessage>(new SimpleStringMessage);
QSignalSpy testFieldChangedSpy(result, &SimpleStringMessage::testFieldStringChanged);
QVERIFY(testFieldChangedSpy.isValid());
_client->streamTestMethodServerStream(request, result);
QTRY_COMPARE(testFieldChangedSpy.count(), 4);
QCOMPARE_EQ(result->testFieldString(), "Stream4");
}
*/
void QtGrpcClientTest::StreamBlobTest()
{
BlobMessage result;
BlobMessage request;
QFile testFile(":/assets/testfile");
QVERIFY(testFile.open(QFile::ReadOnly));
request.setTestBytes(testFile.readAll());
QByteArray dataHash = QCryptographicHash::hash(request.testBytes(), QCryptographicHash::Sha256);
auto stream = _client->streamTestMethodBlobServerStream(request);
QSignalSpy streamFinishedSpy(stream.get(), &QGrpcStream::finished);
QVERIFY(streamFinishedSpy.isValid());
QSignalSpy streamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamErrorSpy.isValid());
QObject::connect(stream.get(), &QGrpcStream::messageReceived, this, [&result, stream]() {
BlobMessage ret = stream->read<BlobMessage>();
result.setTestBytes(ret.testBytes());
});
QTRY_COMPARE_EQ_WITH_TIMEOUT(streamFinishedSpy.count(), 1, MessageLatencyWithThreshold);
QCOMPARE_EQ(streamErrorSpy.count(), 0);
QVERIFY(!result.testBytes().isEmpty());
QByteArray returnDataHash = QCryptographicHash::hash(result.testBytes(),
QCryptographicHash::Sha256);
QCOMPARE_EQ(returnDataHash, dataHash);
}
void QtGrpcClientTest::CallMessageAsyncTest()
{
SimpleStringMessage request;
request.setTestFieldString("Some status message");
QSharedPointer<QGrpcCallReply> reply = _client->testMethodStatusMessage(request);
QSignalSpy replyErrorSpy(reply.get(), &QGrpcCallReply::errorOccurred);
QVERIFY(replyErrorSpy.isValid());
QTRY_COMPARE_EQ_WITH_TIMEOUT(replyErrorSpy.count(), 1, FailTimeout);
QCOMPARE(qvariant_cast<QGrpcStatus>(replyErrorSpy.at(0).first()).message(),
request.testFieldString());
}
void QtGrpcClientTest::CallMessageClientAsyncTest()
{
SimpleStringMessage request;
request.setTestFieldString("Some status message");
QSignalSpy clientErrorSpy(_client.get(), &TestService::Client::errorOccurred);
QVERIFY(clientErrorSpy.isValid());
_client->testMethodStatusMessage(request);
QTRY_COMPARE_EQ_WITH_TIMEOUT(clientErrorSpy.count(), 1, FailTimeout);
QCOMPARE(qvariant_cast<QGrpcStatus>(clientErrorSpy.at(0).first()).message(),
request.testFieldString());
}
void QtGrpcClientTest::CallMessageClientTest()
{
SimpleStringMessage request;
request.setTestFieldString("Some status message");
auto result = QSharedPointer<SimpleStringMessage>::create();
QSignalSpy clientErrorSpy(_client.get(), &TestService::Client::errorOccurred);
QVERIFY(clientErrorSpy.isValid());
_client->testMethodStatusMessage(request, result.get());
QTRY_COMPARE_GE_WITH_TIMEOUT(clientErrorSpy.count(), 1, FailTimeout);
QCOMPARE(qvariant_cast<QGrpcStatus>(clientErrorSpy.at(0).first()).message(),
request.testFieldString());
}
void QtGrpcClientTest::StatusMessageClientTest()
{
SimpleStringMessage request;
request.setTestFieldString("Some status message");
auto result = QSharedPointer<SimpleStringMessage>::create();
QGrpcStatus status = _client->testMethodStatusMessage(request, result.get());
QCOMPARE_EQ(status.message(), request.testFieldString());
}
void QtGrpcClientTest::StreamAndGetAsyncReplyTest()
{
SimpleStringMessage request;
request.setTestFieldString("Some status message");
QGrpcStatus::StatusCode asyncStatus = QGrpcStatus::StatusCode::Ok;
QString statusMessage;
auto reply = _client->testMethodStatusMessage(request);
reply->subscribe(
this, []() { QVERIFY(false); },
[&asyncStatus, &statusMessage](const QGrpcStatus &status) {
asyncStatus = status.code();
statusMessage = status.message();
});
QTRY_COMPARE_WITH_TIMEOUT(statusMessage, request.testFieldString(),
MessageLatencyWithThreshold);
SimpleStringMessage result;
request.setTestFieldString("Hello Qt!");
reply = _client->testMethod(request);
reply->subscribe(this, [reply, &result]() { result = reply->read<SimpleStringMessage>(); });
QTRY_COMPARE_WITH_TIMEOUT(result.testFieldString(), request.testFieldString(),
MessageLatencyWithThreshold);
result.setTestFieldString("");
request.setTestFieldString("Hello Qt1!");
reply = _client->testMethod(request);
reply->subscribe(
this, [reply, &result]() { result = reply->read<SimpleStringMessage>(); },
[]() { QVERIFY(false); });
QTRY_COMPARE_WITH_TIMEOUT(result.testFieldString(), request.testFieldString(),
MessageLatencyWithThreshold);
}
void QtGrpcClientTest::MultipleStreamsTest()
{
const int ExpectedMessageCount = 4;
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Stream");
auto stream = _client->streamTestMethodServerStream(request);
QCOMPARE_EQ(stream, _client->streamTestMethodServerStream(request));
QSignalSpy streamFinishedSpy(stream.get(), &QGrpcStream::finished);
QVERIFY(streamFinishedSpy.isValid());
QSignalSpy streamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamErrorSpy.isValid());
QSignalSpy steamMessageRecievedSpy(stream.get(), &QGrpcStream::messageReceived);
QVERIFY(steamMessageRecievedSpy.isValid());
QObject::connect(stream.get(), &QGrpcStream::messageReceived, this, [&result, stream]() {
SimpleStringMessage ret = stream->read<SimpleStringMessage>();
result.setTestFieldString(result.testFieldString() + ret.testFieldString());
});
QTRY_COMPARE_EQ_WITH_TIMEOUT(streamFinishedSpy.count(), 1,
MessageLatencyWithThreshold * ExpectedMessageCount);
QCOMPARE_EQ(streamErrorSpy.count(), 0);
QCOMPARE_EQ(steamMessageRecievedSpy.count(), ExpectedMessageCount);
QCOMPARE_EQ(result.testFieldString(), "Stream1Stream2Stream3Stream4");
}
void QtGrpcClientTest::MultipleStreamsCancelTest()
{
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Stream");
auto stream = _client->streamTestMethodServerStream(request);
auto streamNext = _client->streamTestMethodServerStream(request);
QCOMPARE_EQ(stream, streamNext);
QSignalSpy streamFinishedSpy(stream.get(), &QGrpcStream::finished);
QVERIFY(streamFinishedSpy.isValid());
QSignalSpy streamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamErrorSpy.isValid());
QSignalSpy streamNextFinishedSpy(streamNext.get(), &QGrpcStream::finished);
QVERIFY(streamNextFinishedSpy.isValid());
QSignalSpy streamNextErrorSpy(streamNext.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamNextErrorSpy.isValid());
streamNext->abort();
QCOMPARE(streamFinishedSpy.count(), 1);
QCOMPARE(streamErrorSpy.count(), 0);
QCOMPARE(streamNextFinishedSpy.count(), 1);
QCOMPARE(streamNextErrorSpy.count(), 0);
stream = _client->streamTestMethodServerStream(request);
QCOMPARE_NE(stream, streamNext);
streamNext = _client->streamTestMethodServerStream(request);
QCOMPARE_EQ(stream, streamNext);
QSignalSpy otherStreamFinishedSpy(stream.get(), &QGrpcStream::finished);
QVERIFY(streamFinishedSpy.isValid());
QSignalSpy otherStreamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(otherStreamErrorSpy.isValid());
QSignalSpy otherStreamNextFinishedSpy(streamNext.get(), &QGrpcStream::finished);
QVERIFY(streamNextFinishedSpy.isValid());
QSignalSpy otherStreamNextErrorSpy(streamNext.get(), &QGrpcStream::errorOccurred);
QVERIFY(otherStreamNextErrorSpy.isValid());
stream->abort();
QCOMPARE(otherStreamFinishedSpy.count(), 1);
QCOMPARE_EQ(otherStreamErrorSpy.count(), 0);
QCOMPARE(otherStreamNextFinishedSpy.count(), 1);
QCOMPARE_EQ(otherStreamNextErrorSpy.count(), 0);
}
void QtGrpcClientTest::CallNonCompatibleArgRetTest()
{
const QtProtobuf::sint32 TestValue = 2048;
const QString TestValueString = QString::number(TestValue);
SimpleIntMessage request;
request.setTestField(TestValue);
auto result = QSharedPointer<SimpleStringMessage>::create();
QCOMPARE_EQ(_client->testMethodNonCompatibleArgRet(request, result.get()), QGrpcStatus::Ok);
QCOMPARE_EQ(result->testFieldString(), TestValueString);
}
void QtGrpcClientTest::CallStringThreadTest()
{
SimpleStringMessage request;
auto result = QSharedPointer<SimpleStringMessage>::create();
request.setTestFieldString("Hello Qt from thread!");
QSignalSpy clientErrorSpy(_client.get(), &TestService::Client::errorOccurred);
QVERIFY(clientErrorSpy.isValid());
bool ok = false;
QSharedPointer<QThread> thread(QThread::create([&]() {
ok = _client->testMethod(request, result.get()) == QGrpcStatus::Ok;
}));
thread->start();
QTRY_COMPARE_EQ_WITH_TIMEOUT(clientErrorSpy.count(), 1, MessageLatencyWithThreshold);
QVERIFY(!ok);
QVERIFY(result->testFieldString().isEmpty());
QVERIFY(qvariant_cast<QGrpcStatus>(clientErrorSpy.at(0).first())
.message()
.startsWith("QAbstractGrpcClient::call is called from different "
"thread."));
}
void QtGrpcClientTest::StringAsyncThreadTest()
{
SimpleStringMessage request;
SimpleStringMessage result;
request.setTestFieldString("Hello Qt from thread!");
QSignalSpy clientErrorSpy(_client.get(), &TestService::Client::errorOccurred);
QVERIFY(clientErrorSpy.isValid());
QSharedPointer<QThread> thread(QThread::create([&]() {
QEventLoop waiter;
QSharedPointer<QGrpcCallReply> reply = _client->testMethod(request);
QObject::connect(reply.get(), &QGrpcCallReply::finished, &waiter,
[reply, &result, &waiter]() {
result = reply->read<SimpleStringMessage>();
waiter.quit();
});
waiter.exec();
}));
thread->start();
QTRY_COMPARE_EQ_WITH_TIMEOUT(clientErrorSpy.count(), 1, FailTimeout);
QTRY_VERIFY(result.testFieldString().isEmpty());
QTRY_VERIFY(qvariant_cast<QGrpcStatus>(clientErrorSpy.at(0).first())
.message()
.startsWith("QAbstractGrpcClient::call is called from different "
"thread."));
}
void QtGrpcClientTest::StreamStringThreadTest()
{
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Stream");
QSignalSpy clientErrorSpy(_client.get(), &TestService::Client::errorOccurred);
QVERIFY(clientErrorSpy.isValid());
int i = 0;
QSharedPointer<QThread> thread(QThread::create([&]() {
QEventLoop waiter;
auto stream = _client->streamTestMethodServerStream(request);
QObject::connect(stream.get(), &QGrpcStream::messageReceived, &waiter,
[&result, &i, &waiter, stream]() {
SimpleStringMessage ret = stream->read<SimpleStringMessage>();
result.setTestFieldString(result.testFieldString()
+ ret.testFieldString());
if (++i == 4)
waiter.quit();
});
waiter.exec();
}));
thread->start();
QTRY_COMPARE_EQ_WITH_TIMEOUT(clientErrorSpy.count(), 1, FailTimeout);
QTRY_VERIFY(result.testFieldString().isEmpty());
QTRY_VERIFY(qvariant_cast<QGrpcStatus>(clientErrorSpy.at(0).first())
.message()
.startsWith("QAbstractGrpcClient::stream is called from different "
"thread."));
}
void QtGrpcClientTest::StreamCancelWhileErrorTimeoutTest()
{
SimpleStringMessage result;
SimpleStringMessage request;
request.setTestFieldString("Stream");
auto stream = _client->streamTestMethodServerStream(request);
QSignalSpy streamFinishedSpy(stream.get(), &QGrpcStream::finished);
QVERIFY(streamFinishedSpy.isValid());
QSignalSpy streamErrorSpy(stream.get(), &QGrpcStream::errorOccurred);
QVERIFY(streamErrorSpy.isValid());
stream->abort();
stream.reset();
QTRY_COMPARE_EQ_WITH_TIMEOUT(streamFinishedSpy.count(), 1, MessageLatencyWithThreshold);
QCOMPARE_EQ(streamErrorSpy.count(), 0);
}
QTEST_MAIN(QtGrpcClientTest)
#include "tst_grpc_client.moc"
Reference in New Issue View Git Blame Copy Permalink