garnet/examples/inspection/echo_connection_unittest.cc

// Copyright 2019 The Fuchsia 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 "echo_connection.h"

#include <lib/gtest/real_loop_fixture.h>
#include <lib/inspect/reader.h>
#include <lib/inspect/testing/inspect.h>
#include <lib/sys/cpp/testing/component_context_provider.h>

#include "lib/fidl/cpp/binding.h"

namespace example {
namespace testing {

using namespace fidl::examples::echo;
using namespace inspect::testing;

class EchoConnectionTest : public gtest::RealLoopFixture {
 public:
  EchoConnectionTest()
      : inspector_(),
        inspect_tree_(inspector_.CreateTree("root")),
        connection_(inspect_tree_.GetRoot().CreateChild("connection"), stats_),
        echo_(),
        binding_(&connection_, echo_.NewRequest().TakeChannel()) {}

 protected:
  inspect::Inspector inspector_;
  inspect::Tree inspect_tree_;
  std::shared_ptr<EchoConnectionStats> stats_;
  EchoConnection connection_;
  EchoPtr echo_;
  fidl::Binding<EchoConnection> binding_;
};

TEST_F(EchoConnectionTest, EchoString_MultipleRequests) {
  ::fidl::StringPtr message = "bogus";
  echo_->EchoString("Hello World!",
                    [&](::fidl::StringPtr retval) { message = retval; });
  RunLoopUntilIdle();
  EXPECT_EQ("Hello World!", message);
  auto hierarchy_result = inspect::ReadFromVmo(inspect_tree_.GetVmo());
  ASSERT_TRUE(hierarchy_result.is_ok());
  EXPECT_THAT(
      hierarchy_result.take_value(),
      ChildrenMatch(::testing::ElementsAre(NodeMatches(AllOf(
          NameMatches("connection"), MetricList(::testing::UnorderedElementsAre(
                                         UIntMetricIs("bytes_processed", 12),
                                         UIntMetricIs("requests", 1))))))));

  // Call the service again.
  echo_->EchoString("Hello Again!",
                    [&](::fidl::StringPtr retval) { message = retval; });
  RunLoopUntilIdle();
  EXPECT_EQ("Hello Again!", message);
  hierarchy_result = inspect::ReadFromVmo(inspect_tree_.GetVmo());
  ASSERT_TRUE(hierarchy_result.is_ok());
  EXPECT_THAT(
      hierarchy_result.take_value(),
      ChildrenMatch(::testing::ElementsAre(NodeMatches(AllOf(
          NameMatches("connection"), MetricList(::testing::UnorderedElementsAre(
                                         UIntMetricIs("bytes_processed", 24),
                                         UIntMetricIs("requests", 2))))))));
}

// Answer "" with ""
TEST_F(EchoConnectionTest, EchoString_Empty) {
  fidl::StringPtr message = "bogus";
  echo_->EchoString("", [&](::fidl::StringPtr retval) { message = retval; });
  RunLoopUntilIdle();
  EXPECT_EQ("", message);
  auto hierarchy_result = inspect::ReadFromVmo(inspect_tree_.GetVmo());
  ASSERT_TRUE(hierarchy_result.is_ok());
  EXPECT_THAT(
      hierarchy_result.take_value(),
      ChildrenMatch(::testing::ElementsAre(NodeMatches(AllOf(
          NameMatches("connection"), MetricList(::testing::UnorderedElementsAre(
                                         UIntMetricIs("bytes_processed", 0),
                                         UIntMetricIs("requests", 1))))))));
}

}  // namespace testing
}  // namespace example