// Copyright 2018 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_server_app.h"
#include "lib/component/cpp/testing/test_with_context.h"

namespace echo2 {
namespace testing {

using namespace fidl::examples::echo;

class EchoServerAppForTest : public EchoServerApp {
 public:
  // Expose injecting constructor so we can pass an instrumented Context
  EchoServerAppForTest(std::unique_ptr<component::StartupContext> context)
      : EchoServerApp(std::move(context), false) {}
};

class EchoServerAppTest : public component::testing::TestWithContext {
 public:
  void SetUp() override {
    TestWithContext::SetUp();
    echoServerApp_.reset(new EchoServerAppForTest(TakeContext()));
  }

  void TearDown() override {
    echoServerApp_.reset();
    TestWithContext::TearDown();
  }

 protected:
  EchoPtr echo() {
    EchoPtr echo;
    controller().outgoing_public_services().ConnectToService(echo.NewRequest());
    return echo;
  }

 private:
  std::unique_ptr<EchoServerAppForTest> echoServerApp_;
};

// Answer "Hello World" with "Hello World"
TEST_F(EchoServerAppTest, EchoString_HelloWorld) {
  EchoPtr echo_ = echo();
  ::fidl::StringPtr message = "bogus";
  echo_->EchoString("Hello World!",
                    [&](::fidl::StringPtr retval) { message = retval; });
  RunLoopUntilIdle();
  EXPECT_EQ("Hello World!", message);
}

// Answer "" with ""
TEST_F(EchoServerAppTest, EchoString_Empty) {
  EchoPtr echo_ = echo();
  fidl::StringPtr message = "bogus";
  echo_->EchoString("", [&](::fidl::StringPtr retval) { message = retval; });
  RunLoopUntilIdle();
  EXPECT_EQ("", message);
}

}  // namespace testing
}  // namespace echo2