131 lines
4.2 KiB
Go
131 lines
4.2 KiB
Go
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"orblivion/lbry-id/auth"
|
|
"orblivion/lbry-id/store"
|
|
"testing"
|
|
)
|
|
|
|
func TestServerAuthHandlerSuccess(t *testing.T) {
|
|
testAuth := TestAuth{TestNewTokenString: auth.TokenString("seekrit")}
|
|
testStore := TestStore{}
|
|
s := Server{&testAuth, &testStore}
|
|
|
|
requestBody := []byte(`{"deviceId": "dev-1", "email": "abc@example.com", "password": "123"}`)
|
|
|
|
req := httptest.NewRequest(http.MethodPost, PathAuthToken, bytes.NewBuffer(requestBody))
|
|
w := httptest.NewRecorder()
|
|
|
|
s.getAuthToken(w, req)
|
|
body, _ := ioutil.ReadAll(w.Body)
|
|
|
|
expectStatusCode(t, w, http.StatusOK)
|
|
|
|
var result auth.AuthToken
|
|
err := json.Unmarshal(body, &result)
|
|
|
|
if err != nil || result.Token != testAuth.TestNewTokenString {
|
|
t.Errorf("Expected auth response to contain token: result: %+v err: %+v", string(body), err)
|
|
}
|
|
|
|
if testStore.Called.SaveToken != testAuth.TestNewTokenString {
|
|
t.Errorf("Expected Store.SaveToken to be called with %s", testAuth.TestNewTokenString)
|
|
}
|
|
}
|
|
|
|
func TestServerAuthHandlerErrors(t *testing.T) {
|
|
tt := []struct {
|
|
name string
|
|
expectedStatusCode int
|
|
expectedErrorString string
|
|
|
|
storeErrors TestStoreFunctionsErrors
|
|
authFailGenToken bool
|
|
}{
|
|
{
|
|
name: "login fail",
|
|
expectedStatusCode: http.StatusUnauthorized,
|
|
expectedErrorString: http.StatusText(http.StatusUnauthorized) + ": No match for email and password",
|
|
|
|
storeErrors: TestStoreFunctionsErrors{GetUserId: store.ErrNoUId},
|
|
},
|
|
{
|
|
name: "generate token fail",
|
|
expectedStatusCode: http.StatusInternalServerError,
|
|
expectedErrorString: http.StatusText(http.StatusInternalServerError),
|
|
|
|
authFailGenToken: true,
|
|
},
|
|
{
|
|
name: "save token fail",
|
|
expectedStatusCode: http.StatusInternalServerError,
|
|
expectedErrorString: http.StatusText(http.StatusInternalServerError),
|
|
|
|
storeErrors: TestStoreFunctionsErrors{SaveToken: fmt.Errorf("TestStore.SaveToken fail")},
|
|
},
|
|
}
|
|
for _, tc := range tt {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
// Set this up to fail according to specification
|
|
testAuth := TestAuth{TestNewTokenString: auth.TokenString("seekrit")}
|
|
testStore := TestStore{Errors: tc.storeErrors}
|
|
if tc.authFailGenToken { // TODO - TestAuth{Errors:authErrors}
|
|
testAuth.FailGenToken = true
|
|
}
|
|
server := Server{&testAuth, &testStore}
|
|
|
|
// Make request
|
|
// So long as the JSON is well-formed, the content doesn't matter here since the password check will be stubbed out
|
|
requestBody := `{"deviceId": "dev-1", "email": "abc@example.com", "password": "123"}`
|
|
req := httptest.NewRequest(http.MethodPost, PathAuthToken, bytes.NewBuffer([]byte(requestBody)))
|
|
w := httptest.NewRecorder()
|
|
|
|
server.getAuthToken(w, req)
|
|
|
|
expectStatusCode(t, w, tc.expectedStatusCode)
|
|
expectErrorString(t, w, tc.expectedErrorString)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestServerValidateAuthRequest(t *testing.T) {
|
|
authRequest := AuthRequest{DeviceId: "dId", Email: "joe@example.com", Password: "aoeu"}
|
|
if !authRequest.validate() {
|
|
t.Fatalf("Expected valid AuthRequest to successfully validate")
|
|
}
|
|
|
|
authRequest = AuthRequest{Email: "joe@example.com", Password: "aoeu"}
|
|
if authRequest.validate() {
|
|
t.Fatalf("Expected AuthRequest with missing device to not successfully validate")
|
|
}
|
|
|
|
authRequest = AuthRequest{DeviceId: "dId", Email: "joe-example.com", Password: "aoeu"}
|
|
if authRequest.validate() {
|
|
t.Fatalf("Expected AuthRequest with invalid email to not successfully validate")
|
|
}
|
|
|
|
// Note that Golang's email address parser, which I use, will accept
|
|
// "Joe <joe@example.com>" so we need to make sure to avoid accepting it. See
|
|
// the implementation.
|
|
authRequest = AuthRequest{DeviceId: "dId", Email: "Joe <joe@example.com>", Password: "aoeu"}
|
|
if authRequest.validate() {
|
|
t.Fatalf("Expected AuthRequest with email with unexpected formatting to not successfully validate")
|
|
}
|
|
|
|
authRequest = AuthRequest{DeviceId: "dId", Password: "aoeu"}
|
|
if authRequest.validate() {
|
|
t.Fatalf("Expected AuthRequest with missing email to not successfully validate")
|
|
}
|
|
|
|
authRequest = AuthRequest{DeviceId: "dId", Email: "joe@example.com"}
|
|
if authRequest.validate() {
|
|
t.Fatalf("Expected AuthRequest with missing password to not successfully validate")
|
|
}
|
|
}
|