95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package http
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"path"
|
|
"testing"
|
|
|
|
platform "github.com/influxdata/influxdb/v2"
|
|
platform2 "github.com/influxdata/influxdb/v2/kit/platform"
|
|
kithttp "github.com/influxdata/influxdb/v2/kit/transport/http"
|
|
"github.com/influxdata/influxdb/v2/mock"
|
|
"github.com/influxdata/influxdb/v2/pkg/testttp"
|
|
"github.com/influxdata/influxdb/v2/tenant"
|
|
platformtesting "github.com/influxdata/influxdb/v2/testing"
|
|
"go.uber.org/zap/zaptest"
|
|
)
|
|
|
|
// NewMockUserBackend returns a UserBackend with mock services.
|
|
func NewMockUserBackend(t *testing.T) *UserBackend {
|
|
return &UserBackend{
|
|
log: zaptest.NewLogger(t),
|
|
UserService: mock.NewUserService(),
|
|
UserOperationLogService: mock.NewUserOperationLogService(),
|
|
PasswordsService: mock.NewPasswordsService(),
|
|
HTTPErrorHandler: kithttp.NewErrorHandler(zaptest.NewLogger(t)),
|
|
}
|
|
}
|
|
|
|
func initUserService(f platformtesting.UserFields, t *testing.T) (platform.UserService, string, func()) {
|
|
t.Helper()
|
|
|
|
store := platformtesting.NewTestInmemStore(t)
|
|
tenantStore := tenant.NewStore(store)
|
|
tenantStore.IDGen = f.IDGenerator
|
|
tenantService := tenant.NewService(tenantStore)
|
|
|
|
ctx := context.Background()
|
|
for _, u := range f.Users {
|
|
if err := tenantService.CreateUser(ctx, u); err != nil {
|
|
t.Fatalf("failed to populate users")
|
|
}
|
|
}
|
|
|
|
userBackend := NewMockUserBackend(t)
|
|
userBackend.HTTPErrorHandler = kithttp.NewErrorHandler(zaptest.NewLogger(t))
|
|
userBackend.UserService = tenantService
|
|
handler := NewUserHandler(zaptest.NewLogger(t), userBackend)
|
|
server := httptest.NewServer(handler)
|
|
|
|
httpClient, err := NewHTTPClient(server.URL, "", false)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
client := UserService{
|
|
Client: httpClient,
|
|
}
|
|
|
|
return &client, "", server.Close
|
|
}
|
|
|
|
func TestUserService(t *testing.T) {
|
|
t.Parallel()
|
|
platformtesting.UserService(initUserService, t)
|
|
}
|
|
|
|
func TestUserHandler_SettingPassword(t *testing.T) {
|
|
be := NewMockUserBackend(t)
|
|
fakePassSVC := mock.NewPasswordsService()
|
|
|
|
userID := platform2.ID(1)
|
|
fakePassSVC.SetPasswordFn = func(_ context.Context, id platform2.ID, newPass string) error {
|
|
if id != userID {
|
|
return errors.New("unexpected id: " + id.String())
|
|
}
|
|
if newPass == "" {
|
|
return errors.New("no password provided")
|
|
}
|
|
return nil
|
|
}
|
|
be.PasswordsService = fakePassSVC
|
|
|
|
h := NewUserHandler(zaptest.NewLogger(t), be)
|
|
|
|
addr := path.Join("/api/v2/users", userID.String(), "/password")
|
|
|
|
testttp.
|
|
PostJSON(t, addr, passwordSetRequest{Password: "newpassword"}).
|
|
Do(h).
|
|
ExpectStatus(http.StatusNoContent)
|
|
}
|