#360 Add Response.Clear() extension.
This commit is contained in:
parent
df33a3cff8
commit
00d5a056d9
|
|
@ -0,0 +1,26 @@
|
||||||
|
// Copyright (c) .NET Foundation. All rights reserved.
|
||||||
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
||||||
|
|
||||||
|
using System;
|
||||||
|
using Microsoft.AspNet.Http.Features;
|
||||||
|
|
||||||
|
namespace Microsoft.AspNet.Http.Extensions
|
||||||
|
{
|
||||||
|
public static class ResponseExtensions
|
||||||
|
{
|
||||||
|
public static void Clear(this HttpResponse response)
|
||||||
|
{
|
||||||
|
if (response.HasStarted)
|
||||||
|
{
|
||||||
|
throw new InvalidOperationException("The response cannot be cleared, it has already started sending.");
|
||||||
|
}
|
||||||
|
response.StatusCode = 200;
|
||||||
|
response.HttpContext.Features.Get<IHttpResponseFeature>().ReasonPhrase = null;
|
||||||
|
response.Headers.Clear();
|
||||||
|
if (response.Body.CanSeek)
|
||||||
|
{
|
||||||
|
response.Body.SetLength(0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,64 @@
|
||||||
|
// Copyright (c) .NET Foundation. All rights reserved.
|
||||||
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
||||||
|
|
||||||
|
using System;
|
||||||
|
using System.Collections.Generic;
|
||||||
|
using System.IO;
|
||||||
|
using System.Threading.Tasks;
|
||||||
|
using Microsoft.AspNet.Http.Features;
|
||||||
|
using Microsoft.AspNet.Http.Internal;
|
||||||
|
using Microsoft.Framework.Primitives;
|
||||||
|
using Xunit;
|
||||||
|
|
||||||
|
namespace Microsoft.AspNet.Http.Extensions
|
||||||
|
{
|
||||||
|
public class ResponseExtensionTests
|
||||||
|
{
|
||||||
|
[Fact]
|
||||||
|
public void Clear_ResetsResponse()
|
||||||
|
{
|
||||||
|
var context = new DefaultHttpContext();
|
||||||
|
context.Response.StatusCode = 201;
|
||||||
|
context.Response.Headers["custom"] = "value";
|
||||||
|
context.Response.Body.Write(new byte[100], 0, 100);
|
||||||
|
|
||||||
|
context.Response.Clear();
|
||||||
|
|
||||||
|
Assert.Equal(200, context.Response.StatusCode);
|
||||||
|
Assert.Equal(string.Empty, context.Response.Headers["custom"].ToString());
|
||||||
|
Assert.Equal(0, context.Response.Body.Length);
|
||||||
|
}
|
||||||
|
|
||||||
|
[Fact]
|
||||||
|
public void Clear_AlreadyStarted_Throws()
|
||||||
|
{
|
||||||
|
var context = new DefaultHttpContext();
|
||||||
|
context.Features.Set<IHttpResponseFeature>(new StartedResponseFeature());
|
||||||
|
|
||||||
|
Assert.Throws<InvalidOperationException>(() => context.Response.Clear());
|
||||||
|
}
|
||||||
|
|
||||||
|
private class StartedResponseFeature : IHttpResponseFeature
|
||||||
|
{
|
||||||
|
public Stream Body { get; set; }
|
||||||
|
|
||||||
|
public bool HasStarted { get { return true; } }
|
||||||
|
|
||||||
|
public IDictionary<string, StringValues> Headers { get; set; }
|
||||||
|
|
||||||
|
public string ReasonPhrase { get; set; }
|
||||||
|
|
||||||
|
public int StatusCode { get; set; }
|
||||||
|
|
||||||
|
public void OnCompleted(Func<object, Task> callback, object state)
|
||||||
|
{
|
||||||
|
throw new NotImplementedException();
|
||||||
|
}
|
||||||
|
|
||||||
|
public void OnStarting(Func<object, Task> callback, object state)
|
||||||
|
{
|
||||||
|
throw new NotImplementedException();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue