src/devices/tests/ddk-fallback-test/test.cc

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

#include <lib/ddk/platform-defs.h>
#include <lib/driver-integration-test/fixture.h>
#include <lib/fdio/directory.h>
#include <zircon/syscalls.h>

#include <unordered_set>

#include <zxtest/zxtest.h>

using driver_integration_test::IsolatedDevmgr;

class FallbackTest : public zxtest::Test {
 public:
  ~FallbackTest() override = default;
  // Set up and launch the devmgr.
  void LaunchDevmgr(IsolatedDevmgr::Args args) {
    board_test::DeviceEntry dev = {};
    dev.vid = PDEV_VID_TEST;
    dev.pid = PDEV_PID_FALLBACK_TEST;
    dev.did = 0;
    args.device_list.push_back(dev);

    zx_status_t status = IsolatedDevmgr::Create(&args, &devmgr_);
    ASSERT_OK(status);
  }

  // Check that the correct driver was bound. `fallback` indicates if we expect the fallback or
  // not-fallback driver to have bound.
  void CheckDriverBound(bool fallback) {
    fbl::unique_fd fd;
    fbl::String path = fbl::StringPrintf("sys/platform/11:16:0/ddk-%s-test",
                                         fallback ? "fallback" : "not-fallback");
    ASSERT_OK(device_watcher::RecursiveWaitForFile(devmgr_.devfs_root(), path.c_str(), &fd));
    ASSERT_GT(fd.get(), 0);
    ASSERT_OK(fdio_get_service_handle(fd.release(), chan_.reset_and_get_address()));
    ASSERT_NE(chan_.get(), ZX_HANDLE_INVALID);
  }

 protected:
  zx::channel chan_;
  IsolatedDevmgr devmgr_;
};

TEST_F(FallbackTest, TestNotFallbackTakesPriorityDfv1) {
  IsolatedDevmgr::Args args;
  args.use_driver_framework_v2 = false;
  ASSERT_NO_FATAL_FAILURE(LaunchDevmgr(std::move(args)));
  ASSERT_NO_FATAL_FAILURE(CheckDriverBound(false));
}

TEST_F(FallbackTest, TestFallbackBoundWhenAloneDfv1) {
  IsolatedDevmgr::Args args;
  args.use_driver_framework_v2 = false;
  args.driver_disable.push_back("ddk_not_fallback_test");
  ASSERT_NO_FATAL_FAILURE(LaunchDevmgr(std::move(args)));
  ASSERT_NO_FATAL_FAILURE(CheckDriverBound(true));
}

TEST_F(FallbackTest, TestFallbackBoundWhenEagerDfv1) {
  IsolatedDevmgr::Args args;
  args.use_driver_framework_v2 = false;
  args.driver_bind_eager.push_back("fuchsia-boot:///#meta/ddk-fallback-test.cm");
  ASSERT_NO_FATAL_FAILURE(LaunchDevmgr(std::move(args)));
  ASSERT_NO_FATAL_FAILURE(CheckDriverBound(true));
}

TEST_F(FallbackTest, TestNotFallbackTakesPriorityDfv2) {
  IsolatedDevmgr::Args args;
  args.use_driver_framework_v2 = true;
  ASSERT_NO_FATAL_FAILURE(LaunchDevmgr(std::move(args)));
  ASSERT_NO_FATAL_FAILURE(CheckDriverBound(false));
}

TEST_F(FallbackTest, TestFallbackBoundWhenAloneDfv2) {
  IsolatedDevmgr::Args args;
  args.use_driver_framework_v2 = true;
  args.driver_disable.push_back("fuchsia-boot:///#meta/ddk-not-fallback-test.cm");
  ASSERT_NO_FATAL_FAILURE(LaunchDevmgr(std::move(args)));
  ASSERT_NO_FATAL_FAILURE(CheckDriverBound(true));
}