sdk/lib/zxio/tests/test_file_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_FILE_SERVER_BASE_H_
#define LIB_ZXIO_TESTS_TEST_FILE_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::File server
// that simply returns ZX_ERR_NOT_SUPPORTED for every operation other than
// fuchsia.io.File/Close.
class TestFileServerBase : public fuchsia_io::testing::File_TestBase {
 public:
  TestFileServerBase() = default;
  virtual ~TestFileServerBase() = 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);
  }
};

// This implementation provides a simple Read() implementation.
class TestReadFileServer : public TestFileServerBase {
 public:
  void Read(ReadRequestView request, ReadCompleter::Sync& completer) final {
    fidl::Arena fidl_allocator;
    fidl::VectorView<uint8_t> read_data(fidl_allocator, sizeof(kTestData));
    memcpy(read_data.mutable_data(), kTestData, sizeof(kTestData));
    completer.Reply(ZX_OK, read_data);
  }

  static constexpr char kTestData[] = "abcdef";
};

class TestVmofileServer : public TestFileServerBase {
 public:
  void set_seek_offset(uint64_t seek_start_offset) { seek_start_offset_ = seek_start_offset; }

 private:
  // Constructing a Vmofile instance requires a server responding to a seek call
  // over the fuchsia.io.File protocol. This is a test implementation smart enough
  // to respond to this call.
  void Seek(SeekRequestView request, SeekCompleter::Sync& completer) final {
    if (request->start != fuchsia_io::wire::SeekOrigin::kStart || request->offset != 0) {
      ADD_FAILURE("unsupported Seek received start %d offset %ld", request->start, request->offset);
      completer.Close(ZX_ERR_NOT_SUPPORTED);
      return;
    }
    completer.Reply(ZX_OK, seek_start_offset_);
  }

  uint64_t seek_start_offset_;
};

}  // namespace zxio_tests

#endif  // LIB_ZXIO_TESTS_TEST_FILE_SERVER_BASE_H_