@@ -8,8 +8,8 @@ namespace TinyPng.Tests;
8
8
9
9
public class FakeResponseHandler : DelegatingHandler
10
10
{
11
- private readonly Dictionary < Uri , HttpResponseMessage > _fakeGetResponses = new ( ) ;
12
- private readonly Dictionary < Uri , HttpResponseMessage > _fakePostResponses = new ( ) ;
11
+ private readonly Dictionary < Uri , HttpResponseMessage > _fakeGetResponses = [ ] ;
12
+ private readonly Dictionary < Uri , HttpResponseMessage > _fakePostResponses = [ ] ;
13
13
14
14
15
15
public void AddFakeGetResponse ( Uri uri , HttpResponseMessage responseMessage )
@@ -23,12 +23,8 @@ public void AddFakePostResponse(Uri uri, HttpResponseMessage responseMessage)
23
23
24
24
protected override Task < HttpResponseMessage > SendAsync ( HttpRequestMessage request , System . Threading . CancellationToken cancellationToken )
25
25
{
26
- var result = request . Method == HttpMethod . Get && _fakeGetResponses . ContainsKey ( request . RequestUri )
27
- ? _fakeGetResponses [ request . RequestUri ]
28
- : request . Method == HttpMethod . Post && _fakePostResponses . ContainsKey ( request . RequestUri )
29
- ? _fakePostResponses [ request . RequestUri ]
30
- : new HttpResponseMessage ( HttpStatusCode . NotFound ) { RequestMessage = request } ;
31
-
32
- return Task . FromResult ( result ) ;
26
+ if ( request . Method == HttpMethod . Get && _fakeGetResponses . TryGetValue ( request . RequestUri , out var getMessage ) ) { return Task . FromResult ( getMessage ) ; }
27
+ else if ( request . Method == HttpMethod . Post && _fakePostResponses . TryGetValue ( request . RequestUri , out var postMessage ) ) { return Task . FromResult ( postMessage ) ; }
28
+ else { return Task . FromResult ( new HttpResponseMessage ( HttpStatusCode . NotFound ) { RequestMessage = request } ) ; }
33
29
}
34
30
}
0 commit comments