packages/webview_flutter/webview_flutter_android/android/src/test/java/io/flutter/plugins/webviewflutter/DownloadListenerTest.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.never;
import static org.mockito.Mockito.reset;
import static org.mockito.Mockito.verify;

import android.webkit.DownloadListener;
import io.flutter.plugins.webviewflutter.DownloadListenerHostApiImpl.DownloadListenerCreator;
import io.flutter.plugins.webviewflutter.DownloadListenerHostApiImpl.DownloadListenerImpl;
import org.junit.After;
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 DownloadListenerTest {
  @Rule public MockitoRule mockitoRule = MockitoJUnit.rule();

  @Mock public DownloadListenerFlutterApiImpl mockFlutterApi;

  InstanceManager instanceManager;
  DownloadListenerHostApiImpl hostApiImpl;
  DownloadListenerImpl downloadListener;

  @Before
  public void setUp() {
    instanceManager = InstanceManager.open(identifier -> {});

    final DownloadListenerCreator downloadListenerCreator =
        new DownloadListenerCreator() {
          @Override
          public DownloadListenerImpl createDownloadListener(
              DownloadListenerFlutterApiImpl flutterApi) {
            downloadListener = super.createDownloadListener(flutterApi);
            return downloadListener;
          }
        };

    hostApiImpl =
        new DownloadListenerHostApiImpl(instanceManager, downloadListenerCreator, mockFlutterApi);
    hostApiImpl.create(0L);
  }

  @After
  public void tearDown() {
    instanceManager.close();
  }

  @Test
  public void postMessage() {
    downloadListener.onDownloadStart(
        "https://www.google.com", "userAgent", "contentDisposition", "mimetype", 54);
    verify(mockFlutterApi)
        .onDownloadStart(
            eq(downloadListener),
            eq("https://www.google.com"),
            eq("userAgent"),
            eq("contentDisposition"),
            eq("mimetype"),
            eq(54L),
            any());

    reset(mockFlutterApi);
    downloadListener.release();
    downloadListener.onDownloadStart("", "", "", "", 23);
    verify(mockFlutterApi, never())
        .onDownloadStart((DownloadListener) any(), any(), any(), any(), any(), eq(23), any());
  }
}