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 "errors.hpp"
17 #include "handler_mock.hpp"
18 #include "helper.hpp"
19 #include "machine_name.hpp"
20 
21 #include <cstdint>
22 #include <cstring>
23 #include <string>
24 #include <vector>
25 
26 #include <gtest/gtest.h>
27 
28 #define MAX_IPMI_BUFFER 64
29 
30 using ::testing::Return;
31 using ::testing::Throw;
32 
33 namespace google
34 {
35 namespace ipmi
36 {
37 
38 TEST(MachineNameCommandTest, InvalidFile)
39 {
40     std::vector<std::uint8_t> request = {};
41     ::testing::StrictMock<HandlerMock> hMock;
42     EXPECT_CALL(hMock, getMachineName()).WillOnce(Throw(IpmiException(5)));
43 
44     EXPECT_EQ(::ipmi::response(5), getMachineName(request, &hMock));
45 }
46 
47 TEST(MachineNameCommandTest, CachesValidRequest)
48 {
49     std::vector<std::uint8_t> request = {};
50     const std::string ret = "Machine";
51 
52     ::testing::StrictMock<HandlerMock> hMock;
53     EXPECT_CALL(hMock, getMachineName()).WillOnce(Return(ret));
54 
55     auto reply = getMachineName(request, &hMock);
56     auto result = ValidateReply(reply);
57     auto& data = result.second;
58 
59     EXPECT_EQ(sizeof(GetMachineNameReply) + ret.size(), data.size());
60     EXPECT_EQ(SysOEMCommands::SysMachineName, result.first);
61     EXPECT_EQ(ret.length(), data[0]);
62     EXPECT_EQ(ret.data(),
63               std::string(data.begin() + sizeof(struct GetMachineNameReply),
64                           data.end()));
65 }
66 
67 } // namespace ipmi
68 } // namespace google
69