// 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_deprecated/reader.h>
#include <lib/inspect_deprecated/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_deprecated::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_deprecated::Inspector inspector_;
  inspect_deprecated::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_deprecated::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_deprecated::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_deprecated::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