base/fuchsia/service_directory_test_base.cc

// Copyright 2018 The Chromium 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 "base/fuchsia/service_directory_test_base.h"

#include <lib/fdio/util.h>
#include <utility>

namespace base {
namespace fuchsia {

ServiceDirectoryTestBase::ServiceDirectoryTestBase() {
  // TODO(https://crbug.com/920920): Remove the ServiceDirectory's implicit
  // "public" sub-directory and update this setup logic.

  // Mount service dir and publish the service.
  fidl::InterfaceHandle<::fuchsia::io::Directory> directory;
  service_directory_ =
      std::make_unique<ServiceDirectory>(directory.NewRequest());
  service_binding_ =
      std::make_unique<ScopedServiceBinding<testfidl::TestInterface>>(
          service_directory_.get(), &test_service_);

  // Create the ServiceDirectoryClient, connected to the "public" sub-directory.
  fidl::InterfaceHandle<::fuchsia::io::Directory> public_directory;
  CHECK_EQ(fdio_open_at(directory.channel().get(), "public", 0,
                        public_directory.NewRequest().TakeChannel().release()),
           ZX_OK);
  public_service_directory_client_ =
      std::make_unique<ServiceDirectoryClient>(std::move(public_directory));

  // Create a ServiceDirectoryClient for the "private" part of the directory.
  root_service_directory_client_ =
      std::make_unique<ServiceDirectoryClient>(std::move(directory));
}

ServiceDirectoryTestBase::~ServiceDirectoryTestBase() = default;

void ServiceDirectoryTestBase::VerifyTestInterface(
    fidl::InterfacePtr<testfidl::TestInterface>* stub,
    zx_status_t expected_error) {
  // Call the service and wait for response.
  base::RunLoop run_loop;
  zx_status_t actual_error = ZX_OK;

  stub->set_error_handler([&run_loop, &actual_error](zx_status_t status) {
    actual_error = status;
    run_loop.Quit();
  });

  (*stub)->Add(2, 2, [&run_loop](int32_t result) {
    EXPECT_EQ(result, 4);
    run_loop.Quit();
  });

  run_loop.Run();

  EXPECT_EQ(expected_error, actual_error);

  // Reset error handler because the current one captures |run_loop| and
  // |error| references which are about to be destroyed.
  stub->set_error_handler(nullptr);
}

}  // namespace fuchsia
}  // namespace base