From f9a3a4b3949dddecae413b97904c895b2cd887bf Mon Sep 17 00:00:00 2001
From: JoramWilander <jwawilander@gmail.com>
Date: Wed, 30 Mar 2016 12:49:29 -0400
Subject: Add MFA functionality

---
 store/sql_user_store_test.go | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

(limited to 'store/sql_user_store_test.go')

diff --git a/store/sql_user_store_test.go b/store/sql_user_store_test.go
index 8f2366136..dcd2440ac 100644
--- a/store/sql_user_store_test.go
+++ b/store/sql_user_store_test.go
@@ -502,3 +502,47 @@ func TestUserUnreadCount(t *testing.T) {
 		t.Fatal("should have 3 unread messages")
 	}
 }
+
+func TestUserStoreUpdateMfaSecret(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	Must(store.User().Save(&u1))
+
+	time.Sleep(100 * time.Millisecond)
+
+	if err := (<-store.User().UpdateMfaSecret(u1.Id, "12345")).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	// should pass, no update will occur though
+	if err := (<-store.User().UpdateMfaSecret("junk", "12345")).Err; err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestUserStoreUpdateMfaActive(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	Must(store.User().Save(&u1))
+
+	time.Sleep(100 * time.Millisecond)
+
+	if err := (<-store.User().UpdateMfaActive(u1.Id, true)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	if err := (<-store.User().UpdateMfaActive(u1.Id, false)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	// should pass, no update will occur though
+	if err := (<-store.User().UpdateMfaActive("junk", true)).Err; err != nil {
+		t.Fatal(err)
+	}
+}
-- 
cgit v1.2.3-1-g7c22