runtime_probe/daemon_test.cc

// Copyright 2022 The ChromiumOS Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <memory>
#include <optional>
#include <utility>

#include <base/files/file_path.h>
#include <base/values.h>
#include <brillo/dbus/mock_dbus_method_response.h>
#include <gmock/gmock.h>
#include <gtest/gtest.h>

#include "runtime_probe/daemon.h"
#include "runtime_probe/proto_bindings/runtime_probe.pb.h"
#include "runtime_probe/system/context_mock_impl.h"

namespace runtime_probe {

namespace {

using ::brillo::dbus_utils::MockDBusMethodResponse;

using ::testing::NiceMock;
using ::testing::Return;

class DaemonTest : public ::testing::Test {
 protected:
  Daemon daemon_;
  org::chromium::RuntimeProbeInterface* const dbus_adaptor_{&daemon_};

 private:
  ContextMockImpl mock_context_;
};

TEST_F(DaemonTest, ProbeCategories_LoadDefaultFailed) {
  ProbeRequest request;
  request.set_probe_default_category(true);
  std::optional<ProbeResult> reply;
  auto response =
      std::make_unique<MockDBusMethodResponse<ProbeResult>>(nullptr);
  response->save_return_args(&reply);
  dbus_adaptor_->ProbeCategories(std::move(response), request);
  EXPECT_TRUE(reply);
  EXPECT_EQ(reply->error(), RUNTIME_PROBE_ERROR_PROBE_CONFIG_INVALID);
}

TEST_F(DaemonTest, ProbeSsfcComponents_LoadDefaultFailed) {
  ProbeSsfcComponentsRequest request;
  std::optional<ProbeSsfcComponentsResponse> reply;
  auto response =
      std::make_unique<MockDBusMethodResponse<ProbeSsfcComponentsResponse>>(
          nullptr);
  response->save_return_args(&reply);
  dbus_adaptor_->ProbeSsfcComponents(std::move(response), request);
  EXPECT_TRUE(reply);
  EXPECT_EQ(reply->error(), RUNTIME_PROBE_ERROR_PROBE_CONFIG_INVALID);
}

// TODO(kevinptt): Add more test cases for D-Bus methods.

}  // namespace

}  // namespace runtime_probe