Files
mattermost/utils/testutils/mocked_http_service.go
2018-11-06 16:28:55 +08:00

31 lines
673 B
Go

// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.
package testutils
import (
"net/http"
"net/http/httptest"
"github.com/mattermost/mattermost-server/services/httpservice"
)
type MockedHTTPService struct {
Server *httptest.Server
}
func MakeMockedHTTPService(handler http.Handler) *MockedHTTPService {
return &MockedHTTPService{
Server: httptest.NewServer(handler),
}
}
func (h *MockedHTTPService) MakeClient(trustURLs bool) *httpservice.Client {
return &httpservice.Client{Client: h.Server.Client()}
}
func (h *MockedHTTPService) Close() {
h.Server.CloseClientConnections()
h.Server.Close()
}