runtime_probe/utils/function_test_utils_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 "runtime_probe/utils/function_test_utils.h"

#include <string>
#include <utility>

#include <base/json/json_reader.h>
#include <gtest/gtest.h>

#include "runtime_probe/probe_function.h"

namespace runtime_probe {
namespace {

class MockProbeFunction : public ProbeFunction {
  using ProbeFunction::ProbeFunction;

 public:
  NAME_PROBE_FUNCTION("mock_function");

  DataType EvalImpl() const override { return {}; }
};

class FuntionTestUtilsTest : public BaseFunctionTest {};

TEST_F(FuntionTestUtilsTest, FakeProbeFunction) {
  auto probe_result = R"JSON(
      [{
        "field_1": "value_1",
        "field_2": "value_2"
      }]
    )JSON";
  FakeProbeFunction probe_function(probe_result);
  auto result = EvalProbeFunction(&probe_function);
  auto ans = CreateProbeResultFromJson(probe_result);
  EXPECT_EQ(result, ans);
}

}  // namespace
}  // namespace runtime_probe