persona_test.go

package persona

import (
	"errors"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
)

func TestSignupWithUsername(t *testing.T) {

	w := httptest.NewRecorder()

	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	db.DropTable(&User{})
	db.DropTable(&Sessionusername{})

	db.AutoMigrate(&User{})

	Config(db, "username")

	user := User{Username: "Erwan", Password: HashPassword("0000"), Mail: "e@e.e"}
	err = Signup(&user, user.Username, w)
	if err != nil {
		t.Error(err)
	}

	resp := w.Result()
	if resp.Cookies()[0].Value == "" {
		t.Error(errors.New("no session cookie"))
	}
}

func TestLogoutWithUsername(t *testing.T) {
	w := httptest.NewRecorder()

	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	Config(db, "username")

	user := User{Username: "Erwan", Password: "0000", Mail: "e@e.e"}

	err = Logout(user.Username, w)
	if err != nil {
		t.Error(err)
	}

	resp := w.Result()
	if resp.Cookies()[0].Value != "" {
		t.Error(errors.New("cookie was not deleted"))
	}
}

func TestSignupWithEmail(t *testing.T) {

	w := httptest.NewRecorder()

	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	db.DropTable(&User{})
	db.DropTable(&Sessionusername{})

	db.AutoMigrate(&User{})

	Config(db, "email")

	user := User{Username: "Erwan", Password: HashPassword("0000"), Mail: "e@e.e"}
	err = Signup(&user, user.Mail, w)
	if err != nil {
		t.Error(err)
	}

	resp := w.Result()
	if resp.Cookies()[0].Value == "" {
		t.Error(errors.New("no session cookie"))
	}
}

func TestLogoutWithEmail(t *testing.T) {
	w := httptest.NewRecorder()

	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	Config(db, "email")

	user := User{Username: "Erwan", Password: "0000", Mail: "e@e.e"}

	err = Logout(user.Mail, w)
	if err != nil {
		t.Error(err)
	}

	resp := w.Result()
	if resp.Cookies()[0].Value != "" {
		t.Error(errors.New("cookie was not deleted"))
	}
}

func TestLoginWithUsername(t *testing.T) {

	w := httptest.NewRecorder()

	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	Config(db, "username")

	user := User{Username: "Erwan", Password: "0000", Mail: "e@e.e"}

	err = Login(user.Username, user.Password, w)
	if err != nil {
		t.Error(err)
	}

	resp := w.Result()
	if resp.Cookies()[0].Value == "" {
		t.Error(errors.New("no session cookie"))
	}
}

func TestLoginWithEmail(t *testing.T) {

	w := httptest.NewRecorder()

	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	Config(db, "email")

	user := User{Username: "Erwan", Password: "0000", Mail: "e@e.e"}

	err = Login(user.Mail, user.Password, w)
	if err != nil {
		t.Error(err)
	}

	resp := w.Result()
	if resp.Cookies()[0].Value == "" {
		t.Error(errors.New("no session cookie"))
	}
}

func TestCurrentUser(t *testing.T) {
	w := httptest.NewRecorder()

	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	Config(db, "username")

	user := User{Username: "Erwan", Password: "0000", Mail: "e@e.e"}

	var session Sessionusername
	db.Table("sessionusernames").Where("username = ?", user.Username).First(&session)

	http.SetCookie(w, &http.Cookie{Name: "session-persona", Value: session.Token})
	r := &http.Request{Header: http.Header{"Cookie": w.HeaderMap["Set-Cookie"]}}

	username, err := GetCurrentUser(r)
	if err != nil {
		t.Error(err)
	}

	if username == "" {
		t.Error("no user loggedin")
	}
}

func TestRecoverPassword(t *testing.T) {
	db, err := gorm.Open("sqlite3", "test.db")
	if err != nil {
		t.Error(err)
	}
	defer db.Close()

	Config(db, "username")

	user := User{Username: "Erwan", Password: "0000", Mail: "e@e.e"}

	err = RecoverPassword(user.Username, user.Password, "0001")
	if err != nil {
		t.Error(err)
	}

	var updatedUser User
	db.Table("users").Where("username = ?", user.Username).First(&updatedUser)

	if !checkPasswordHash("0001", updatedUser.Password) {
		t.Error(errors.New("password wasn't changed"))
	}
}