blob: 0cbd156d2ded24c6618eae1145f2a58f724c1c5c [file] [log] [blame] [edit]
// Copyright 2019 The Dawn Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#ifndef SRC_DAWN_TESTS_UNITTESTS_WIRE_WIRETEST_H_
#define SRC_DAWN_TESTS_UNITTESTS_WIRE_WIRETEST_H_
#include <memory>
#include "dawn/mock_webgpu.h"
#include "gtest/gtest.h"
// Definition of a "Lambda predicate matcher" for GMock to allow checking deep structures
// are passed correctly by the wire.
// Helper templates to extract the argument type of a lambda.
template <typename T>
struct MatcherMethodArgument;
template <typename Lambda, typename Arg>
struct MatcherMethodArgument<bool (Lambda::*)(Arg) const> {
using Type = Arg;
};
template <typename Lambda>
using MatcherLambdaArgument = typename MatcherMethodArgument<decltype(&Lambda::operator())>::Type;
// The matcher itself, unfortunately it isn't able to return detailed information like other
// matchers do.
template <typename Lambda, typename Arg>
class LambdaMatcherImpl : public testing::MatcherInterface<Arg> {
public:
explicit LambdaMatcherImpl(Lambda lambda) : mLambda(lambda) {}
void DescribeTo(std::ostream* os) const override { *os << "with a custom matcher"; }
bool MatchAndExplain(Arg value, testing::MatchResultListener* listener) const override {
if (!mLambda(value)) {
*listener << "which doesn't satisfy the custom predicate";
return false;
}
return true;
}
private:
Lambda mLambda;
};
// Use the MatchesLambda as follows:
//
// EXPECT_CALL(foo, Bar(MatchesLambda([](ArgType arg) -> bool {
// return CheckPredicateOnArg(arg);
// })));
template <typename Lambda>
inline testing::Matcher<MatcherLambdaArgument<Lambda>> MatchesLambda(Lambda lambda) {
return MakeMatcher(new LambdaMatcherImpl<Lambda, MatcherLambdaArgument<Lambda>>(lambda));
}
class StringMessageMatcher : public testing::MatcherInterface<const char*> {
public:
StringMessageMatcher() {}
bool MatchAndExplain(const char* message,
testing::MatchResultListener* listener) const override {
if (message == nullptr) {
*listener << "missing error message";
return false;
}
if (std::strlen(message) <= 1) {
*listener << "message is truncated";
return false;
}
return true;
}
void DescribeTo(std::ostream* os) const override { *os << "valid error message"; }
void DescribeNegationTo(std::ostream* os) const override { *os << "invalid error message"; }
};
inline testing::Matcher<const char*> ValidStringMessage() {
return MakeMatcher(new StringMessageMatcher());
}
namespace dawn::wire {
class WireClient;
class WireServer;
namespace client {
class MemoryTransferService;
} // namespace client
namespace server {
class MemoryTransferService;
} // namespace server
} // namespace dawn::wire
namespace utils {
class TerribleCommandBuffer;
}
class WireTest : public testing::Test {
protected:
WireTest();
~WireTest() override;
void SetUp() override;
void TearDown() override;
void FlushClient(bool success = true);
void FlushServer(bool success = true);
void DefaultApiDeviceWasReleased();
testing::StrictMock<MockProcTable> api;
WGPUDevice apiDevice;
WGPUQueue apiQueue;
WGPUDevice device;
WGPUQueue queue;
dawn::wire::WireServer* GetWireServer();
dawn::wire::WireClient* GetWireClient();
void DeleteServer();
void DeleteClient();
private:
void SetupIgnoredCallExpectations();
virtual dawn::wire::client::MemoryTransferService* GetClientMemoryTransferService();
virtual dawn::wire::server::MemoryTransferService* GetServerMemoryTransferService();
std::unique_ptr<dawn::wire::WireServer> mWireServer;
std::unique_ptr<dawn::wire::WireClient> mWireClient;
std::unique_ptr<utils::TerribleCommandBuffer> mS2cBuf;
std::unique_ptr<utils::TerribleCommandBuffer> mC2sBuf;
};
#endif // SRC_DAWN_TESTS_UNITTESTS_WIRE_WIRETEST_H_