lib/ui/window/platform_message_response_dart_unittests.cc

// Copyright 2013 The Flutter 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 "flutter/common/task_runners.h"
#include "flutter/fml/mapping.h"
#include "flutter/fml/synchronization/waitable_event.h"
#include "flutter/lib/ui/window/platform_message_response_dart.h"
#include "flutter/runtime/dart_vm.h"
#include "flutter/shell/common/shell_test.h"
#include "flutter/shell/common/thread_host.h"
#include "flutter/testing/testing.h"

namespace flutter {
namespace testing {

TEST_F(ShellTest, PlatformMessageResponseDart) {
  bool did_pass = false;
  auto message_latch = std::make_shared<fml::AutoResetWaitableEvent>();
  TaskRunners task_runners("test",                  // label
                           GetCurrentTaskRunner(),  // platform
                           CreateNewThread(),       // raster
                           CreateNewThread(),       // ui
                           CreateNewThread()        // io
  );

  auto nativeCallPlatformMessageResponseDart =
      [ui_task_runner =
           task_runners.GetUITaskRunner()](Dart_NativeArguments args) {
        auto dart_state = std::make_shared<tonic::DartState>();
        auto response = fml::MakeRefCounted<PlatformMessageResponseDart>(
            tonic::DartPersistentValue(tonic::DartState::Current(),
                                       Dart_GetNativeArgument(args, 0)),
            ui_task_runner, "foobar");
        uint8_t* data = static_cast<uint8_t*>(malloc(100));
        auto mapping = std::make_unique<fml::MallocMapping>(data, 100);
        response->Complete(std::move(mapping));
      };

  AddNativeCallback("CallPlatformMessageResponseDart",
                    CREATE_NATIVE_ENTRY(nativeCallPlatformMessageResponseDart));

  auto nativeFinishCallResponse = [message_latch,
                                   &did_pass](Dart_NativeArguments args) {
    did_pass =
        tonic::DartConverter<bool>::FromDart(Dart_GetNativeArgument(args, 0));
    message_latch->Signal();
  };

  AddNativeCallback("FinishCallResponse",
                    CREATE_NATIVE_ENTRY(nativeFinishCallResponse));

  Settings settings = CreateSettingsForFixture();

  std::unique_ptr<Shell> shell = CreateShell(settings, task_runners);

  ASSERT_TRUE(shell->IsSetup());
  auto configuration = RunConfiguration::InferFromSettings(settings);
  configuration.SetEntrypoint("platformMessageResponseTest");

  shell->RunEngine(std::move(configuration), [](auto result) {
    ASSERT_EQ(result, Engine::RunStatus::Success);
  });

  message_latch->Wait();

  ASSERT_TRUE(did_pass);
  DestroyShell(std::move(shell), task_runners);
}

}  // namespace testing
}  // namespace flutter