services/network/mojo_socket_test_util.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 "services/network/mojo_socket_test_util.h"

#include "testing/gtest/include/gtest/gtest.h"

namespace network {

TestSocketObserver::TestSocketObserver() : binding_(this) {}

TestSocketObserver::~TestSocketObserver() {
  EXPECT_EQ(net::OK, read_error_);
  EXPECT_EQ(net::OK, write_error_);
}

mojom::SocketObserverPtr TestSocketObserver::GetObserverPtr() {
  DCHECK(!binding_);

  mojom::SocketObserverPtr ptr;
  binding_.Bind(mojo::MakeRequest(&ptr));
  return ptr;
}

int TestSocketObserver::WaitForReadError() {
  read_loop_.Run();
  int error = read_error_;
  read_error_ = net::OK;
  return error;
}

int TestSocketObserver::WaitForWriteError() {
  write_loop_.Run();
  int error = write_error_;
  write_error_ = net::OK;
  return error;
}

void TestSocketObserver::OnReadError(int net_error) {
  read_error_ = net_error;
  read_loop_.Quit();
}

void TestSocketObserver::OnWriteError(int net_error) {
  write_error_ = net_error;
  write_loop_.Quit();
}

}  // namespace network