vultr: minor fixups

Signed-off-by: Julien Pivotto <roidelapluie@o11y.eu>
This commit is contained in:
Julien Pivotto 2022-05-24 18:12:13 +02:00 committed by Julien Pivotto
parent 3ef153b00c
commit c9d4e5fc94
2 changed files with 9 additions and 9 deletions

View File

@ -27,25 +27,25 @@ type SDMock struct {
Mux *http.ServeMux Mux *http.ServeMux
} }
// NewSDMock returns a new SDMock. // NewSDMock returns a new Vultr mock server.
func NewSDMock(t *testing.T) *SDMock { func NewSDMock(t *testing.T) *SDMock {
return &SDMock{ return &SDMock{
t: t, t: t,
} }
} }
// Endpoint returns the URI to the mock server // Endpoint returns the URI to the mock server.
func (m *SDMock) Endpoint() string { func (m *SDMock) Endpoint() string {
return m.Server.URL + "/" return m.Server.URL + "/"
} }
// Setup creates the mock server // Setup creates the mock server.
func (m *SDMock) Setup() { func (m *SDMock) Setup() {
m.Mux = http.NewServeMux() m.Mux = http.NewServeMux()
m.Server = httptest.NewServer(m.Mux) m.Server = httptest.NewServer(m.Mux)
} }
// ShutdownServer creates the mock server // ShutdownServer shuts down the mock server.
func (m *SDMock) ShutdownServer() { func (m *SDMock) ShutdownServer() {
m.Server.Close() m.Server.Close()
} }