Corrigir não enviar dois User-Agent quando o usuário envia o seu próprio

tiagohm · Jan 26, 2020 · f38aaab · f38aaab
1 parent 58144eb
commit f38aaab
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/lib/src/http2_transport.dart b/lib/src/http2_transport.dart
@@ -265,8 +265,10 @@ class Http2Transport implements Transport {
 
   static int _obtainContentLength(Headers headers) {
     final contentLength = headers.first(HttpHeaders.contentLengthHeader);
-    if (contentLength == null)
+
+    if (contentLength == null) {
       return -1;
+    }
 
     try {
       return int.parse(contentLength[0]);

diff --git a/lib/src/http_transport.dart b/lib/src/http_transport.dart
@@ -153,7 +153,13 @@ class HttpTransport implements Transport {
 
       // Headers.
       request.headers?.forEach((key, value) {
-        clientRequest.headers.add(key, value);
+        switch (key) {
+          case HttpHeaders.userAgentHeader:
+            clientRequest.headers.set(key, value);
+            break;
+          default:
+            clientRequest.headers.add(key, value);
+        }
       });
 
       // Body.

diff --git a/test/recipes_test.dart b/test/recipes_test.dart
@@ -63,7 +63,7 @@ void main() {
 
     expect(response.body.contentType.type, 'application');
     expect(response.body.contentType.subType, 'json');
-    expect(response.body.contentLength, 221);
+    expect(response.body.contentLength, 429);
     expect(response.code, 200);
     expect(response.isSuccess, true);
     expect(response.message, 'OK');
@@ -345,7 +345,9 @@ void main() {
     final data = await response.body.data.raw();
     await response.body.close();
 
-    expect(data.length, 50);
+    print(data);
+
+    expect(data.length, 30);
   });
 
   test('Gzip', () async {