summaryrefslogtreecommitdiffstats
path: root/utils/testutils/mocked_http_service.go
blob: 8f9bc42be906dbd480f75ef9b4b89f02ed8ac55a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package testutils

import (
	"github.com/mattermost/mattermost-server/services/httpservice"
	"net/http"
	"net/http/httptest"
)

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()
}