woodpecker/server/database/user_test.go

package database

import (
	"database/sql"
	"testing"

	"github.com/drone/drone/server/database/schema"
	"github.com/drone/drone/server/database/testdata"
	"github.com/drone/drone/server/database/testdatabase"
	"github.com/drone/drone/shared/model"
)

// in-memory database instance for unit testing
var db *sql.DB

// setup the test database and test fixtures
func setup() {
	db, _ = testdatabase.Open()
	schema.Load(db)
	testdata.Load(db)
}

// teardown the test database
func teardown() {
	db.Close()
}

func TestUserFind(t *testing.T) {
	setup()
	defer teardown()

	users := NewUserManager(db)
	user, err := users.Find(1)
	if err != nil {
		t.Errorf("Want User from ID, got %s", err)
	}

	testUser(t, user)
}

func TestUserFindLogin(t *testing.T) {
	setup()
	defer teardown()

	users := NewUserManager(db)
	user, err := users.FindLogin("github.com", "smellypooper")
	if err != nil {
		t.Errorf("Want User from Login, got %s", err)
	}

	testUser(t, user)
}

func TestUserFindToken(t *testing.T) {
	setup()
	defer teardown()

	users := NewUserManager(db)
	user, err := users.FindToken("e42080dddf012c718e476da161d21ad5")
	if err != nil {
		t.Errorf("Want User from Token, got %s", err)
	}

	testUser(t, user)
}

func TestUserList(t *testing.T) {
	setup()
	defer teardown()

	users := NewUserManager(db)
	all, err := users.List()
	if err != nil {
		t.Errorf("Want Users, got %s", err)
	}

	var got, want = len(all), 4
	if got != want {
		t.Errorf("Want %v Users, got %v", want, got)
	}

	testUser(t, all[0])
}

func TestUserInsert(t *testing.T) {
	setup()
	defer teardown()

	user := model.NewUser("github.com", "winkle", "winkle@caltech.edu")
	users := NewUserManager(db)
	if err := users.Insert(user); err != nil {
		t.Errorf("Want User created, got %s", err)
	}

	var got, want = user.ID, int64(5)
	if want != got {
		t.Errorf("Want User ID %v, got %v", want, got)
	}

	// verify unique remote + remote login constraint
	var err = users.Insert(&model.User{Remote: user.Remote, Login: user.Login, Token: "f71eb4a81a2cca56035dd7f6f2942e41"})
	if err == nil {
		t.Error("Want Token unique constraint violated")
	}

	// verify unique token constraint
	err = users.Insert(&model.User{Remote: "gitlab.com", Login: user.Login, Token: user.Token})
	if err == nil {
		t.Error("Want Token unique constraint violated")
	}
}

func TestUserUpdate(t *testing.T) {
	setup()
	defer teardown()

	users := NewUserManager(db)
	user, err := users.Find(4)
	if err != nil {
		t.Errorf("Want User from ID, got %s", err)
	}

	// update the user's access token
	user.Access = "fc47f37716fa04e9dfa9ac7eb22b5718"
	user.Secret = "d1c65427c978f2c9ad4baed72628dba0"
	if err := users.Update(user); err != nil {
		t.Errorf("Want User updated, got %s", err)
	}

	updated, _ := users.Find(4)
	var got, want = updated.Access, user.Access
	if got != want {
		t.Errorf("Want updated Access %s, got %s", want, got)
	}

	got, want = updated.Secret, user.Secret
	if got != want {
		t.Errorf("Want updated Secret %s, got %s", want, got)
	}
}

func TestUserDelete(t *testing.T) {
	setup()
	defer teardown()

	users := NewUserManager(db)
	user, err := users.Find(1)
	if err != nil {
		t.Errorf("Want User from ID, got %s", err)
	}

	// delete the user
	if err := users.Delete(user); err != nil {
		t.Errorf("Want User deleted, got %s", err)
	}

	// check to see if the deleted user is actually gone
	if _, err := users.Find(1); err != sql.ErrNoRows {
		t.Errorf("Want ErrNoRows, got %s", err)
	}
}

// testUser is a helper function that compares the user
// to an expected set of fixed field values.
func testUser(t *testing.T, user *model.User) {
	var got, want = user.Login, "smellypooper"
	if got != want {
		t.Errorf("Want Token %v, got %v", want, got)
	}

	got, want = user.Remote, "github.com"
	if got != want {
		t.Errorf("Want Token %v, got %v", want, got)
	}

	got, want = user.Access, "f0b461ca586c27872b43a0685cbc2847"
	if got != want {
		t.Errorf("Want Access Token %v, got %v", want, got)
	}

	got, want = user.Secret, "976f22a5eef7caacb7e678d6c52f49b1"
	if got != want {
		t.Errorf("Want Token Secret %v, got %v", want, got)
	}

	got, want = user.Name, "Dr. Cooper"
	if got != want {
		t.Errorf("Want Name %v, got %v", want, got)
	}

	got, want = user.Email, "drcooper@caltech.edu"
	if got != want {
		t.Errorf("Want Email %v, got %v", want, got)
	}

	got, want = user.Gravatar, "b9015b0857e16ac4d94a0ffd9a0b79c8"
	if got != want {
		t.Errorf("Want Gravatar %v, got %v", want, got)
	}

	got, want = user.Token, "e42080dddf012c718e476da161d21ad5"
	if got != want {
		t.Errorf("Want Token %v, got %v", want, got)
	}

	var gotBool, wantBool = user.Active, true
	if gotBool != wantBool {
		t.Errorf("Want Active %v, got %v", wantBool, gotBool)
	}

	gotBool, wantBool = user.Admin, true
	if gotBool != wantBool {
		t.Errorf("Want Admin %v, got %v", wantBool, gotBool)
	}

	var gotInt64, wantInt64 = user.ID, int64(1)
	if gotInt64 != wantInt64 {
		t.Errorf("Want ID %v, got %v", wantInt64, gotInt64)
	}

	gotInt64, wantInt64 = user.Created, int64(1398065343)
	if gotInt64 != wantInt64 {
		t.Errorf("Want Created %v, got %v", wantInt64, gotInt64)
	}

	gotInt64, wantInt64 = user.Updated, int64(1398065344)
	if gotInt64 != wantInt64 {
		t.Errorf("Want Updated %v, got %v", wantInt64, gotInt64)
	}

	gotInt64, wantInt64 = user.Synced, int64(1398065345)
	if gotInt64 != wantInt64 {
		t.Errorf("Want Synced %v, got %v", wantInt64, gotInt64)
	}
}