Add support for replying byte data (#146)

* Add support for replying byte data * fix format
lomsa-dev · Apr 30, 2023 · e5488a5 · e5488a5 · maple135790 · May 1, 2023
1 parent af264ae
commit e5488a5
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 0 deletions.
diff --git a/lib/src/handlers/request_handler.dart b/lib/src/handlers/request_handler.dart
@@ -1,4 +1,5 @@
 import 'dart:convert';
+import 'dart:typed_data';
 
 import 'package:dio/dio.dart';
 import 'package:http_mock_adapter/http_mock_adapter.dart';
@@ -50,6 +51,16 @@ class RequestHandler implements MockServer {
         false;
 
     mockResponse = (requestOptions) {
+      if (data is Uint8List) {
+        return MockResponseBody.fromBytes(
+          data,
+          statusCode,
+          headers: headers,
+          statusMessage: statusMessage,
+          isRedirect: isRedirect,
+          delay: delay,
+        );
+      }
       var rawData = data;
       if (data is MockDataCallback) {
         rawData = data(requestOptions);

diff --git a/lib/src/response.dart b/lib/src/response.dart
@@ -54,4 +54,21 @@ class MockResponseBody extends ResponseBody implements MockResponse {
         isRedirect: isRedirect,
         delay: delay,
       );
+
+  static MockResponseBody fromBytes(
+    Uint8List bytes,
+    int statusCode, {
+    required Map<String, List<String>> headers,
+    String? statusMessage,
+    required bool isRedirect,
+    Duration? delay,
+  }) =>
+      MockResponseBody(
+        Stream.value(bytes),
+        statusCode,
+        headers: headers,
+        statusMessage: statusMessage,
+        isRedirect: isRedirect,
+        delay: delay,
+      );
 }
diff --git a/test/handlers/request_handler_test.dart b/test/handlers/request_handler_test.dart
@@ -1,4 +1,6 @@
+import 'dart:ffi';
 import 'dart:io';
+import 'dart:typed_data';
 
 import 'package:dio/dio.dart';
 import 'package:http_mock_adapter/http_mock_adapter.dart';
@@ -13,6 +15,28 @@ void main() {
     setUp(() {
       requestHandler = RequestHandler();
     });
+    test('sets response data for a status with Uint8List', () async {
+      const statusCode = HttpStatus.ok;
+      final inputData = Uint8List.fromList([1, 2, 3, 4, 5]);
+
+      requestHandler.reply(
+        statusCode,
+        inputData,
+      );
+
+      final statusHandler = requestHandler.mockResponse;
+
+      expect(statusHandler, isNotNull);
+
+      final mockResponseBody =
+          statusHandler(RequestOptions(path: '')) as MockResponseBody;
+      final resolvedData = await BackgroundTransformer().transformResponse(
+        RequestOptions(path: '', responseType: ResponseType.bytes),
+        mockResponseBody,
+      );
+
+      expect(resolvedData, inputData);
+    });
 
     test('sets response data for a status with JSON by default', () async {
       const statusCode = HttpStatus.ok;