From beed8748c0b7072db3cc1d8a2680a8b4856922a6 Mon Sep 17 00:00:00 2001 From: Lifubang Date: Tue, 16 Oct 2018 15:34:54 +0800 Subject: [PATCH] add test case TestNewAPIClientFromFlagsForDefaultSchema Signed-off-by: Lifubang --- cli/command/cli_test.go | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/cli/command/cli_test.go b/cli/command/cli_test.go index a4b06e6997..6ac120fa22 100644 --- a/cli/command/cli_test.go +++ b/cli/command/cli_test.go @@ -43,6 +43,26 @@ func TestNewAPIClientFromFlags(t *testing.T) { assert.Check(t, is.Equal(api.DefaultVersion, apiclient.ClientVersion())) } +func TestNewAPIClientFromFlagsForDefaultSchema(t *testing.T) { + host := ":2375" + opts := &flags.CommonOptions{Hosts: []string{host}} + configFile := &configfile.ConfigFile{ + HTTPHeaders: map[string]string{ + "My-Header": "Custom-Value", + }, + } + apiclient, err := NewAPIClientFromFlags(opts, configFile) + assert.NilError(t, err) + assert.Check(t, is.Equal("tcp://localhost"+host, apiclient.DaemonHost())) + + expectedHeaders := map[string]string{ + "My-Header": "Custom-Value", + "User-Agent": UserAgent(), + } + assert.Check(t, is.DeepEqual(expectedHeaders, apiclient.(*client.Client).CustomHTTPHeaders())) + assert.Check(t, is.Equal(api.DefaultVersion, apiclient.ClientVersion())) +} + func TestNewAPIClientFromFlagsWithAPIVersionFromEnv(t *testing.T) { customVersion := "v3.3.3" defer env.Patch(t, "DOCKER_API_VERSION", customVersion)()