sdk/lib/zxio/tests/test_directory_server_base.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_DIRECTORY_SERVER_BASE_H_
#define LIB_ZXIO_TESTS_TEST_DIRECTORY_SERVER_BASE_H_

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

#include <zxtest/zxtest.h>

namespace zxio_tests {

// This is a test friendly implementation of a fuchsia_io::Directory server
// that simply returns ZX_ERR_NOT_SUPPORTED for every operation other than
// fuchsia.io.Directory/Close.
class TestDirectoryServerBase : public fuchsia_io::testing::Directory_TestBase {
 public:
  TestDirectoryServerBase() = default;
  virtual ~TestDirectoryServerBase() = default;

  void NotImplemented_(const std::string& name, fidl::CompleterBase& completer) final {
    ADD_FAILURE("unexpected message received: %s", name.c_str());
    completer.Close(ZX_ERR_NOT_SUPPORTED);
  }

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

  // Exercised by |zxio_close|.
  void Close2(Close2RequestView request, Close2Completer::Sync& completer) override {
    completer.Reply({});
    // After the reply, we should close the connection.
    completer.Close(ZX_OK);
  }
};

}  // namespace zxio_tests

#endif  // LIB_ZXIO_TESTS_TEST_DIRECTORY_SERVER_BASE_H_