sdk/lib/zxio/tests/test_node_server.h

// Copyright 2021 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.

#ifndef LIB_ZXIO_TESTS_TEST_NODE_SERVER_H_
#define LIB_ZXIO_TESTS_TEST_NODE_SERVER_H_

#include <fidl/fuchsia.io/cpp/wire_test_base.h>
#include <lib/fit/function.h>

#include <zxtest/zxtest.h>

namespace zxio_tests {

// Implementation of a fuchsia_io::Node server that implements Close() and
// creates a test failure for all other messages.
class CloseOnlyNodeServer : public fidl::testing::WireTestBase<fuchsia_io::Node> {
  void NotImplemented_(const std::string& name, fidl::CompleterBase& completer) final {
    ADD_FAILURE() << "unexpected message received: " << name;
    completer.Close(ZX_ERR_NOT_SUPPORTED);
  }

  // Exercised by |zxio_close|.
  void Close(CloseCompleter::Sync& completer) final {
    completer.ReplySuccess();
    // After the reply, we should close the connection.
    completer.Close(ZX_OK);
  }
};

}  // namespace zxio_tests

#endif  // LIB_ZXIO_TESTS_TEST_NODE_SERVER_H_