harness-drone/server/datastore/database/user_test.go

package database

import (
	"testing"

	"github.com/drone/drone/shared/model"
	"github.com/franela/goblin"
)

func TestUserstore(t *testing.T) {
	db := MustConnect("sqlite3", ":memory:")
	us := NewUserstore(db)
	defer db.Close()

	g := goblin.Goblin(t)
	g.Describe("Userstore", func() {

		// before each test be sure to purge the package
		// table data from the database.
		g.BeforeEach(func() {
			db.Exec("DELETE FROM users")
		})

		g.It("Should Put a User", func() {
			user := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			err := us.PostUser(&user)
			g.Assert(err == nil).IsTrue()
			g.Assert(user.ID != 0).IsTrue()
		})

		g.It("Should Post a User", func() {
			user := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			err := us.PostUser(&user)
			g.Assert(err == nil).IsTrue()
			g.Assert(user.ID != 0).IsTrue()
		})

		g.It("Should Get a User", func() {
			user := model.User{
				Login:    "joe",
				Remote:   "github.com",
				Access:   "f0b461ca586c27872b43a0685cbc2847",
				Secret:   "976f22a5eef7caacb7e678d6c52f49b1",
				Name:     "Joe Sixpack",
				Email:    "foo@bar.com",
				Gravatar: "b9015b0857e16ac4d94a0ffd9a0b79c8",
				Token:    "e42080dddf012c718e476da161d21ad5",
				Active:   true,
				Admin:    true,
				Created:  1398065343,
				Updated:  1398065344,
				Synced:   1398065345,
			}
			us.PostUser(&user)
			getuser, err := us.GetUser(user.ID)
			g.Assert(err == nil).IsTrue()
			g.Assert(user.ID).Equal(getuser.ID)
			g.Assert(user.Login).Equal(getuser.Login)
			g.Assert(user.Remote).Equal(getuser.Remote)
			g.Assert(user.Access).Equal(getuser.Access)
			g.Assert(user.Secret).Equal(getuser.Secret)
			g.Assert(user.Name).Equal(getuser.Name)
			g.Assert(user.Email).Equal(getuser.Email)
			g.Assert(user.Gravatar).Equal(getuser.Gravatar)
			g.Assert(user.Token).Equal(getuser.Token)
			g.Assert(user.Active).Equal(getuser.Active)
			g.Assert(user.Admin).Equal(getuser.Admin)
			g.Assert(user.Created).Equal(getuser.Created)
			g.Assert(user.Updated).Equal(getuser.Updated)
			g.Assert(user.Synced).Equal(getuser.Synced)
		})

		g.It("Should Get a User By Login", func() {
			user := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			us.PostUser(&user)
			getuser, err := us.GetUserLogin(user.Remote, user.Login)
			g.Assert(err == nil).IsTrue()
			g.Assert(user.ID).Equal(getuser.ID)
			g.Assert(user.Login).Equal(getuser.Login)
			g.Assert(user.Remote).Equal(getuser.Remote)
		})

		g.It("Should Get a User By Token", func() {
			user := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			us.PostUser(&user)
			getuser, err := us.GetUserToken(user.Token)
			g.Assert(err == nil).IsTrue()
			g.Assert(user.ID).Equal(getuser.ID)
			g.Assert(user.Token).Equal(getuser.Token)
		})

		g.It("Should Enforce Unique User Token", func() {
			user1 := model.User{
				Login:  "jane",
				Remote: "github.com",
				Name:   "Jane Doe",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			user2 := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			err1 := us.PostUser(&user1)
			err2 := us.PostUser(&user2)
			g.Assert(err1 == nil).IsTrue()
			g.Assert(err2 == nil).IsFalse()
		})

		g.It("Should Enforce Unique User Remote and Login", func() {
			user1 := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			user2 := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "ab20g0ddaf012c744e136da16aa21ad9",
			}
			err1 := us.PostUser(&user1)
			err2 := us.PostUser(&user2)
			g.Assert(err1 == nil).IsTrue()
			g.Assert(err2 == nil).IsFalse()
		})

		g.It("Should Get a User List", func() {
			user1 := model.User{
				Login:  "jane",
				Remote: "github.com",
				Name:   "Jane Doe",
				Email:  "foo@bar.com",
				Token:  "ab20g0ddaf012c744e136da16aa21ad9",
			}
			user2 := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			us.PostUser(&user1)
			us.PostUser(&user2)
			users, err := us.GetUserList()
			g.Assert(err == nil).IsTrue()
			g.Assert(len(users)).Equal(2)
			g.Assert(users[0].Login).Equal(user1.Login)
			g.Assert(users[0].Remote).Equal(user1.Remote)
			g.Assert(users[0].Name).Equal(user1.Name)
			g.Assert(users[0].Email).Equal(user1.Email)
			g.Assert(users[0].Token).Equal(user1.Token)
		})

		g.It("Should Del a User", func() {
			user := model.User{
				Login:  "joe",
				Remote: "github.com",
				Name:   "Joe Sixpack",
				Email:  "foo@bar.com",
				Token:  "e42080dddf012c718e476da161d21ad5",
			}
			us.PostUser(&user)
			_, err1 := us.GetUser(user.ID)
			err2 := us.DelUser(&user)
			_, err3 := us.GetUser(user.ID)
			g.Assert(err1 == nil).IsTrue()
			g.Assert(err2 == nil).IsTrue()
			g.Assert(err3 == nil).IsFalse()
		})

	})
}