thermokarst/jwt
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Placeholder for generating token

Browse source
This commit is contained in:
Matthew Dillon 2015-04-17 16:08:36 -08:00
parent 7871c1db59
commit e0d56f0a29
2 changed files with 24 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
jwt.go
View file

@ -8,8 +8,11 @@ import (
type Config struct {
Secret string
Auth AuthFunc
}
type AuthFunc func(string, string) (bool, error)
type JWTMiddleware struct {
config Config
}
@ -28,11 +31,19 @@ func (m *JWTMiddleware) Secure(h http.Handler) http.Handler {
})
}
func (m *JWTMiddleware) Auth(w http.ResponseWriter, r *http.Request) {
var b interface{}
func (m *JWTMiddleware) GenerateToken(w http.ResponseWriter, r *http.Request) {
var b map[string]string
err := json.NewDecoder(r.Body).Decode(&b)
if err != nil {
panic(err)
}
w.Write([]byte("test"))
result, err := m.config.Auth(b["email"], b["password"])
if err != nil {
panic(err)
}
resp := "failure"
if result {
resp = "success"
}
w.Write([]byte(resp))
}

14
jwt_test.go
View file

@ -13,9 +13,14 @@ var testHandler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request)
w.Write([]byte("test"))
})
var authFunc = func(email, password string) (bool, error) {
return true, nil
}
func newJWTMiddlewareOrFatal(t *testing.T) *JWTMiddleware {
config := &Config{
Secret: "password",
Auth: authFunc,
}
middleware, err := NewMiddleware(config)
if err != nil {
@ -29,6 +34,7 @@ func TestNewJWTMiddleware(t *testing.T) {
if middleware.config.Secret != "password" {
t.Errorf("expected 'password', got %v", middleware.config.Secret)
}
// TODO: test auth func init
}
func TestNewJWTMiddlewareNoConfig(t *testing.T) {
@ -48,7 +54,7 @@ func TestSecureHandler(t *testing.T) {
}
}
func TestAuthHandler(t *testing.T) {
func TestGenerateTokenHandler(t *testing.T) {
middleware := newJWTMiddlewareOrFatal(t)
authBody := map[string]interface{}{
"email": "user@example.com",
@ -58,7 +64,7 @@ func TestAuthHandler(t *testing.T) {
if err != nil {
t.Error(err)
}
ts := httptest.NewServer(http.HandlerFunc(middleware.Auth))
ts := httptest.NewServer(http.HandlerFunc(middleware.GenerateToken))
defer ts.Close()
resp, err := http.Post(ts.URL, "application/json", bytes.NewReader(body))
respBody, err := ioutil.ReadAll(resp.Body)
@ -66,7 +72,7 @@ func TestAuthHandler(t *testing.T) {
if err != nil {
t.Error(err)
}
if string(respBody) != "test" {
t.Errorf("expected 'test', got %v", respBody)
if string(respBody) != "success" {
t.Errorf("expected 'success', got %v", string(respBody))
}
}