Implement IUserTwoFactorStore

Author: g0t4

src/AspNet.Identity.MongoDB/IdentityUser.cs

@@ -3,7 +3,6 @@
 	using System.Collections.Generic;
 	using System.Linq;
 	using System.Security.Claims;
-	using System.Threading.Tasks;
 	using global::MongoDB.Bson;
 	using global::MongoDB.Bson.Serialization.Attributes;
 	using Microsoft.AspNet.Identity;
@@ -33,6 +32,8 @@ public IdentityUser()
 
 		public virtual bool PhoneNumberConfirmed { get; set; }
 
+		public virtual bool TwoFactorEnabled { get; set; }
+
 		[BsonIgnoreIfNull]
 		public List<string> Roles { get; set; }
 

src/AspNet.Identity.MongoDB/UserStore.cs

@@ -17,7 +17,8 @@ public class UserStore<TUser> : IUserStore<TUser>,
 		IUserEmailStore<TUser>,
 		IUserClaimStore<TUser>,
 		IQueryableUserStore<TUser>,
-		IUserPhoneNumberStore<TUser>
+		IUserPhoneNumberStore<TUser>,
+		IUserTwoFactorStore<TUser, string>
 		where TUser : IdentityUser
 	{
 		private readonly IdentityContext _Context;
@@ -205,5 +206,16 @@ public Task SetPhoneNumberConfirmedAsync(TUser user, bool confirmed)
 			user.PhoneNumberConfirmed = confirmed;
 			return Task.FromResult(0);
 		}
+
+		public Task SetTwoFactorEnabledAsync(TUser user, bool enabled)
+		{
+			user.TwoFactorEnabled = enabled;
+			return Task.FromResult(0);
+		}
+
+		public Task<bool> GetTwoFactorEnabledAsync(TUser user)
+		{
+			return Task.FromResult(user.TwoFactorEnabled);
+		}
 	}
 }

src/IntegrationTests/IntegrationTests.csproj

@@ -70,6 +70,7 @@
     <Compile Include="UserRoleStoreTests.cs" />
     <Compile Include="UserSecurityStampStoreTests.cs" />
     <Compile Include="UserStoreTests.cs" />
+    <Compile Include="UserTwoFactorStoreTests.cs" />
   </ItemGroup>
   <ItemGroup>
     <None Include="packages.config" />

src/IntegrationTests/UserTwoFactorStoreTests.cs

@@ -0,0 +1,35 @@
+namespace IntegrationTests
+{
+	using AspNet.Identity.MongoDB;
+	using Microsoft.AspNet.Identity;
+	using NUnit.Framework;
+
+	[TestFixture]
+	public class UserTwoFactorStoreTests : UserIntegrationTestsBase
+	{
+		[Test]
+		public void SetTwoFactorEnabled()
+		{
+			var user = new IdentityUser {UserName = "bob"};
+			var manager = GetUserManager();
+			manager.Create(user);
+
+			manager.SetTwoFactorEnabled(user.Id, true);
+
+			Expect(manager.GetTwoFactorEnabled(user.Id));
+		}
+
+		[Test]
+		public void ClearTwoFactorEnabled_PreviouslyEnabled_NotEnabled()
+		{
+			var user = new IdentityUser {UserName = "bob"};
+			var manager = GetUserManager();
+			manager.Create(user);
+			manager.SetTwoFactorEnabled(user.Id, true);
+
+			manager.SetTwoFactorEnabled(user.Id, false);
+
+			Expect(manager.GetTwoFactorEnabled(user.Id), Is.False);
+		}
+	}
+}