flutter_flutter/packages/flutter/test/services/default_binary_messenger_test.dart

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

import 'dart:convert';
import 'dart:typed_data';
import 'dart:ui' as ui;
import 'package:flutter/services.dart';
import 'package:flutter_test/flutter_test.dart';

void main() {
  TestWidgetsFlutterBinding.ensureInitialized();

  ByteData _makeByteData(String str) {
    final List<int> list = utf8.encode(str);
    final ByteBuffer buffer =
        list is Uint8List ? list.buffer : Uint8List.fromList(list).buffer;
    return ByteData.view(buffer);
  }

  test('default binary messenger calls callback once', () async {
    int count = 0;
    const String channel = 'foo';
    ServicesBinding.instance!.defaultBinaryMessenger.handlePlatformMessage(
      channel,
      _makeByteData('bar'),
      (ByteData? message) async {
        count += 1;
      },
    );
    expect(count, equals(0));
    await ui.channelBuffers.drain(channel, (ByteData? data, ui.PlatformMessageResponseCallback callback) async {
      callback(null);
    });
    expect(count, equals(1));
  });

  test('can check the handler', () {
    Future<ByteData> handler(ByteData? call) => Future<ByteData>.value(null);
    final BinaryMessenger messenger = ServicesBinding.instance!.defaultBinaryMessenger;

    expect(messenger.checkMessageHandler('test_channel', null), true);
    expect(messenger.checkMessageHandler('test_channel', handler), false);
    messenger.setMessageHandler('test_channel', handler);
    expect(messenger.checkMessageHandler('test_channel', handler), true);
    messenger.setMessageHandler('test_channel', null);
  });

  test('can check the mock handler', () {
    Future<ByteData> handler(ByteData? call) => Future<ByteData>.value(null);
    final BinaryMessenger messenger = ServicesBinding.instance!.defaultBinaryMessenger;

    expect(messenger.checkMockMessageHandler('test_channel', null), true);
    expect(messenger.checkMockMessageHandler('test_channel', handler), false);
    messenger.setMockMessageHandler('test_channel', handler);
    expect(messenger.checkMockMessageHandler('test_channel', handler), true);
    messenger.setMockMessageHandler('test_channel', null);
  });
}