Implement IUserLoginStore

- update nuspec for deploy 1.0.1

Author: g0t4

src/AspNet.Identity.MongoDB/AspNet.Identity.MongoDB.csproj

@@ -57,6 +57,7 @@
     <Compile Include="UserStore.cs" />
   </ItemGroup>
   <ItemGroup>
+    <None Include="AspNet.Identity.MongoDB.nuspec" />
     <None Include="packages.config" />
   </ItemGroup>
   <Import Project="$(MSBuildToolsPath)\Microsoft.CSharp.targets" />

src/AspNet.Identity.MongoDB/AspNet.Identity.MongoDB.nuspec

@@ -2,15 +2,15 @@
 <package >
   <metadata>
     <id>AspNet.Identity.MongoDB</id>
-    <version>1.0.0-alpha1</version>
+    <version>1.0.1-alpha1</version>
     <authors>Wes McClure</authors>
     <owners>Wes McClure</owners>
     <licenseUrl>https://github.com/g0t4/aspnet-identity-mongo/blob/master/LICENSE</licenseUrl>
     <projectUrl>https://github.com/g0t4/aspnet-identity-mongo</projectUrl>
     <iconUrl>https://github.com/g0t4/aspnet-identity-mongo</iconUrl>
     <requireLicenseAcceptance>false</requireLicenseAcceptance>
     <description>A mongodb provider for the new ASP.NET Identity framework. My aim is to ensure this project is well tested and configurable.</description>
-    <releaseNotes>Initial release</releaseNotes>
+    <releaseNotes>Adding IUserLoginStore, IUserRoleStore and IUserPasswordStore interfaces.</releaseNotes>
     <copyright>MIT</copyright>
     <tags>mongo mongodb aspnet identity</tags>
   </metadata>

src/AspNet.Identity.MongoDB/IdentityUser.cs

