blob: 2531b53ad2cfe56082438e076beaab2ff84e56cb [file] [log] [blame] [edit]
/*
* Copyright (C) 2016 The Android Open Source Project
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* * Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* * Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in
* the documentation and/or other materials provided with the
* distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
* FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
* COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
* OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/
#include "socket_mock.h"
#include <gtest/gtest.h>
SocketMock::SocketMock() : Socket(INVALID_SOCKET) {}
SocketMock::~SocketMock() {
if (!events_.empty()) {
ADD_FAILURE() << events_.size() << " event(s) were not handled";
}
}
bool SocketMock::Send(const void* data, size_t length) {
if (events_.empty()) {
ADD_FAILURE() << "Send() was called when no message was expected";
return false;
}
if (events_.front().type != EventType::kSend) {
ADD_FAILURE() << "Send() was called out-of-order";
return false;
}
std::string message(reinterpret_cast<const char*>(data), length);
if (events_.front().message != message) {
ADD_FAILURE() << "Send() expected " << events_.front().message << ", but got " << message;
return false;
}
bool return_value = events_.front().status;
events_.pop();
return return_value;
}
// Mock out multi-buffer send to be one large send, since that's what it should looks like from
// the user's perspective.
bool SocketMock::Send(std::vector<cutils_socket_buffer_t> buffers) {
std::string data;
for (const auto& buffer : buffers) {
data.append(reinterpret_cast<const char*>(buffer.data), buffer.length);
}
return Send(data.data(), data.size());
}
ssize_t SocketMock::Receive(void* data, size_t length, int /*timeout_ms*/) {
if (events_.empty()) {
ADD_FAILURE() << "Receive() was called when no message was ready";
return -1;
}
const Event& event = events_.front();
if (event.type != EventType::kReceive) {
ADD_FAILURE() << "Receive() was called out-of-order";
return -1;
}
const std::string& message = event.message;
if (message.length() > length) {
ADD_FAILURE() << "Receive(): not enough bytes (" << length << ") for " << message;
return -1;
}
receive_timed_out_ = event.status;
ssize_t return_value = message.length();
// Empty message indicates failure.
if (message.empty()) {
return_value = -1;
} else {
memcpy(data, message.data(), message.length());
}
events_.pop();
return return_value;
}
int SocketMock::Close() {
return 0;
}
std::unique_ptr<Socket> SocketMock::Accept() {
if (events_.empty()) {
ADD_FAILURE() << "Accept() was called when no socket was ready";
return nullptr;
}
if (events_.front().type != EventType::kAccept) {
ADD_FAILURE() << "Accept() was called out-of-order";
return nullptr;
}
std::unique_ptr<Socket> sock = std::move(events_.front().sock);
events_.pop();
return sock;
}
void SocketMock::ExpectSend(std::string message) {
events_.push(Event(EventType::kSend, std::move(message), true, nullptr));
}
void SocketMock::ExpectSendFailure(std::string message) {
events_.push(Event(EventType::kSend, std::move(message), false, nullptr));
}
void SocketMock::AddReceive(std::string message) {
events_.push(Event(EventType::kReceive, std::move(message), false, nullptr));
}
void SocketMock::AddReceiveTimeout() {
events_.push(Event(EventType::kReceive, "", true, nullptr));
}
void SocketMock::AddReceiveFailure() {
events_.push(Event(EventType::kReceive, "", false, nullptr));
}
void SocketMock::AddAccept(std::unique_ptr<Socket> sock) {
events_.push(Event(EventType::kAccept, "", false, std::move(sock)));
}
SocketMock::Event::Event(EventType _type, std::string _message, ssize_t _status,
std::unique_ptr<Socket> _sock)
: type(_type), message(_message), status(_status), sock(std::move(_sock)) {}