mirror of https://github.com/polybar/polybar.git
97 lines
3.1 KiB
C++
97 lines
3.1 KiB
C++
#include "ipc/decoder.hpp"
|
|
|
|
#include <cstring>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "common/test.hpp"
|
|
#include "components/logger.hpp"
|
|
#include "gmock/gmock.h"
|
|
#include "ipc/msg.hpp"
|
|
|
|
using namespace polybar;
|
|
using namespace ipc;
|
|
using ::testing::InSequence;
|
|
|
|
static std::vector<uint8_t> get_msg(decltype(MAGIC) magic, uint32_t version, type_t type, const std::string& msg) {
|
|
std::vector<uint8_t> data(HEADER_SIZE);
|
|
header* header = reinterpret_cast<ipc::header*>(data.data());
|
|
std::copy(magic.begin(), magic.end(), header->s.magic);
|
|
header->s.version = version;
|
|
header->s.size = msg.size();
|
|
header->s.type = type;
|
|
|
|
data.insert(data.end(), msg.begin(), msg.end());
|
|
|
|
return data;
|
|
}
|
|
|
|
static decltype(MAGIC) MAGIC_WRONG = {'0', '0', '0', '0', '0', '0', '0'};
|
|
static type_t TYPE_ACTION = to_integral(v0::ipc_type::ACTION);
|
|
|
|
static auto MSG1 = get_msg(MAGIC, VERSION, TYPE_ACTION, "foobar");
|
|
static auto MSG2 = get_msg(MAGIC, VERSION, TYPE_ACTION, "");
|
|
static auto MSG_WRONG1 = get_msg(MAGIC_WRONG, VERSION, TYPE_ACTION, "");
|
|
static auto MSG_WRONG2 = get_msg(MAGIC, 120, TYPE_ACTION, "");
|
|
static auto MSG_WRONG3 = get_msg(MAGIC_WRONG, 120, TYPE_ACTION, "");
|
|
|
|
class MockCallback {
|
|
public:
|
|
MOCK_METHOD(void, cb, (uint8_t version, type_t, const vector<uint8_t>&));
|
|
};
|
|
|
|
static logger null_logger(loglevel::NONE);
|
|
|
|
class DecoderTest : public ::testing::Test {
|
|
protected:
|
|
MockCallback cb;
|
|
decoder dec{null_logger, [this](uint8_t version, auto type, const auto& data) { cb.cb(version, type, data); }};
|
|
};
|
|
|
|
TEST_F(DecoderTest, single_msg1) {
|
|
EXPECT_CALL(cb, cb(0, TYPE_ACTION, vector<uint8_t>(MSG1.begin() + HEADER_SIZE, MSG1.end()))).Times(1);
|
|
EXPECT_NO_THROW(dec.on_read(MSG1.data(), MSG1.size()));
|
|
}
|
|
|
|
TEST_F(DecoderTest, single_msg2) {
|
|
EXPECT_CALL(cb, cb(0, TYPE_ACTION, vector<uint8_t>(MSG2.begin() + HEADER_SIZE, MSG2.end()))).Times(1);
|
|
EXPECT_NO_THROW(dec.on_read(MSG2.data(), MSG2.size()));
|
|
}
|
|
|
|
TEST_F(DecoderTest, single_msg_wrong1) {
|
|
EXPECT_THROW(dec.on_read(MSG_WRONG1.data(), MSG_WRONG1.size()), decoder::error);
|
|
// After an error, any further read fails
|
|
EXPECT_THROW(dec.on_read(MSG1.data(), MSG1.size()), decoder::error);
|
|
}
|
|
|
|
TEST_F(DecoderTest, single_msg_wrong2) {
|
|
EXPECT_THROW(dec.on_read(MSG_WRONG2.data(), MSG_WRONG2.size()), decoder::error);
|
|
// After an error, any further read fails
|
|
EXPECT_THROW(dec.on_read(MSG1.data(), MSG1.size()), decoder::error);
|
|
}
|
|
|
|
TEST_F(DecoderTest, single_msg_wrong3) {
|
|
EXPECT_THROW(dec.on_read(MSG_WRONG3.data(), MSG_WRONG3.size()), decoder::error);
|
|
// After an error, any further read fails
|
|
EXPECT_THROW(dec.on_read(MSG1.data(), MSG1.size()), decoder::error);
|
|
}
|
|
|
|
TEST_F(DecoderTest, byte_by_byte) {
|
|
EXPECT_CALL(cb, cb(0, TYPE_ACTION, vector<uint8_t>(MSG1.begin() + HEADER_SIZE, MSG1.end()))).Times(1);
|
|
for (const uint8_t c : MSG1) {
|
|
EXPECT_NO_THROW(dec.on_read(&c, 1));
|
|
}
|
|
}
|
|
|
|
TEST_F(DecoderTest, multiple) {
|
|
static constexpr int NUM_ITER = 10;
|
|
{
|
|
InSequence seq;
|
|
EXPECT_CALL(cb, cb(0, TYPE_ACTION, vector<uint8_t>(MSG1.begin() + HEADER_SIZE, MSG1.end()))).Times(NUM_ITER);
|
|
}
|
|
|
|
for (int i = 0; i < NUM_ITER; i++) {
|
|
EXPECT_NO_THROW(dec.on_read(MSG1.data(), MSG1.size()));
|
|
}
|
|
}
|