@@ -1,6 +1,7 @@
 namespace AspNet.Identity.MongoDB
 {
 	using System.Collections.Generic;
+	using System.Linq;
 	using global::MongoDB.Bson;
 	using global::MongoDB.Bson.Serialization.Attributes;
 	using Microsoft.AspNet.Identity;
@@ -11,6 +12,7 @@ public IdentityUser()
 		{
 			Id = ObjectId.GenerateNewId().ToString();
 			Roles = new List<string>();
+			Logins = new List<UserLoginInfo>();
 		}
 
 		[BsonRepresentation(BsonType.ObjectId)]
@@ -34,6 +36,23 @@ public virtual void RemoveRole(string role)
 		[BsonIgnoreIfNull]
 		public virtual string PasswordHash { get; set; }
 
+		[BsonIgnoreIfNull]
+		public List<UserLoginInfo> Logins { get; set; }
+
+		public virtual void AddLogin(UserLoginInfo login)
+		{
+			Logins.Add(login);
+		}
+
+		public virtual void RemoveLogin(UserLoginInfo login)
+		{
+			var loginsToRemove = Logins
+				.Where(l => l.LoginProvider == login.LoginProvider)
+				.Where(l => l.ProviderKey == login.ProviderKey);
+
+			Logins = Logins.Except(loginsToRemove).ToList();
+		}
+
 		public virtual bool HasPassword()
 		{
 			return false;

src/AspNet.Identity.MongoDB/UserStore.cs

@@ -1,12 +1,14 @@
 namespace AspNet.Identity.MongoDB
 {
 	using System.Collections.Generic;
+	using System.Linq;
 	using System.Threading.Tasks;
 	using global::MongoDB.Bson;
 	using global::MongoDB.Driver.Builders;
+	using global::MongoDB.Driver.Linq;
 	using Microsoft.AspNet.Identity;
 
-	public class UserStore<TUser> : IUserStore<TUser>, IUserPasswordStore<TUser>, IUserRoleStore<TUser>
+	public class UserStore<TUser> : IUserStore<TUser>, IUserPasswordStore<TUser>, IUserRoleStore<TUser>, IUserLoginStore<TUser>
 		where TUser : IdentityUser
 	{
 		private readonly IdentityContext _Context;
@@ -86,5 +88,30 @@ public Task<bool> IsInRoleAsync(TUser user, string roleName)
 		{
 			return Task.FromResult(user.Roles.Contains(roleName));
 		}
+
+		public Task AddLoginAsync(TUser user, UserLoginInfo login)
+		{
+			user.AddLogin(login);
+			return Task.FromResult(0);
+		}
+
+		public Task RemoveLoginAsync(TUser user, UserLoginInfo login)
+		{
+			user.RemoveLogin(login);
+			return Task.FromResult(0);
+		}
+
+		public Task<IList<UserLoginInfo>> GetLoginsAsync(TUser user)
+		{
+			return Task.FromResult((IList<UserLoginInfo>) user.Logins);
+		}
+
+		public Task<TUser> FindAsync(UserLoginInfo login)
+		{
+			return Task.Factory
+				.StartNew(() => _Context.Users.AsQueryable<TUser>()
+					.FirstOrDefault(u => u.Logins
+						.Any(l => l.LoginProvider == login.LoginProvider && l.ProviderKey == login.ProviderKey)));
+		}
 	}
 }

src/IntegrationTests/IntegrationTests.csproj

@@ -61,6 +61,7 @@
     <Compile Include="IdentityUserTests.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />
     <Compile Include="UserIntegrationTestsBase.cs" />
+    <Compile Include="UserLoginStoreTests.cs" />
     <Compile Include="UserPasswordStoreTests.cs" />
     <Compile Include="UserRoleStoreTests.cs" />
     <Compile Include="UserStoreTests.cs" />

src/IntegrationTests/UserLoginStoreTests.cs

@@ -0,0 +1,100 @@
+namespace IntegrationTests
+{
+	using System.Linq;
+	using AspNet.Identity.MongoDB;
+	using Microsoft.AspNet.Identity;
+	using NUnit.Framework;
+
+	[TestFixture]
+	public class UserLoginStoreTests : UserIntegrationTestsBase
+	{
+		[Test]
+		public void AddLogin_NewLogin_Adds()
+		{
+			var manager = GetUserManager();
+			var login = new UserLoginInfo("provider", "key");
+			var user = new IdentityUser {UserName = "bob"};
+			manager.Create(user);
+
+			manager.AddLogin(user.Id, login);
+
+			var savedLogin = Users.FindAll().Single().Logins.Single();
+			Expect(savedLogin.LoginProvider, Is.EqualTo("provider"));
+			Expect(savedLogin.ProviderKey, Is.EqualTo("key"));
+		}
+
+
+		[Test]
+		public void RemoveLogin_NewLogin_Removes()
+		{
+			var manager = GetUserManager();
+			var login = new UserLoginInfo("provider", "key");
+			var user = new IdentityUser {UserName = "bob"};
+			manager.Create(user);
+			manager.AddLogin(user.Id, login);
+
+			manager.RemoveLogin(user.Id, login);
+
+			var savedUser = Users.FindAll().Single();
+			Expect(savedUser.Logins, Is.Empty);
+		}
+
+		[Test]
+		public void GetLogins_OneLogin_ReturnsLogin()
+		{
+			var manager = GetUserManager();
+			var login = new UserLoginInfo("provider", "key");
+			var user = new IdentityUser {UserName = "bob"};
+			manager.Create(user);
+			manager.AddLogin(user.Id, login);
+
+			var logins = manager.GetLogins(user.Id);
+
+			var savedLogin = logins.Single();
+			Expect(savedLogin.LoginProvider, Is.EqualTo("provider"));
+			Expect(savedLogin.ProviderKey, Is.EqualTo("key"));
+		}
+
+		[Test]
+		public void Find_UserWithLogin_FindsUser()
+		{
+			var manager = GetUserManager();
+			var login = new UserLoginInfo("provider", "key");
+			var user = new IdentityUser {UserName = "bob"};
+			manager.Create(user);
+			manager.AddLogin(user.Id, login);
+
+			var findUser = manager.Find(login);
+
+			Expect(findUser, Is.Not.Null);
+		}
+
+		[Test]
+		public void Find_UserWithDifferentKey_DoesNotFindUser()
+		{
+			var manager = GetUserManager();
+			var login = new UserLoginInfo("provider", "key");
+			var user = new IdentityUser {UserName = "bob"};
+			manager.Create(user);
+			manager.AddLogin(user.Id, login);
+
+			var findUser = manager.Find(new UserLoginInfo("provider", "otherkey"));
+
+			Expect(findUser, Is.Null);
+		}
+
+		[Test]
+		public void Find_UserWithDifferentProvider_DoesNotFindUser()
+		{
+			var manager = GetUserManager();
+			var login = new UserLoginInfo("provider", "key");
+			var user = new IdentityUser {UserName = "bob"};
+			manager.Create(user);
+			manager.AddLogin(user.Id, login);
+
+			var findUser = manager.Find(new UserLoginInfo("otherprovider", "key"));
+
+			Expect(findUser, Is.Null);
+		}
+	}
+}

src/Tests/IdentityUserTests.cs

@@ -1,6 +1,5 @@
 namespace Tests
 {
-	using System.Collections.Generic;
 	using AspNet.Identity.MongoDB;
 	using MongoDB.Bson;
 	using NUnit.Framework;
@@ -60,7 +59,27 @@ public void Create_NullRoles_DoesNotSerializeRoles()
 
 			Expect(document.Contains("Roles"), Is.False);
 		}
-		
-		// todo consider if we want to not serialize the empty Roles array
+
+		// todo consider if we want to not serialize the empty Roles array, also empty Logins array
+
+		[Test]
+		public void Create_NewIdentityUser_LoginsNotNull()
+		{
+			var user = new IdentityUser();
+
+			Expect(user.Logins, Is.Not.Null);
+		}
+
+		[Test]
+		public void Create_NullLogins_DoesNotSerializeLogins()
+		{
+			// serialized nulls can cause havoc in deserialization, overwriting the constructor's initial empty list 
+			var user = new IdentityUser();
+			user.Logins = null;
+
+			var document = user.ToBsonDocument();
+
+			Expect(document.Contains("Logins"), Is.False);
+		}
 	}
 }