Fix container name filters in docker input (#3331)
(cherry picked from commit fa25e123d8
)
This commit is contained in:
parent
0cc5fc0ce4
commit
17377b4942
|
@ -77,3 +77,40 @@ func compileFilterNoGlob(filters []string) Filter {
|
||||||
}
|
}
|
||||||
return &out
|
return &out
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type IncludeExcludeFilter struct {
|
||||||
|
include Filter
|
||||||
|
exclude Filter
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewIncludeExcludeFilter(
|
||||||
|
include []string,
|
||||||
|
exclude []string,
|
||||||
|
) (Filter, error) {
|
||||||
|
in, err := Compile(include)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
ex, err := Compile(exclude)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return &IncludeExcludeFilter{in, ex}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *IncludeExcludeFilter) Match(s string) bool {
|
||||||
|
if f.include != nil {
|
||||||
|
if !f.include.Match(s) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if f.exclude != nil {
|
||||||
|
if f.exclude.Match(s) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
|
@ -20,16 +20,6 @@ import (
|
||||||
"github.com/influxdata/telegraf/plugins/inputs"
|
"github.com/influxdata/telegraf/plugins/inputs"
|
||||||
)
|
)
|
||||||
|
|
||||||
type DockerLabelFilter struct {
|
|
||||||
labelInclude filter.Filter
|
|
||||||
labelExclude filter.Filter
|
|
||||||
}
|
|
||||||
|
|
||||||
type DockerContainerFilter struct {
|
|
||||||
containerInclude filter.Filter
|
|
||||||
containerExclude filter.Filter
|
|
||||||
}
|
|
||||||
|
|
||||||
// Docker object
|
// Docker object
|
||||||
type Docker struct {
|
type Docker struct {
|
||||||
Endpoint string
|
Endpoint string
|
||||||
|
@ -41,11 +31,9 @@ type Docker struct {
|
||||||
TagEnvironment []string `toml:"tag_env"`
|
TagEnvironment []string `toml:"tag_env"`
|
||||||
LabelInclude []string `toml:"docker_label_include"`
|
LabelInclude []string `toml:"docker_label_include"`
|
||||||
LabelExclude []string `toml:"docker_label_exclude"`
|
LabelExclude []string `toml:"docker_label_exclude"`
|
||||||
LabelFilter DockerLabelFilter
|
|
||||||
|
|
||||||
ContainerInclude []string `toml:"container_name_include"`
|
ContainerInclude []string `toml:"container_name_include"`
|
||||||
ContainerExclude []string `toml:"container_name_exclude"`
|
ContainerExclude []string `toml:"container_name_exclude"`
|
||||||
ContainerFilter DockerContainerFilter
|
|
||||||
|
|
||||||
SSLCA string `toml:"ssl_ca"`
|
SSLCA string `toml:"ssl_ca"`
|
||||||
SSLCert string `toml:"ssl_cert"`
|
SSLCert string `toml:"ssl_cert"`
|
||||||
|
@ -55,10 +43,12 @@ type Docker struct {
|
||||||
newEnvClient func() (Client, error)
|
newEnvClient func() (Client, error)
|
||||||
newClient func(string, *tls.Config) (Client, error)
|
newClient func(string, *tls.Config) (Client, error)
|
||||||
|
|
||||||
client Client
|
client Client
|
||||||
httpClient *http.Client
|
httpClient *http.Client
|
||||||
engine_host string
|
engine_host string
|
||||||
filtersCreated bool
|
filtersCreated bool
|
||||||
|
labelFilter filter.Filter
|
||||||
|
containerFilter filter.Filter
|
||||||
}
|
}
|
||||||
|
|
||||||
// KB, MB, GB, TB, PB...human friendly
|
// KB, MB, GB, TB, PB...human friendly
|
||||||
|
@ -291,12 +281,8 @@ func (d *Docker) gatherContainer(
|
||||||
"container_version": imageVersion,
|
"container_version": imageVersion,
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(d.ContainerInclude) > 0 || len(d.ContainerExclude) > 0 {
|
if !d.containerFilter.Match(cname) {
|
||||||
if len(d.ContainerInclude) == 0 || !d.ContainerFilter.containerInclude.Match(cname) {
|
return nil
|
||||||
if len(d.ContainerExclude) == 0 || d.ContainerFilter.containerExclude.Match(cname) {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ctx, cancel := context.WithTimeout(context.Background(), d.Timeout.Duration)
|
ctx, cancel := context.WithTimeout(context.Background(), d.Timeout.Duration)
|
||||||
|
@ -317,10 +303,8 @@ func (d *Docker) gatherContainer(
|
||||||
|
|
||||||
// Add labels to tags
|
// Add labels to tags
|
||||||
for k, label := range container.Labels {
|
for k, label := range container.Labels {
|
||||||
if len(d.LabelInclude) == 0 || d.LabelFilter.labelInclude.Match(k) {
|
if d.labelFilter.Match(k) {
|
||||||
if len(d.LabelExclude) == 0 || !d.LabelFilter.labelExclude.Match(k) {
|
tags[k] = label
|
||||||
tags[k] = label
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -666,46 +650,25 @@ func parseSize(sizeStr string) (int64, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Docker) createContainerFilters() error {
|
func (d *Docker) createContainerFilters() error {
|
||||||
|
// Backwards compatibility for deprecated `container_names` parameter.
|
||||||
if len(d.ContainerNames) > 0 {
|
if len(d.ContainerNames) > 0 {
|
||||||
d.ContainerInclude = append(d.ContainerInclude, d.ContainerNames...)
|
d.ContainerInclude = append(d.ContainerInclude, d.ContainerNames...)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(d.ContainerInclude) != 0 {
|
filter, err := filter.NewIncludeExcludeFilter(d.ContainerInclude, d.ContainerExclude)
|
||||||
var err error
|
if err != nil {
|
||||||
d.ContainerFilter.containerInclude, err = filter.Compile(d.ContainerInclude)
|
return err
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
d.containerFilter = filter
|
||||||
if len(d.ContainerExclude) != 0 {
|
|
||||||
var err error
|
|
||||||
d.ContainerFilter.containerExclude, err = filter.Compile(d.ContainerExclude)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d *Docker) createLabelFilters() error {
|
func (d *Docker) createLabelFilters() error {
|
||||||
if len(d.LabelInclude) != 0 {
|
filter, err := filter.NewIncludeExcludeFilter(d.LabelInclude, d.LabelExclude)
|
||||||
var err error
|
if err != nil {
|
||||||
d.LabelFilter.labelInclude, err = filter.Compile(d.LabelInclude)
|
return err
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
d.labelFilter = filter
|
||||||
if len(d.LabelExclude) != 0 {
|
|
||||||
var err error
|
|
||||||
d.LabelFilter.labelExclude, err = filter.Compile(d.LabelExclude)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -44,21 +44,23 @@ func (c *MockClient) ContainerInspect(
|
||||||
return c.ContainerInspectF(ctx, containerID)
|
return c.ContainerInspectF(ctx, containerID)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var baseClient = MockClient{
|
||||||
|
InfoF: func(context.Context) (types.Info, error) {
|
||||||
|
return info, nil
|
||||||
|
},
|
||||||
|
ContainerListF: func(context.Context, types.ContainerListOptions) ([]types.Container, error) {
|
||||||
|
return containerList, nil
|
||||||
|
},
|
||||||
|
ContainerStatsF: func(context.Context, string, bool) (types.ContainerStats, error) {
|
||||||
|
return containerStats(), nil
|
||||||
|
},
|
||||||
|
ContainerInspectF: func(context.Context, string) (types.ContainerJSON, error) {
|
||||||
|
return containerInspect, nil
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
func newClient(host string, tlsConfig *tls.Config) (Client, error) {
|
func newClient(host string, tlsConfig *tls.Config) (Client, error) {
|
||||||
return &MockClient{
|
return &baseClient, nil
|
||||||
InfoF: func(context.Context) (types.Info, error) {
|
|
||||||
return info, nil
|
|
||||||
},
|
|
||||||
ContainerListF: func(context.Context, types.ContainerListOptions) ([]types.Container, error) {
|
|
||||||
return containerList, nil
|
|
||||||
},
|
|
||||||
ContainerStatsF: func(context.Context, string, bool) (types.ContainerStats, error) {
|
|
||||||
return containerStats(), nil
|
|
||||||
},
|
|
||||||
ContainerInspectF: func(context.Context, string) (types.ContainerJSON, error) {
|
|
||||||
return containerInspect, nil
|
|
||||||
},
|
|
||||||
}, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDockerGatherContainerStats(t *testing.T) {
|
func TestDockerGatherContainerStats(t *testing.T) {
|
||||||
|
@ -234,82 +236,291 @@ func TestDocker_WindowsMemoryContainerStats(t *testing.T) {
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestDockerGatherLabels(t *testing.T) {
|
func TestContainerLabels(t *testing.T) {
|
||||||
var gatherLabelsTests = []struct {
|
var tests = []struct {
|
||||||
include []string
|
name string
|
||||||
exclude []string
|
container types.Container
|
||||||
expected []string
|
include []string
|
||||||
notexpected []string
|
exclude []string
|
||||||
|
expected map[string]string
|
||||||
}{
|
}{
|
||||||
{[]string{}, []string{}, []string{"label1", "label2"}, []string{}},
|
{
|
||||||
{[]string{"*"}, []string{}, []string{"label1", "label2"}, []string{}},
|
name: "Nil filters matches all",
|
||||||
{[]string{"lab*"}, []string{}, []string{"label1", "label2"}, []string{}},
|
container: types.Container{
|
||||||
{[]string{"label1"}, []string{}, []string{"label1"}, []string{"label2"}},
|
Labels: map[string]string{
|
||||||
{[]string{"label1*"}, []string{}, []string{"label1"}, []string{"label2"}},
|
"a": "x",
|
||||||
{[]string{}, []string{"*"}, []string{}, []string{"label1", "label2"}},
|
},
|
||||||
{[]string{}, []string{"lab*"}, []string{}, []string{"label1", "label2"}},
|
},
|
||||||
{[]string{}, []string{"label1"}, []string{"label2"}, []string{"label1"}},
|
include: nil,
|
||||||
{[]string{"*"}, []string{"*"}, []string{}, []string{"label1", "label2"}},
|
exclude: nil,
|
||||||
|
expected: map[string]string{
|
||||||
|
"a": "x",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Empty filters matches all",
|
||||||
|
container: types.Container{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"a": "x",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
include: []string{},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: map[string]string{
|
||||||
|
"a": "x",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Must match include",
|
||||||
|
container: types.Container{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"a": "x",
|
||||||
|
"b": "y",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
include: []string{"a"},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: map[string]string{
|
||||||
|
"a": "x",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Must not match exclude",
|
||||||
|
container: types.Container{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"a": "x",
|
||||||
|
"b": "y",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
include: []string{},
|
||||||
|
exclude: []string{"b"},
|
||||||
|
expected: map[string]string{
|
||||||
|
"a": "x",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Include Glob",
|
||||||
|
container: types.Container{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"aa": "x",
|
||||||
|
"ab": "y",
|
||||||
|
"bb": "z",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
include: []string{"a*"},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: map[string]string{
|
||||||
|
"aa": "x",
|
||||||
|
"ab": "y",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Exclude Glob",
|
||||||
|
container: types.Container{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"aa": "x",
|
||||||
|
"ab": "y",
|
||||||
|
"bb": "z",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
include: []string{},
|
||||||
|
exclude: []string{"a*"},
|
||||||
|
expected: map[string]string{
|
||||||
|
"bb": "z",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Excluded Includes",
|
||||||
|
container: types.Container{
|
||||||
|
Labels: map[string]string{
|
||||||
|
"aa": "x",
|
||||||
|
"ab": "y",
|
||||||
|
"bb": "z",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
include: []string{"a*"},
|
||||||
|
exclude: []string{"*b"},
|
||||||
|
expected: map[string]string{
|
||||||
|
"aa": "x",
|
||||||
|
},
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
for _, tt := range gatherLabelsTests {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
t.Run("", func(t *testing.T) {
|
|
||||||
var acc testutil.Accumulator
|
var acc testutil.Accumulator
|
||||||
d := Docker{
|
|
||||||
newClient: newClient,
|
newClientFunc := func(host string, tlsConfig *tls.Config) (Client, error) {
|
||||||
|
client := baseClient
|
||||||
|
client.ContainerListF = func(context.Context, types.ContainerListOptions) ([]types.Container, error) {
|
||||||
|
return []types.Container{tt.container}, nil
|
||||||
|
}
|
||||||
|
return &client, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, label := range tt.include {
|
d := Docker{
|
||||||
d.LabelInclude = append(d.LabelInclude, label)
|
newClient: newClientFunc,
|
||||||
}
|
LabelInclude: tt.include,
|
||||||
for _, label := range tt.exclude {
|
LabelExclude: tt.exclude,
|
||||||
d.LabelExclude = append(d.LabelExclude, label)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
err := d.Gather(&acc)
|
err := d.Gather(&acc)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
for _, label := range tt.expected {
|
// Grab tags from a container metric
|
||||||
if !acc.HasTag("docker_container_cpu", label) {
|
var actual map[string]string
|
||||||
t.Errorf("Didn't get expected label of %s. Test was: Include: %s Exclude %s",
|
for _, metric := range acc.Metrics {
|
||||||
label, tt.include, tt.exclude)
|
if metric.Measurement == "docker_container_cpu" {
|
||||||
|
actual = metric.Tags
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, label := range tt.notexpected {
|
for k, v := range tt.expected {
|
||||||
if acc.HasTag("docker_container_cpu", label) {
|
require.Equal(t, v, actual[k])
|
||||||
t.Errorf("Got unexpected label of %s. Test was: Include: %s Exclude %s",
|
|
||||||
label, tt.include, tt.exclude)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestContainerNames(t *testing.T) {
|
func TestContainerNames(t *testing.T) {
|
||||||
var gatherContainerNames = []struct {
|
var tests = []struct {
|
||||||
include []string
|
name string
|
||||||
exclude []string
|
containers [][]string
|
||||||
expected []string
|
include []string
|
||||||
notexpected []string
|
exclude []string
|
||||||
|
expected []string
|
||||||
}{
|
}{
|
||||||
{[]string{}, []string{}, []string{"etcd", "etcd2"}, []string{}},
|
{
|
||||||
{[]string{"*"}, []string{}, []string{"etcd", "etcd2"}, []string{}},
|
name: "Nil filters matches all",
|
||||||
{[]string{"etc*"}, []string{}, []string{"etcd", "etcd2"}, []string{}},
|
containers: [][]string{
|
||||||
{[]string{"etcd"}, []string{}, []string{"etcd"}, []string{"etcd2"}},
|
{"/etcd"},
|
||||||
{[]string{"etcd2*"}, []string{}, []string{"etcd2"}, []string{"etcd"}},
|
{"/etcd2"},
|
||||||
{[]string{}, []string{"etc*"}, []string{}, []string{"etcd", "etcd2"}},
|
},
|
||||||
{[]string{}, []string{"etcd"}, []string{"etcd2"}, []string{"etcd"}},
|
include: nil,
|
||||||
{[]string{"*"}, []string{"*"}, []string{"etcd", "etcd2"}, []string{}},
|
exclude: nil,
|
||||||
{[]string{}, []string{"*"}, []string{""}, []string{"etcd", "etcd2"}},
|
expected: []string{"etcd", "etcd2"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Empty filters matches all",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: []string{"etcd", "etcd2"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Match all containers",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{"*"},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: []string{"etcd", "etcd2"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Include prefix match",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{"etc*"},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: []string{"etcd", "etcd2"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Exact match",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{"etcd"},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: []string{"etcd"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Star matches zero length",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{"etcd2*"},
|
||||||
|
exclude: []string{},
|
||||||
|
expected: []string{"etcd2"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Exclude matches all",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{},
|
||||||
|
exclude: []string{"etc*"},
|
||||||
|
expected: []string{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Exclude single",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{},
|
||||||
|
exclude: []string{"etcd"},
|
||||||
|
expected: []string{"etcd2"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Exclude all",
|
||||||
|
containers: [][]string{
|
||||||
|
{"/etcd"},
|
||||||
|
{"/etcd2"},
|
||||||
|
},
|
||||||
|
include: []string{"*"},
|
||||||
|
exclude: []string{"*"},
|
||||||
|
expected: []string{},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Exclude item matching include",
|
||||||
|
containers: [][]string{
|
||||||
|
{"acme"},
|
||||||
|
{"foo"},
|
||||||
|
{"acme-test"},
|
||||||
|
},
|
||||||
|
include: []string{"acme*"},
|
||||||
|
exclude: []string{"*test*"},
|
||||||
|
expected: []string{"acme"},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Exclude item no wildcards",
|
||||||
|
containers: [][]string{
|
||||||
|
{"acme"},
|
||||||
|
{"acme-test"},
|
||||||
|
},
|
||||||
|
include: []string{"acme*"},
|
||||||
|
exclude: []string{"test"},
|
||||||
|
expected: []string{"acme", "acme-test"},
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
for _, tt := range gatherContainerNames {
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
t.Run("", func(t *testing.T) {
|
|
||||||
var acc testutil.Accumulator
|
var acc testutil.Accumulator
|
||||||
|
|
||||||
|
newClientFunc := func(host string, tlsConfig *tls.Config) (Client, error) {
|
||||||
|
client := baseClient
|
||||||
|
client.ContainerListF = func(context.Context, types.ContainerListOptions) ([]types.Container, error) {
|
||||||
|
var containers []types.Container
|
||||||
|
for _, names := range tt.containers {
|
||||||
|
containers = append(containers, types.Container{
|
||||||
|
Names: names,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return containers, nil
|
||||||
|
}
|
||||||
|
return &client, nil
|
||||||
|
}
|
||||||
|
|
||||||
d := Docker{
|
d := Docker{
|
||||||
newClient: newClient,
|
newClient: newClientFunc,
|
||||||
ContainerInclude: tt.include,
|
ContainerInclude: tt.include,
|
||||||
ContainerExclude: tt.exclude,
|
ContainerExclude: tt.exclude,
|
||||||
}
|
}
|
||||||
|
@ -317,39 +528,21 @@ func TestContainerNames(t *testing.T) {
|
||||||
err := d.Gather(&acc)
|
err := d.Gather(&acc)
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
|
|
||||||
|
// Set of expected names
|
||||||
|
var expected = make(map[string]bool)
|
||||||
|
for _, v := range tt.expected {
|
||||||
|
expected[v] = true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Set of actual names
|
||||||
|
var actual = make(map[string]bool)
|
||||||
for _, metric := range acc.Metrics {
|
for _, metric := range acc.Metrics {
|
||||||
if metric.Measurement == "docker_container_cpu" {
|
if name, ok := metric.Tags["container_name"]; ok {
|
||||||
if val, ok := metric.Tags["container_name"]; ok {
|
actual[name] = true
|
||||||
var found bool = false
|
|
||||||
for _, cname := range tt.expected {
|
|
||||||
if val == cname {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
t.Errorf("Got unexpected container of %s. Test was -> Include: %s, Exclude: %s", val, tt.include, tt.exclude)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, metric := range acc.Metrics {
|
require.Equal(t, expected, actual)
|
||||||
if metric.Measurement == "docker_container_cpu" {
|
|
||||||
if val, ok := metric.Tags["container_name"]; ok {
|
|
||||||
var found bool = false
|
|
||||||
for _, cname := range tt.notexpected {
|
|
||||||
if val == cname {
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if found {
|
|
||||||
t.Errorf("Got unexpected container of %s. Test was -> Include: %s, Exclude: %s", val, tt.include, tt.exclude)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue