recorders_test.go

// Copyright The OpenTelemetry Authors
// SPDX-License-Identifier: Apache-2.0

//go:build windows

package sqlserverreceiver // import "github.com/open-telemetry/opentelemetry-collector-contrib/receiver/sqlserverreceiver"

import (
	"bufio"
	"fmt"
	"os"
	"path/filepath"
	"strings"
	"testing"

	"github.com/stretchr/testify/require"
)

// TestPerfCounterRecorders requires each perf counter record created to exist in the available counters.
func TestPerfCounterRecorders(t *testing.T) {
	t.Run("default", func(t *testing.T) {
		expectedCounters := getAvailableCounters(t)

		for _, counter := range perfCounterRecorders {
			for counterPath := range counter.recorders {
				counterFullName := fmt.Sprintf("%s:%s %s", defaultObjectName, counter.object, counterPath)
				_, ok := expectedCounters[counterFullName]
				require.True(t, ok, fmt.Sprintf("counter %s not found", counterFullName))
			}
		}
	})

	t.Run("named", func(t *testing.T) {
		expectedCounters := getAvailableNamedInstanceCounters(t)

		for _, counter := range perfCounterRecorders {
			for counterPath := range counter.recorders {
				counterFullName := fmt.Sprintf("MSSQL$TEST_NAME:%s %s", counter.object, counterPath)
				_, ok := expectedCounters[counterFullName]
				require.True(t, ok, fmt.Sprintf("counter %s not found", counterFullName))
			}
		}
	})

}

// getAvailableCounters populates a map containing all available counters.
func getAvailableCounters(t *testing.T) map[string]bool {
	file, err := os.Open(filepath.Join("testdata", "counters.txt"))
	require.NoError(t, err)

	defer file.Close()

	scanner := bufio.NewScanner(file)
	counterNames := map[string]bool{}
	for scanner.Scan() {
		if scanner.Text() != "" {
			nameNoBackslash := strings.ReplaceAll(scanner.Text(), "\\", " ")
			nameNoInstance := strings.TrimSpace(strings.ReplaceAll(nameNoBackslash, "(*)", ""))
			counterNames[nameNoInstance] = true
		}
	}

	return counterNames
}

// getAvailableCounters populates a map containing all available counters.
func getAvailableNamedInstanceCounters(t *testing.T) map[string]bool {
	file, err := os.Open(filepath.Join("testdata", "named-instance-counters.txt"))
	require.NoError(t, err)

	defer file.Close()

	scanner := bufio.NewScanner(file)
	counterNames := map[string]bool{}
	for scanner.Scan() {
		if scanner.Text() != "" {
			nameNoBackslash := strings.ReplaceAll(scanner.Text(), "\\", " ")
			nameNoInstance := strings.TrimSpace(strings.ReplaceAll(nameNoBackslash, "(*)", ""))
			counterNames[nameNoInstance] = true
		}
	}

	return counterNames
}