From f9c4a5d462314658c1477ddd68b8420006885e68 Mon Sep 17 00:00:00 2001 From: Pavel Date: Fri, 25 Oct 2019 13:18:43 +0300 Subject: [PATCH] More tests --- core/engine_test.go | 33 +++++++++++++++++++++++++++++++++ core/http_client_builder.go | 4 ++++ 2 files changed, 37 insertions(+) diff --git a/core/engine_test.go b/core/engine_test.go index 0d33b43..da87fdf 100644 --- a/core/engine_test.go +++ b/core/engine_test.go @@ -147,6 +147,39 @@ func (e *EngineTest) Test_ConfigureRouter() { }) } +func (e *EngineTest) Test_BuildHTTPClient() { + e.engine.Config = &Config{ + HTTPClientConfig: &HTTPClientConfig{ + Timeout: 30, + SSLVerification: true, + }, + } + e.engine.BuildHTTPClient() + + assert.NotNil(e.T(), e.engine.httpClient) +} + +func (e *EngineTest) Test_SetHTTPClient() { + var err error + + e.engine.httpClient = nil + e.engine.httpClient, err = NewHTTPClientBuilder().Build() + + assert.NoError(e.T(), err) + assert.NotNil(e.T(), e.engine.httpClient) +} + +func (e *EngineTest) Test_HTTPClient() { + var err error + + e.engine.httpClient = nil + assert.NotNil(e.T(), e.engine.HTTPClient()) + + e.engine.httpClient, err = NewHTTPClientBuilder().Build() + assert.NoError(e.T(), err) + assert.NotNil(e.T(), e.engine.httpClient) +} + func (e *EngineTest) Test_Run_Fail() { defer func() { assert.NotNil(e.T(), recover()) diff --git a/core/http_client_builder.go b/core/http_client_builder.go index 2b53b4c..9836ba1 100644 --- a/core/http_client_builder.go +++ b/core/http_client_builder.go @@ -149,6 +149,10 @@ func (b *HTTPClientBuilder) buildDialer() *HTTPClientBuilder { // parseAddress parses address and returns error in case of error (port is necessary) func (b *HTTPClientBuilder) parseAddress() error { + if b.mockAddress == "" { + return nil + } + if host, port, err := net.SplitHostPort(b.mockAddress); err == nil { b.mockHost = host b.mockPort = port