summaryrefslogtreecommitdiffstats
path: root/utils/testutils/mocked_http_service.go
blob: b1e7f6963f0bedaf445bcdb82ccd1900a294ff16 (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
// Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package testutils

import (
	"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) *http.Client {
	return h.Server.Client()
}

func (h *MockedHTTPService) Close() {
	h.Server.CloseClientConnections()
	h.Server.Close()
}