Merge pull request #1373 from thaJeztah/18.09_backport_move_test_function_in_there

[18.09] backport getEngineConfigFilePath is only used during test so moving it in test files for now.
This commit is contained in:
Silvin Lubecki 2018-09-14 13:20:12 +02:00 committed by GitHub
commit 3b991ec615
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 30 additions and 30 deletions

View File

@ -4,7 +4,6 @@ import (
"context" "context"
"fmt" "fmt"
"io" "io"
"strings"
"syscall" "syscall"
"time" "time"
@ -92,35 +91,6 @@ func (c *baseClient) getEngineImage(engine containerd.Container) (string, error)
return image.Name(), nil return image.Name(), nil
} }
// getEngineConfigFilePath will extract the config file location from the engine flags
func (c baseClient) getEngineConfigFilePath(ctx context.Context, engine containerd.Container) (string, error) {
spec, err := engine.Spec(ctx)
configFile := ""
if err != nil {
return configFile, err
}
for i := 0; i < len(spec.Process.Args); i++ {
arg := spec.Process.Args[i]
if strings.HasPrefix(arg, "--config-file") {
if strings.Contains(arg, "=") {
split := strings.SplitN(arg, "=", 2)
configFile = split[1]
} else {
if i+1 >= len(spec.Process.Args) {
return configFile, ErrMalformedConfigFileParam
}
configFile = spec.Process.Args[i+1]
}
}
}
if configFile == "" {
// TODO - any more diagnostics to offer?
return configFile, ErrEngineConfigLookupFailure
}
return configFile, nil
}
var ( var (
engineWaitInterval = 500 * time.Millisecond engineWaitInterval = 500 * time.Millisecond
engineWaitTimeout = 60 * time.Second engineWaitTimeout = 60 * time.Second

View File

@ -4,6 +4,7 @@ import (
"bytes" "bytes"
"context" "context"
"fmt" "fmt"
"strings"
"syscall" "syscall"
"testing" "testing"
"time" "time"
@ -538,3 +539,32 @@ func TestGetEngineConfigFilePathMalformed1(t *testing.T) {
_, err := client.getEngineConfigFilePath(ctx, container) _, err := client.getEngineConfigFilePath(ctx, container)
assert.Assert(t, err == ErrMalformedConfigFileParam) assert.Assert(t, err == ErrMalformedConfigFileParam)
} }
// getEngineConfigFilePath will extract the config file location from the engine flags
func (c baseClient) getEngineConfigFilePath(ctx context.Context, engine containerd.Container) (string, error) {
spec, err := engine.Spec(ctx)
configFile := ""
if err != nil {
return configFile, err
}
for i := 0; i < len(spec.Process.Args); i++ {
arg := spec.Process.Args[i]
if strings.HasPrefix(arg, "--config-file") {
if strings.Contains(arg, "=") {
split := strings.SplitN(arg, "=", 2)
configFile = split[1]
} else {
if i+1 >= len(spec.Process.Args) {
return configFile, ErrMalformedConfigFileParam
}
configFile = spec.Process.Args[i+1]
}
}
}
if configFile == "" {
// TODO - any more diagnostics to offer?
return configFile, ErrEngineConfigLookupFailure
}
return configFile, nil
}