packages/webview_flutter/webview_flutter_android/android/src/test/java/io/flutter/plugins/webviewflutter/JavaScriptChannelTest.java

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

package io.flutter.plugins.webviewflutter;

import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.verify;

import android.os.Handler;
import io.flutter.plugins.webviewflutter.GeneratedAndroidWebView.JavaScriptChannelFlutterApi;
import io.flutter.plugins.webviewflutter.JavaScriptChannelHostApiImpl.JavaScriptChannelCreator;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

public class JavaScriptChannelTest {
  @Rule public MockitoRule mockitoRule = MockitoJUnit.rule();

  @Mock public GeneratedAndroidWebView.JavaScriptChannelFlutterApi mockFlutterApi;

  InstanceManager testInstanceManager;
  JavaScriptChannelHostApiImpl testHostApiImpl;
  JavaScriptChannel testJavaScriptChannel;

  @Before
  public void setUp() {
    testInstanceManager = new InstanceManager();

    final JavaScriptChannelCreator javaScriptChannelCreator =
        new JavaScriptChannelCreator() {
          @Override
          JavaScriptChannel createJavaScriptChannel(
              Long instanceId,
              JavaScriptChannelFlutterApi javaScriptChannelFlutterApi,
              String channelName,
              Handler platformThreadHandler) {
            testJavaScriptChannel =
                super.createJavaScriptChannel(
                    instanceId, javaScriptChannelFlutterApi, channelName, platformThreadHandler);
            return testJavaScriptChannel;
          }
        };

    testHostApiImpl =
        new JavaScriptChannelHostApiImpl(
            testInstanceManager, javaScriptChannelCreator, mockFlutterApi, new Handler());
    testHostApiImpl.create(0L, "aChannelName");
  }

  @Test
  public void postMessage() {
    testJavaScriptChannel.postMessage("A message post.");
    verify(mockFlutterApi).postMessage(eq(0L), eq("A message post."), any());
  }
}