1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Copyright (c) Microsoft Corporation. All rights reserved. See License.txt in the project root for license information.
using System.IO;
using System.Linq;
using Moq;
using Xunit;
using Assert = Microsoft.TestCommon.AssertEx;
namespace System.Web.Mvc.Test
{
public class FileStreamResultTest
{
private static readonly Random _random = new Random();
[Fact]
public void ConstructorSetsFileStreamProperty()
{
// Arrange
Stream stream = new MemoryStream();
// Act
FileStreamResult result = new FileStreamResult(stream, "contentType");
// Assert
Assert.Same(stream, result.FileStream);
}
[Fact]
public void ConstructorThrowsIfFileStreamIsNull()
{
// Act & assert
Assert.ThrowsArgumentNull(
delegate { new FileStreamResult(null, "contentType"); }, "fileStream");
}
[Fact]
public void WriteFileCopiesProvidedStreamToOutputStream()
{
// Arrange
int byteLen = 0x1234;
byte[] originalBytes = GetRandomByteArray(byteLen);
MemoryStream originalStream = new MemoryStream(originalBytes);
MemoryStream outStream = new MemoryStream();
Mock<HttpResponseBase> mockResponse = new Mock<HttpResponseBase>();
mockResponse.Setup(r => r.OutputStream).Returns(outStream);
FileStreamResultHelper helper = new FileStreamResultHelper(originalStream, "application/octet-stream");
// Act
helper.PublicWriteFile(mockResponse.Object);
// Assert
byte[] outBytes = outStream.ToArray();
Assert.True(originalBytes.SequenceEqual(outBytes));
mockResponse.Verify();
}
private static byte[] GetRandomByteArray(int length)
{
byte[] bytes = new byte[length];
_random.NextBytes(bytes);
return bytes;
}
private class FileStreamResultHelper : FileStreamResult
{
public FileStreamResultHelper(Stream fileStream, string contentType)
: base(fileStream, contentType)
{
}
public void PublicWriteFile(HttpResponseBase response)
{
WriteFile(response);
}
}
}
}
|
