1 files changed, 58 insertions, 0 deletions
diff --git a/src/DotNetOpenId.Test/Mocks/MockIdentifier.cs b/src/DotNetOpenId.Test/Mocks/MockIdentifier.cs
new file mode 100644
index 0000000..7ab2d74
--- /dev/null
+++ b/src/DotNetOpenId.Test/Mocks/MockIdentifier.cs
@@ -0,0 +1,58 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using DotNetOpenId.RelyingParty;
+
+namespace DotNetOpenId.Test.Mocks {
+	/// <summary>
+	/// Performs similar to an ordinary <see cref="Identifier"/>, but when called upon
+	/// to perform discovery, it returns a preset list of sevice endpoints to avoid
+	/// having a dependency on a hosted web site to actually perform discovery on.
+	/// </summary>
+	class MockIdentifier : Identifier {
+		IEnumerable<ServiceEndpoint> endpoints;
+		Identifier wrappedIdentifier;
+
+		public MockIdentifier(Identifier wrappedIdentifier, IEnumerable<ServiceEndpoint> endpoints)
+			: base(false) {
+			if (wrappedIdentifier == null) throw new ArgumentNullException("wrappedIdentifier");
+			if (endpoints == null) throw new ArgumentNullException("endpoints");
+			this.wrappedIdentifier = wrappedIdentifier;
+			this.endpoints = endpoints;
+
+			// Register a mock HTTP response to enable discovery of this identifier within the RP
+			// without having to host an ASP.NET site within the test.
+			MockHttpRequest.RegisterMockXrdsResponse(new Uri(wrappedIdentifier.ToString()), endpoints);
+		}
+
+		internal override IEnumerable<ServiceEndpoint> Discover() {
+			return endpoints;
+		}
+
+		internal override Identifier TrimFragment() {
+			return this;
+		}
+
+		internal override bool TryRequireSsl(out Identifier secureIdentifier) {
+			// We take special care to make our wrapped identifier secure, but still
+			// return a mocked (secure) identifier.
+			Identifier secureWrappedIdentifier;
+			bool result = wrappedIdentifier.TryRequireSsl(out secureWrappedIdentifier);
+			secureIdentifier = new MockIdentifier(secureWrappedIdentifier, endpoints);
+			return result;
+		}
+
+		public override string ToString() {
+			return wrappedIdentifier.ToString();
+		}
+
+		public override bool Equals(object obj) {
+			return wrappedIdentifier.Equals(obj);
+		}
+
+		public override int GetHashCode() {
+			return wrappedIdentifier.GetHashCode();
+		}
+	}
+}