104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
|
package server
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/base64"
|
||
|
|
"encoding/json"
|
||
|
|
"fmt"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"lbryio/lbry-id/auth"
|
||
|
|
"lbryio/lbry-id/store"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestServerGetClientSalt(t *testing.T) {
|
||
|
|
tt := []struct {
|
||
|
|
name string
|
||
|
|
emailGetParam string
|
||
|
|
emailCallExpected auth.Email
|
||
|
|
|
||
|
|
expectedStatusCode int
|
||
|
|
expectedErrorString string
|
||
|
|
|
||
|
|
storeErrors TestStoreFunctionsErrors
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "success",
|
||
|
|
emailGetParam: base64.StdEncoding.EncodeToString([]byte("good@example.com")),
|
||
|
|
emailCallExpected: "good@example.com",
|
||
|
|
expectedStatusCode: http.StatusOK,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "invalid email",
|
||
|
|
emailGetParam: base64.StdEncoding.EncodeToString([]byte("bad-example.com")),
|
||
|
|
|
||
|
|
expectedStatusCode: http.StatusBadRequest,
|
||
|
|
expectedErrorString: http.StatusText(http.StatusBadRequest) + ": Invalid email",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "account not found",
|
||
|
|
emailGetParam: base64.StdEncoding.EncodeToString([]byte("nonexistent@example.com")),
|
||
|
|
emailCallExpected: "nonexistent@example.com",
|
||
|
|
expectedStatusCode: http.StatusNotFound,
|
||
|
|
expectedErrorString: http.StatusText(http.StatusNotFound) + ": No match for email",
|
||
|
|
|
||
|
|
storeErrors: TestStoreFunctionsErrors{GetClientSaltSeed: store.ErrWrongCredentials},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "db error getting client salt",
|
||
|
|
emailGetParam: base64.StdEncoding.EncodeToString([]byte("good@example.com")),
|
||
|
|
emailCallExpected: "good@example.com",
|
||
|
|
expectedStatusCode: http.StatusInternalServerError,
|
||
|
|
expectedErrorString: http.StatusText(http.StatusInternalServerError),
|
||
|
|
|
||
|
|
storeErrors: TestStoreFunctionsErrors{GetClientSaltSeed: fmt.Errorf("Some random DB Error!")},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tc := range tt {
|
||
|
|
t.Run(tc.name, func(t *testing.T) {
|
||
|
|
|
||
|
|
testAuth := TestAuth{}
|
||
|
|
testStore := TestStore{
|
||
|
|
TestClientSaltSeed: auth.ClientSaltSeed("abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234abcd1234"),
|
||
|
|
|
||
|
|
Errors: tc.storeErrors,
|
||
|
|
}
|
||
|
|
|
||
|
|
s := Server{&testAuth, &testStore}
|
||
|
|
|
||
|
|
req := httptest.NewRequest(http.MethodGet, PathClientSaltSeed, nil)
|
||
|
|
q := req.URL.Query()
|
||
|
|
q.Add("email", string(tc.emailGetParam))
|
||
|
|
req.URL.RawQuery = q.Encode()
|
||
|
|
w := httptest.NewRecorder()
|
||
|
|
|
||
|
|
s.getClientSaltSeed(w, req)
|
||
|
|
|
||
|
|
body, _ := ioutil.ReadAll(w.Body)
|
||
|
|
|
||
|
|
expectStatusCode(t, w, tc.expectedStatusCode)
|
||
|
|
expectErrorString(t, body, tc.expectedErrorString)
|
||
|
|
|
||
|
|
// In this case, a salt seed is expected iff there is no error string
|
||
|
|
expectSaltSeed := len(tc.expectedErrorString) == 0
|
||
|
|
|
||
|
|
if !expectSaltSeed {
|
||
|
|
return // The rest of the test does not apply
|
||
|
|
}
|
||
|
|
|
||
|
|
var result ClientSaltSeedResponse
|
||
|
|
err := json.Unmarshal(body, &result)
|
||
|
|
|
||
|
|
if err != nil || result.ClientSaltSeed != testStore.TestClientSaltSeed {
|
||
|
|
t.Errorf("Expected client salt seed response to have the test client salt secret: result: %+v err: %+v", string(body), err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if testStore.Called.GetClientSaltSeed != tc.emailCallExpected {
|
||
|
|
t.Errorf("Expected Store.GetClientSaltSeed to be called with %s got %s", tc.emailCallExpected, testStore.Called.GetClientSaltSeed)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|