1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55


base / fuchsia / test_interface_impl.cc [blame]

// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/fuchsia/test_interface_impl.h"

#include <utility>

#include "base/memory/weak_ptr.h"
#include "base/run_loop.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace base {

TestInterfaceImpl::TestInterfaceImpl() = default;
TestInterfaceImpl::~TestInterfaceImpl() = default;

void TestInterfaceImpl::Add(int32_t a, int32_t b, AddCallback callback) {
  callback(a + b);
}

zx_status_t VerifyTestInterface(
    fidl::InterfacePtr<testfidl::TestInterface>& ptr) {
  // Call the service and wait for response.
  RunLoop run_loop;
  zx_status_t result = ZX_ERR_INTERNAL;
  base::WeakPtrFactory<zx_status_t> weak_result(&result);

  ptr.set_error_handler(
      [quit = run_loop.QuitClosure(),
       weak_result = weak_result.GetWeakPtr()](zx_status_t status) {
        if (weak_result)
          *weak_result = status;
        std::move(quit).Run();
      });

  ptr->Add(2, 2,
           [quit = run_loop.QuitClosure(),
            weak_result = weak_result.GetWeakPtr()](int32_t value) {
             EXPECT_EQ(value, 4);
             if (weak_result)
               *weak_result = ZX_OK;
             std::move(quit).Run();
           });

  run_loop.Run();

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

  return result;
}

}  // namespace base