1 // Copyright 2021 Google LLC
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //      http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 #include "commands.hpp"
16 #include "flash_size.hpp"
17 #include "handler_mock.hpp"
18 
19 #include <cstdint>
20 #include <cstring>
21 #include <string>
22 #include <vector>
23 
24 #include <gtest/gtest.h>
25 
26 #define MAX_IPMI_BUFFER 64
27 
28 using ::testing::Return;
29 
30 namespace google
31 {
32 namespace ipmi
33 {
34 
35 TEST(FlashSizeCommandTest, InvalidCommandLength)
36 {
37     // GetFlashSizeRequest is one byte, let's send 0.
38     std::vector<std::uint8_t> request = {};
39     size_t dataLen = request.size();
40     std::uint8_t reply[MAX_IPMI_BUFFER];
41 
42     HandlerMock hMock;
43     EXPECT_EQ(IPMI_CC_REQ_DATA_LEN_INVALID,
44               getFlashSize(request.data(), reply, &dataLen, &hMock));
45 }
46 
47 TEST(FlashSizeCommandTest, ValidRequest)
48 {
49     std::vector<std::uint8_t> request = {SysOEMCommands::SysGetFlashSize};
50     size_t dataLen = request.size();
51     std::uint8_t reply[MAX_IPMI_BUFFER];
52     uint32_t flashSize = 5422312; // 0x52BCE8
53 
54     HandlerMock hMock;
55     EXPECT_CALL(hMock, getFlashSize()).WillOnce(Return(flashSize));
56     EXPECT_EQ(IPMI_CC_OK,
57               getFlashSize(request.data(), reply, &dataLen, &hMock));
58     EXPECT_EQ(dataLen, 5);
59     EXPECT_EQ(reply[4], 0);
60     EXPECT_EQ(reply[3], 0x52);
61     EXPECT_EQ(reply[2], 0xBC);
62     EXPECT_EQ(reply[1], 0xE8);
63 }
64 
65 } // namespace ipmi
66 } // namespace google
67