Accesso ai cookie all'interno del test unitario nel contesto AspNet.TestHost.TestServer su ASP.NET 5/MVC 6

Question

Non esiste un modo semplice per ottenere un accesso a nell'oggetto response eseguendo test di integrazione con AspNet.TestHost.TestServer. I cookie devono essere impostati dall'azione del controller. Qual è il modo migliore per farlo?

  var client = TestServer.Create(app => 
      { 
       app.UseMvc(routes => 
         routes.MapRoute("default", "{controller}/{action}/{id?}")); 
       app.UseIdentity(); 
      }).CreateClient(); 

      var request = new HttpRequestMessage(HttpMethod.Get, "account/login"); 
      var response = await client.SendAsync(request); 

      // how to get an access to cookie container ????????? 
      // response.Cookies prop doesn't exist 
      Assert.NotEmpty(response.Cookies["auth"]); 

soluzione che vedo è quello di estendere istanza del TestServer, tornare istanza di una classe CustomClientHandler : ClientHandler e sovrascrivere l'intero processo di invio di una richiesta in tale gestore, ma ha bisogno letteralmente di cambiare ogni logica, tranne relativamente piccolo codice del TestServer.

Qualche suggerimento migliore su come implementare un accesso ai cookie in una risposta?

Answer 1

I cookie sono solo intestazioni. Ecco un esempio di prendere un cookie da una risposta e aggiungerlo alla successiva richiesta: https://github.com/aspnet/Mvc/blob/538cd9c19121f8d3171cbfddd5d842cbb756df3e/test/Microsoft.AspNet.Mvc.FunctionalTests/TempDataTest.cs#L201-L202

Answer 2

Ho implementato HttpMessageHandler personalizzato che tiene traccia dei cookie.

Usa la riflessione per richiamare l'attuale gestore e legge/imposta le intestazioni dei cookie.

class TestMessageHandler : HttpMessageHandler 
{ 
    delegate Task<HttpResponseMessage> HandlerSendAsync(HttpRequestMessage message, CancellationToken token); 

    private readonly HandlerSendAsync nextDelegate; 
    private readonly CookieContainer cookies = new System.Net.CookieContainer(); 

    public TestMessageHandler(HttpMessageHandler next) 
    { 
     if(next == null) throw new ArgumentNullException(nameof(next)); 

     nextDelegate = (HandlerSendAsync) 
          next.GetType() 
           .GetTypeInfo() 
           .GetMethod("SendAsync", BindingFlags.NonPublic | BindingFlags.Instance) 
           .CreateDelegate(typeof(HandlerSendAsync), next); 
    } 

    protected override async Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken) 
    { 
     request.Headers.Add("Cookie", cookies.GetCookieHeader(request.RequestUri)); 

     var resp = await nextDelegate(request, cancellationToken).ConfigureAwait(false); 

     if (resp.Headers.TryGetValues("Set-Cookie", out var newCookies)) 
     { 
      foreach (var item in SetCookieHeaderValue.ParseList(newCookies.ToList())) 
      { 
       cookies.Add(request.RequestUri, new Cookie(item.Name, item.Value, item.Path)); 
      } 
     } 

     return resp; 
    } 
} 

E poi si crea il HttpClient in questo modo:

var httpClient = new HttpClient(
        new TestMessageHandler(
         server.CreateHandler())); 

TestMessageHandler ora si prende cura dei tracking cookie.