wallet-sync-server/server/client_test.go

package server

import (
	"encoding/base64"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"

	"lbryio/lbry-id/auth"
	"lbryio/lbry-id/server/paths"
	"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, &TestEnv{}, &TestMail{}, TestPort}

			req := httptest.NewRequest(http.MethodGet, paths.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)
			}
		})
	}
}