1 files changed, 52 insertions, 0 deletions
diff --git a/src/DotNetOpenId.Test/RelyingParty/IProviderEndpointTests.cs b/src/DotNetOpenId.Test/RelyingParty/IProviderEndpointTests.cs
new file mode 100644
index 0000000..04f5873
--- /dev/null
+++ b/src/DotNetOpenId.Test/RelyingParty/IProviderEndpointTests.cs
@@ -0,0 +1,52 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using NUnit.Framework;
+using DotNetOpenId.RelyingParty;
+using DotNetOpenId.Extensions.SimpleRegistration;
+using DotNetOpenId.Extensions.AttributeExchange;
+
+namespace DotNetOpenId.Test.RelyingParty {
+	[TestFixture]
+	public class IProviderEndpointTests {
+		IRelyingPartyApplicationStore store;
+		Realm realm = new Realm(TestSupport.GetFullUrl(TestSupport.ConsumerPage).AbsoluteUri);
+		Uri returnTo = TestSupport.GetFullUrl(TestSupport.ConsumerPage);
+
+		[SetUp]
+		public void SetUp() {
+			store = new ApplicationMemoryStore();
+			if (!UntrustedWebRequest.WhitelistHosts.Contains("localhost"))
+				UntrustedWebRequest.WhitelistHosts.Add("localhost");
+		}
+
+		[Test]
+		public void IsExtensionSupportedTest() {
+			OpenIdRelyingParty rp = new OpenIdRelyingParty(store, null, null);
+			Identifier id = TestSupport.GetFullUrl("xrdsdiscovery/xrds20.aspx");
+			IAuthenticationRequest request = rp.CreateRequest(id, realm, returnTo);
+			IProviderEndpoint provider = request.Provider;
+			Assert.IsTrue(provider.IsExtensionSupported<ClaimsRequest>());
+			Assert.IsTrue(provider.IsExtensionSupported(typeof(ClaimsRequest)));
+			Assert.IsFalse(provider.IsExtensionSupported<FetchRequest>());
+			Assert.IsFalse(provider.IsExtensionSupported(typeof(FetchRequest)));
+
+			// Test the AdditionalTypeUris list by pulling from an XRDS page with one of the
+			// TypeURIs that only shows up in that list.
+			id = TestSupport.GetFullUrl("xrdsdiscovery/xrds10.aspx");
+			request = rp.CreateRequest(id, realm, returnTo);
+			Assert.IsTrue(provider.IsExtensionSupported<ClaimsRequest>());
+			Assert.IsTrue(provider.IsExtensionSupported(typeof(ClaimsRequest)));
+		}
+
+		[Test]
+		public void UriTest() {
+			OpenIdRelyingParty rp = new OpenIdRelyingParty(store, null, null);
+			Identifier id = TestSupport.GetFullUrl("xrdsdiscovery/xrds20.aspx");
+			IAuthenticationRequest request = rp.CreateRequest(id, realm, returnTo);
+			IProviderEndpoint provider = request.Provider;
+			Assert.AreEqual(new Uri("http://a/b"), provider.Uri);
+		}
+	}
+}