Ed Tanous | 40e9b92 | 2024-09-10 13:50:16 -0700 | [diff] [blame] | 1 | // SPDX-License-Identifier: Apache-2.0 |
| 2 | // SPDX-FileCopyrightText: Copyright OpenBMC Authors |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 3 | #include "credential_pipe.hpp" |
| 4 | |
Ed Tanous | 41fe81c | 2024-09-02 15:08:41 -0700 | [diff] [blame] | 5 | #include <boost/asio/buffer.hpp> |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 6 | #include <boost/asio/io_context.hpp> |
Ed Tanous | 3bfa3b2 | 2024-01-31 12:18:03 -0800 | [diff] [blame] | 7 | #include <boost/asio/read.hpp> |
| 8 | #include <boost/asio/readable_pipe.hpp> |
| 9 | #include <boost/system/error_code.hpp> |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 10 | |
Ed Tanous | f0b59af | 2024-03-20 13:38:04 -0700 | [diff] [blame] | 11 | #include <array> |
| 12 | #include <cstddef> |
| 13 | #include <functional> |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 14 | #include <string> |
| 15 | |
| 16 | #include <gmock/gmock.h> |
Ed Tanous | f0b59af | 2024-03-20 13:38:04 -0700 | [diff] [blame] | 17 | #include <gtest/gtest.h> |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 18 | |
| 19 | using ::testing::ElementsAre; |
| 20 | |
| 21 | static void handler(boost::asio::io_context& io, |
| 22 | const boost::system::error_code& ec, size_t sent) |
| 23 | { |
| 24 | io.stop(); |
| 25 | EXPECT_FALSE(ec); |
| 26 | EXPECT_EQ(sent, 18); |
| 27 | } |
| 28 | |
| 29 | TEST(CredentialsPipe, BasicSend) |
| 30 | { |
Ed Tanous | 3bfa3b2 | 2024-01-31 12:18:03 -0800 | [diff] [blame] | 31 | boost::asio::io_context io; |
| 32 | boost::asio::readable_pipe testPipe(io); |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 33 | { |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 34 | CredentialsPipe pipe(io); |
Ed Tanous | 3bfa3b2 | 2024-01-31 12:18:03 -0800 | [diff] [blame] | 35 | testPipe = boost::asio::readable_pipe(io, pipe.releaseFd()); |
| 36 | ASSERT_GT(testPipe.native_handle(), 0); |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 37 | pipe.asyncWrite("username", "password", |
| 38 | std::bind_front(handler, std::ref(io))); |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 39 | } |
Ed Tanous | 3bfa3b2 | 2024-01-31 12:18:03 -0800 | [diff] [blame] | 40 | io.run(); |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 41 | std::array<char, 18> buff{}; |
| 42 | boost::system::error_code ec; |
Ed Tanous | 3bfa3b2 | 2024-01-31 12:18:03 -0800 | [diff] [blame] | 43 | boost::asio::read(testPipe, boost::asio::buffer(buff), ec); |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 44 | ASSERT_FALSE(ec); |
Ed Tanous | 11e8f60 | 2023-08-24 14:25:18 -0700 | [diff] [blame] | 45 | |
| 46 | EXPECT_THAT(buff, |
| 47 | ElementsAre('u', 's', 'e', 'r', 'n', 'a', 'm', 'e', '\0', 'p', |
| 48 | 'a', 's', 's', 'w', 'o', 'r', 'd', '\0')); |
| 49 | } |