// Copyright 2021 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include "libec/get_version_command.h"

namespace ec {
namespace {

using ::testing::Return;

TEST(GetVersionCommand, GetVersionCommand) {
  GetVersionCommand cmd;
  EXPECT_EQ(cmd.Version(), 0);
  EXPECT_EQ(cmd.Command(), EC_CMD_GET_VERSION);
}

// Mock the underlying EcCommand to test.
class GetVersionCommandTest : public testing::Test {
 public:
  class MockGetVersionCommand : public GetVersionCommand {
   public:
    using GetVersionCommand::GetVersionCommand;
    MOCK_METHOD(struct ec_response_get_version*, Resp, (), (override));
    MOCK_METHOD(bool, EcCommandRun, (int fd), (override));
  };
};

TEST_F(GetVersionCommandTest, Success) {
  MockGetVersionCommand mock_command;
  struct ec_response_get_version response = {.version_string_ro = "ro_version",
                                             .version_string_rw = "rw_version",
                                             .current_image = EC_IMAGE_RW};
  EXPECT_CALL(mock_command, Resp).WillRepeatedly(Return(&response));
  EXPECT_CALL(mock_command, EcCommandRun).WillOnce(Return(true));

  EXPECT_TRUE(mock_command.Run(-1));

  EXPECT_EQ(mock_command.ROVersion(), "ro_version");
  EXPECT_EQ(mock_command.RWVersion(), "rw_version");
  EXPECT_EQ(mock_command.Image(), EC_IMAGE_RW);
}

TEST_F(GetVersionCommandTest, ROVersionNotNullTerminated) {
  MockGetVersionCommand mock_command;
  struct ec_response_get_version response {};
  std::fill(response.version_string_ro,
            response.version_string_ro + sizeof(response.version_string_ro),
            'a');
  EXPECT_CALL(mock_command, Resp).WillRepeatedly(Return(&response));
  EXPECT_CALL(mock_command, EcCommandRun).WillOnce(Return(true));
  EXPECT_TRUE(mock_command.Run(-1));

  EXPECT_EQ(mock_command.ROVersion(), "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
}

TEST_F(GetVersionCommandTest, RWVersionNotNullTerminated) {
  MockGetVersionCommand mock_command;
  struct ec_response_get_version response {};
  std::fill(response.version_string_rw,
            response.version_string_rw + sizeof(response.version_string_rw),
            'a');
  EXPECT_CALL(mock_command, Resp).WillRepeatedly(Return(&response));
  EXPECT_CALL(mock_command, EcCommandRun).WillOnce(Return(true));
  EXPECT_TRUE(mock_command.Run(-1));

  EXPECT_EQ(mock_command.RWVersion(), "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
}

}  // namespace
}  // namespace ec