diff options
author | =Corey Hulen <corey@hulen.com> | 2015-06-14 23:53:32 -0800 |
---|---|---|
committer | =Corey Hulen <corey@hulen.com> | 2015-06-14 23:53:32 -0800 |
commit | 56e74239d6b34df8f30ef046f0b0ff4ff0866a71 (patch) | |
tree | 044da29848cf0f5c8607eac34de69065171669cf /store/sql_user_store_test.go | |
download | chat-56e74239d6b34df8f30ef046f0b0ff4ff0866a71.tar.gz chat-56e74239d6b34df8f30ef046f0b0ff4ff0866a71.tar.bz2 chat-56e74239d6b34df8f30ef046f0b0ff4ff0866a71.zip |
first commit
Diffstat (limited to 'store/sql_user_store_test.go')
-rw-r--r-- | store/sql_user_store_test.go | 276 |
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") + } + } +} |