blob: 03dbd6b0dbba05ca7185a35e8f6da8005a8f9f50 (
plain)
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
//-----------------------------------------------------------------------
// <copyright file="TestWebRequestHandler.cs" company="Andrew Arnott">
// Copyright (c) Andrew Arnott. All rights reserved.
// </copyright>
//-----------------------------------------------------------------------
namespace DotNetOpenAuth.Test.Mocks {
using System;
using System.IO;
using System.Net;
using System.Text;
using DotNetOpenAuth.Messaging;
using DotNetOpenAuth.OAuth.ChannelElements;
internal class TestWebRequestHandler : IDirectWebRequestHandler {
private Stream postEntity;
/// <summary>
/// Gets or sets the callback used to provide the mock response for the mock request.
/// </summary>
internal Func<HttpWebRequest, IncomingWebResponse> Callback { get; set; }
/// <summary>
/// Gets the stream that was written out as if on an HTTP request.
/// </summary>
internal Stream RequestEntityStream {
get {
if (this.postEntity == null) {
return null;
}
Stream result = new MemoryStream();
long originalPosition = this.postEntity.Position;
this.postEntity.Position = 0;
this.postEntity.CopyTo(result);
this.postEntity.Position = originalPosition;
result.Position = 0;
return result;
}
}
/// <summary>
/// Gets the stream that was written out as if on an HTTP request as an ordinary string.
/// </summary>
internal string RequestEntityAsString {
get {
if (this.postEntity == null) {
return null;
}
StreamReader reader = new StreamReader(this.RequestEntityStream);
return reader.ReadToEnd();
}
}
#region IWebRequestHandler Members
public bool CanSupport(DirectWebRequestOptions options) {
return true;
}
/// <summary>
/// Prepares an <see cref="HttpWebRequest"/> that contains an POST entity for sending the entity.
/// </summary>
/// <param name="request">The <see cref="HttpWebRequest"/> that should contain the entity.</param>
/// <returns>
/// The writer the caller should write out the entity data to.
/// </returns>
public Stream GetRequestStream(HttpWebRequest request) {
return this.GetRequestStream(request, DirectWebRequestOptions.None);
}
public Stream GetRequestStream(HttpWebRequest request, DirectWebRequestOptions options) {
this.postEntity = new MemoryStream();
return this.postEntity;
}
/// <summary>
/// Processes an <see cref="HttpWebRequest"/> and converts the
/// <see cref="HttpWebResponse"/> to a <see cref="Response"/> instance.
/// </summary>
/// <param name="request">The <see cref="HttpWebRequest"/> to handle.</param>
/// <returns>
/// An instance of <see cref="Response"/> describing the response.
/// </returns>
public IncomingWebResponse GetResponse(HttpWebRequest request) {
return this.GetResponse(request, DirectWebRequestOptions.None);
}
public IncomingWebResponse GetResponse(HttpWebRequest request, DirectWebRequestOptions options) {
if (this.Callback == null) {
throw new InvalidOperationException("Set the Callback property first.");
}
return this.Callback(request);
}
#endregion
#region IDirectSslWebRequestHandler Members
public Stream GetRequestStream(HttpWebRequest request, bool requireSsl) {
ErrorUtilities.VerifyProtocol(!requireSsl || request.RequestUri.Scheme == Uri.UriSchemeHttps, "disallowed request");
return this.GetRequestStream(request);
}
public IncomingWebResponse GetResponse(HttpWebRequest request, bool requireSsl) {
ErrorUtilities.VerifyProtocol(!requireSsl || request.RequestUri.Scheme == Uri.UriSchemeHttps, "disallowed request");
var result = this.GetResponse(request);
ErrorUtilities.VerifyProtocol(!requireSsl || result.FinalUri.Scheme == Uri.UriSchemeHttps, "disallowed request");
return result;
}
#endregion
}
}
|
