Fix --disable_metrics parsing to accept empty string as input.

Signed-off-by: Vishnu kannan <vishnuk@google.com>
This commit is contained in:
Vishnu kannan 2016-04-06 11:52:44 -07:00
parent 8ddbae4202
commit a663e835c4
2 changed files with 10 additions and 0 deletions

View File

@ -76,6 +76,10 @@ func (ml *metricSetValue) String() string {
}
func (ml *metricSetValue) Set(value string) error {
ignoreMetrics = metricSetValue{}
if value == "" {
return nil
}
for _, metric := range strings.Split(value, ",") {
if ignoreWhitelist.Has(container.MetricKind(metric)) {
(*ml).Add(container.MetricKind(metric))

View File

@ -27,3 +27,9 @@ func TestTcpMetricsAreDisabledByDefault(t *testing.T) {
flag.Parse()
assert.True(t, ignoreMetrics.Has(container.NetworkTcpUsageMetrics))
}
func TestTcpMetricsAreEnabledOnDemand(t *testing.T) {
assert.True(t, ignoreMetrics.Has(container.NetworkTcpUsageMetrics))
ignoreMetrics.Set("")
assert.False(t, ignoreMetrics.Has(container.NetworkTcpUsageMetrics))
}