1 files changed, 276 insertions, 0 deletions
diff --git a/store/sql_user_store_test.go b/store/sql_user_store_test.go
new file mode 100644
index 000000000..4231920a5
--- /dev/null
+++ b/store/sql_user_store_test.go
@@ -0,0 +1,276 @@
+// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved.
+// See License.txt for license information.
+
+package store
+
+import (
+	"github.com/mattermost/platform/model"
+	"strings"
+	"testing"
+)
+
+func TestUserStoreSave(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.Email = model.NewId()
+	u1.Username = model.NewId()
+	u1.TeamId = model.NewId()
+
+	if err := (<-store.User().Save(&u1)).Err; err != nil {
+		t.Fatal("couldn't save user", err)
+	}
+
+	if err := (<-store.User().Save(&u1)).Err; err == nil {
+		t.Fatal("shouldn't be able to update user from save")
+	}
+
+	u1.Id = ""
+	if err := (<-store.User().Save(&u1)).Err; err == nil {
+		t.Fatal("should be unique email")
+	}
+
+	u1.Email = ""
+	if err := (<-store.User().Save(&u1)).Err; err == nil {
+		t.Fatal("should be unique username")
+	}
+
+	u1.Email = strings.Repeat("0123456789", 20)
+	u1.Username = ""
+	if err := (<-store.User().Save(&u1)).Err; err == nil {
+		t.Fatal("should be unique username")
+	}
+
+	for i := 0; i < 150; i++ {
+		u1.Id = ""
+		u1.Email = model.NewId()
+		u1.Username = model.NewId()
+		if err := (<-store.User().Save(&u1)).Err; err != nil {
+			t.Fatal("couldn't save item", err)
+		}
+	}
+
+	u1.Id = ""
+	u1.Email = model.NewId()
+	u1.Username = model.NewId()
+	if err := (<-store.User().Save(&u1)).Err; err == nil {
+		t.Fatal("should be the limit", err)
+	}
+}
+
+func TestUserStoreUpdate(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	if err := (<-store.User().Update(&u1, false)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	u1.Id = "missing"
+	if err := (<-store.User().Update(&u1, false)).Err; err == nil {
+		t.Fatal("Update should have failed because of missing key")
+	}
+
+	u1.Id = model.NewId()
+	if err := (<-store.User().Update(&u1, false)).Err; err == nil {
+		t.Fatal("Update should have faile because id change")
+	}
+}
+
+func TestUserStoreUpdateLastPingAt(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	if err := (<-store.User().UpdateLastPingAt(u1.Id, 1234567890)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	if r1 := <-store.User().Get(u1.Id); r1.Err != nil {
+		t.Fatal(r1.Err)
+	} else {
+		if r1.Data.(*model.User).LastPingAt != 1234567890 {
+			t.Fatal("LastPingAt not updated correctly")
+		}
+	}
+
+}
+
+func TestUserStoreUpdateLastActivityAt(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	if err := (<-store.User().UpdateLastActivityAt(u1.Id, 1234567890)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	if r1 := <-store.User().Get(u1.Id); r1.Err != nil {
+		t.Fatal(r1.Err)
+	} else {
+		if r1.Data.(*model.User).LastActivityAt != 1234567890 {
+			t.Fatal("LastActivityAt not updated correctly")
+		}
+	}
+
+}
+
+func TestUserStoreUpdateUserAndSessionActivity(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	s1 := model.Session{}
+	s1.UserId = u1.Id
+	s1.TeamId = u1.TeamId
+	<-store.Session().Save(&s1)
+
+	if err := (<-store.User().UpdateUserAndSessionActivity(u1.Id, s1.Id, 1234567890)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	if r1 := <-store.User().Get(u1.Id); r1.Err != nil {
+		t.Fatal(r1.Err)
+	} else {
+		if r1.Data.(*model.User).LastActivityAt != 1234567890 {
+			t.Fatal("LastActivityAt not updated correctly for user")
+		}
+	}
+
+	if r2 := <-store.Session().Get(s1.Id); r2.Err != nil {
+		t.Fatal(r2.Err)
+	} else {
+		if r2.Data.(*model.Session).LastActivityAt != 1234567890 {
+			t.Fatal("LastActivityAt not updated correctly for session")
+		}
+	}
+
+}
+
+func TestUserStoreGet(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	if r1 := <-store.User().Get(u1.Id); r1.Err != nil {
+		t.Fatal(r1.Err)
+	} else {
+		if r1.Data.(*model.User).ToJson() != u1.ToJson() {
+			t.Fatal("invalid returned user")
+		}
+	}
+
+	if err := (<-store.User().Get("")).Err; err == nil {
+		t.Fatal("Missing id should have failed")
+	}
+}
+
+func TestUserStoreGetProfiles(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	u2 := model.User{}
+	u2.TeamId = u1.TeamId
+	u2.Email = model.NewId()
+	<-store.User().Save(&u2)
+
+	if r1 := <-store.User().GetProfiles(u1.TeamId); r1.Err != nil {
+		t.Fatal(r1.Err)
+	} else {
+		users := r1.Data.(map[string]*model.User)
+		if len(users) != 2 {
+			t.Fatal("invalid returned users")
+		}
+
+		if users[u1.Id].Id != u1.Id {
+			t.Fatal("invalid returned user")
+		}
+	}
+
+	if r2 := <-store.User().GetProfiles("123"); r2.Err != nil {
+		t.Fatal(r2.Err)
+	} else {
+		if len(r2.Data.(map[string]*model.User)) != 0 {
+			t.Fatal("should have returned empty map")
+		}
+	}
+}
+
+func TestUserStoreGetByEmail(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	if err := (<-store.User().GetByEmail(u1.TeamId, u1.Email)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	if err := (<-store.User().GetByEmail("", "")).Err; err == nil {
+		t.Fatal("Should have failed because of missing email")
+	}
+}
+
+func TestUserStoreGetByUsername(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	u1.Username = model.NewId()
+	<-store.User().Save(&u1)
+
+	if err := (<-store.User().GetByUsername(u1.TeamId, u1.Username)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	if err := (<-store.User().GetByUsername("", "")).Err; err == nil {
+		t.Fatal("Should have failed because of missing username")
+	}
+}
+
+func TestUserStoreUpdatePassword(t *testing.T) {
+	Setup()
+
+	u1 := model.User{}
+	u1.TeamId = model.NewId()
+	u1.Email = model.NewId()
+	<-store.User().Save(&u1)
+
+	hashedPassword := model.HashPassword("newpwd")
+
+	if err := (<-store.User().UpdatePassword(u1.Id, hashedPassword)).Err; err != nil {
+		t.Fatal(err)
+	}
+
+	if r1 := <-store.User().GetByEmail(u1.TeamId, u1.Email); r1.Err != nil {
+		t.Fatal(r1.Err)
+	} else {
+		user := r1.Data.(*model.User)
+		if user.Password != hashedPassword {
+			t.Fatal("Password was not updated correctly")
+		}
+	}
+}