content/browser/handwriting/handwriting_recognition_service_impl_unittest.cc

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "content/browser/handwriting/handwriting_recognition_service_impl.h"

#include <utility>

#include "base/run_loop.h"
#include "base/test/bind.h"
#include "content/browser/handwriting/handwriting_recognizer_impl.h"
#include "content/public/test/test_renderer_host.h"
#include "mojo/public/cpp/bindings/pending_receiver.h"
#include "mojo/public/cpp/bindings/pending_remote.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "third_party/blink/public/mojom/handwriting/handwriting.mojom.h"

namespace content {

class HandwritingRecognitionServiceImplTest : public RenderViewHostTestHarness {
};

TEST_F(HandwritingRecognitionServiceImplTest, CreateHandwritingRecognizer) {
  mojo::Remote<handwriting::mojom::HandwritingRecognitionService> remote;
  HandwritingRecognitionServiceImpl::Create(
      remote.BindNewPipeAndPassReceiver());
  auto model_constraint = handwriting::mojom::HandwritingModelConstraint::New();
  bool is_callback_called = false;
  base::RunLoop runloop;
  remote->CreateHandwritingRecognizer(
      std::move(model_constraint),
      base::BindLambdaForTesting(
          [&](handwriting::mojom::CreateHandwritingRecognizerResult result,
              mojo::PendingRemote<handwriting::mojom::HandwritingRecognizer>
                  remote) {
            EXPECT_EQ(result,
                      handwriting::mojom::CreateHandwritingRecognizerResult::
                          kNotSupported);
            EXPECT_TRUE(!remote);
            is_callback_called = true;
            runloop.Quit();
          }));
  runloop.Run();
  EXPECT_TRUE(is_callback_called);
}

TEST_F(HandwritingRecognitionServiceImplTest, QueryHandwritingRecognizer) {
  mojo::Remote<handwriting::mojom::HandwritingRecognitionService> remote;
  HandwritingRecognitionServiceImpl::Create(
      remote.BindNewPipeAndPassReceiver());

  auto constraint = handwriting::mojom::HandwritingModelConstraint::New();
  constraint->languages.push_back("en");

  bool is_callback_called = false;
  base::RunLoop runloop;
  remote->QueryHandwritingRecognizer(
      std::move(constraint),
      base::BindLambdaForTesting(
          [&](handwriting::mojom::QueryHandwritingRecognizerResultPtr result) {
            // We do not support anything here.
            EXPECT_TRUE(result.is_null());
            is_callback_called = true;
            runloop.Quit();
          }));
  runloop.Run();
  EXPECT_TRUE(is_callback_called);
}

}  // namespace content