Jason Ling | 84bff8b | 2020-11-06 13:45:09 -0800 | [diff] [blame^] | 1 | #include "create_action_map.hpp" |
Benjamin Fair | a67e9e9 | 2019-11-15 14:47:42 -0800 | [diff] [blame] | 2 | #include "firmware_handler.hpp" |
| 3 | #include "flags.hpp" |
| 4 | #include "image_mock.hpp" |
Benjamin Fair | a67e9e9 | 2019-11-15 14:47:42 -0800 | [diff] [blame] | 5 | #include "util.hpp" |
| 6 | |
| 7 | #include <memory> |
| 8 | #include <utility> |
| 9 | #include <vector> |
| 10 | |
| 11 | #include <gtest/gtest.h> |
| 12 | |
| 13 | namespace ipmi_flash |
| 14 | { |
| 15 | namespace |
| 16 | { |
| 17 | |
| 18 | class FirmwareOpenFailTest : public ::testing::TestWithParam<std::uint16_t> |
Patrick Venture | 9b37b09 | 2020-05-28 20:58:57 -0700 | [diff] [blame] | 19 | {}; |
Benjamin Fair | a67e9e9 | 2019-11-15 14:47:42 -0800 | [diff] [blame] | 20 | |
| 21 | TEST_P(FirmwareOpenFailTest, WithFlags) |
| 22 | { |
Patrick Venture | 4934daa | 2020-09-22 16:37:44 -0700 | [diff] [blame] | 23 | std::vector<DataHandlerPack> data; |
| 24 | data.emplace_back(FirmwareFlags::UpdateFlags::ipmi, nullptr); |
| 25 | data.emplace_back(FirmwareFlags::UpdateFlags::p2a, nullptr); |
| 26 | data.emplace_back(FirmwareFlags::UpdateFlags::lpc, nullptr); |
Benjamin Fair | a67e9e9 | 2019-11-15 14:47:42 -0800 | [diff] [blame] | 27 | |
| 28 | std::vector<HandlerPack> blobs; |
Patrick Venture | c6ba8ff | 2020-09-23 12:42:57 -0700 | [diff] [blame] | 29 | blobs.emplace_back(hashBlobId, std::make_unique<ImageHandlerMock>()); |
| 30 | blobs.emplace_back("asdf", std::make_unique<ImageHandlerMock>()); |
Benjamin Fair | a67e9e9 | 2019-11-15 14:47:42 -0800 | [diff] [blame] | 31 | |
| 32 | auto handler = FirmwareBlobHandler::CreateFirmwareBlobHandler( |
Patrick Venture | da4d4a4 | 2020-09-28 11:41:05 -0700 | [diff] [blame] | 33 | std::move(blobs), std::move(data), CreateActionMap("asdf")); |
Benjamin Fair | a67e9e9 | 2019-11-15 14:47:42 -0800 | [diff] [blame] | 34 | |
| 35 | EXPECT_FALSE(handler->open(0, GetParam(), "asdf")); |
| 36 | } |
| 37 | |
| 38 | const std::vector<std::uint16_t> OpenFailParams{ |
| 39 | /* These first 4 fail because they don't have the "write" flag */ |
| 40 | 0b000 << 8, |
| 41 | 0b110 << 8, |
| 42 | 0b101 << 8, |
| 43 | 0b011 << 8, |
| 44 | /* Next 1 doesn't specify any transport */ |
| 45 | blobs::OpenFlags::write | 0b000 << 8, |
| 46 | /* Next 3 specify 2 reserved transport bits at the same time. This isn't |
| 47 | * allowed because older code expects these first 3 bits to be mutually |
| 48 | * exclusive. |
| 49 | */ |
| 50 | blobs::OpenFlags::write | 0b110 << 8, |
| 51 | blobs::OpenFlags::write | 0b101 << 8, |
| 52 | blobs::OpenFlags::write | 0b011 << 8, |
| 53 | }; |
| 54 | |
| 55 | INSTANTIATE_TEST_CASE_P(WithFlags, FirmwareOpenFailTest, |
| 56 | ::testing::ValuesIn(OpenFailParams)); |
| 57 | |
| 58 | } // namespace |
| 59 | } // namespace ipmi_flash |