stackdriver_exporter_test.go

// Copyright 2024 The Prometheus Authors
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import "testing"
import "reflect"

func TestParseMetricTypePrefixes(t *testing.T) {
	inputPrefixes := []string{
		"redis.googleapis.com/stats/memory/usage",
		"loadbalancing.googleapis.com/https/request_count",
		"loadbalancing.googleapis.com",
		"redis.googleapis.com/stats/memory/usage_ratio",
		"redis.googleapis.com/stats/memory/usage_ratio",
	}
	expectedOutputPrefixes := []string{
		"loadbalancing.googleapis.com",
		"redis.googleapis.com/stats/memory/usage",
	}

	outputPrefixes := parseMetricTypePrefixes(inputPrefixes)

	if !reflect.DeepEqual(outputPrefixes, expectedOutputPrefixes) {
		t.Errorf("Metric type prefix sanitization did not produce expected output. Expected:\n%s\nGot:\n%s", expectedOutputPrefixes, outputPrefixes)
	}
}

func TestFilterMetricTypePrefixes(t *testing.T) {
	metricPrefixes := []string{
		"redis.googleapis.com/stats/",
	}

	h := &handler{
		metricsPrefixes: metricPrefixes,
	}

	inputFilters := map[string]bool{
		"redis.googleapis.com/stats/memory/usage":       true,
		"redis.googleapis.com/stats/memory/usage_ratio": true,
		"redis.googleapis.com":                          true,
	}

	expectedOutputPrefixes := []string{
		"redis.googleapis.com/stats/memory/usage",
	}

	outputPrefixes := h.filterMetricTypePrefixes(inputFilters)

	if !reflect.DeepEqual(outputPrefixes, expectedOutputPrefixes) {
		t.Errorf("filterMetricTypePrefixes did not produce expected output. Expected:\n%s\nGot:\n%s", expectedOutputPrefixes, outputPrefixes)
	}
}