1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
#include "internal_sys_mock.hpp"
#include "io_mock.hpp"
#include "lpc.hpp"
#include "progress_mock.hpp"
#include <cstring>
#include <ipmiblob/test/blob_interface_mock.hpp>
#include <gtest/gtest.h>
namespace host_tool
{
namespace
{
using ::testing::_;
using ::testing::ContainerEq;
using ::testing::Gt;
using ::testing::Invoke;
using ::testing::NotNull;
using ::testing::Return;
using ::testing::StrEq;
TEST(LpcHandleTest, verifySendsFileContents)
{
internal::InternalSysMock sysMock;
ipmiblob::BlobInterfaceMock blobMock;
HostIoInterfaceMock ioMock;
ProgressMock progMock;
const std::uint32_t address = 0xfedc1000;
const std::uint32_t length = 0x1000;
LpcDataHandler handler(&blobMock, &ioMock, address, length, &progMock,
&sysMock);
std::uint16_t session = 0xbeef;
std::string filePath = "/asdf";
int fileDescriptor = 5;
const int fakeFileSize = 100;
LpcRegion host_lpc_buf;
host_lpc_buf.address = address;
host_lpc_buf.length = length;
std::vector<std::uint8_t> bytes(sizeof(host_lpc_buf));
std::memcpy(bytes.data(), &host_lpc_buf, sizeof(host_lpc_buf));
EXPECT_CALL(blobMock, writeMeta(session, 0, ContainerEq(bytes)));
std::vector<std::uint8_t> data = {0x01, 0x02, 0x03};
EXPECT_CALL(sysMock, open(StrEq(filePath.c_str()), _))
.WillOnce(Return(fileDescriptor));
EXPECT_CALL(sysMock, getSize(StrEq(filePath.c_str())))
.WillOnce(Return(fakeFileSize));
EXPECT_CALL(sysMock, read(_, NotNull(), Gt(data.size())))
.WillOnce(Invoke([&data](int, void* buf, std::size_t) {
std::memcpy(buf, data.data(), data.size());
return data.size();
}))
.WillOnce(Return(0));
EXPECT_CALL(ioMock, write(_, data.size(), _))
.WillOnce(Invoke([&data](const std::size_t, const std::size_t length,
const void* const source) {
EXPECT_THAT(std::memcmp(source, data.data(), data.size()), 0);
return true;
}));
EXPECT_CALL(blobMock, writeBytes(session, 0, _));
EXPECT_CALL(sysMock, close(fileDescriptor)).WillOnce(Return(0));
EXPECT_TRUE(handler.sendContents(filePath, session));
}
} // namespace
} // namespace host_tool
|
