mirror of
https://github.com/prometheus-community/windows_exporter.git
synced 2026-02-08 05:56:37 +00:00
290 lines
7.9 KiB
Go
290 lines
7.9 KiB
Go
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
// Copyright 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.
|
|
|
|
//go:build windows
|
|
|
|
package mi_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/prometheus-community/windows_exporter/internal/mi"
|
|
"github.com/prometheus-community/windows_exporter/internal/utils/testutils"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
type win32Process struct {
|
|
Name string `mi:"Name"`
|
|
}
|
|
|
|
func Test_MI_Application_Initialize(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_Application_TestConnection(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
err = session.TestConnection()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_Query(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operation, err := session.QueryInstances(mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, "select Name from win32_process where handle = 0")
|
|
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
instance, moreResults, err := operation.GetInstance()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, instance)
|
|
|
|
count, err := instance.GetElementCount()
|
|
require.NoError(t, err)
|
|
require.NotZero(t, count)
|
|
|
|
element, err := instance.GetElement("Name")
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, element)
|
|
|
|
value, err := element.GetValue()
|
|
require.NoError(t, err)
|
|
require.Equal(t, "System Idle Process", value)
|
|
require.NotEmpty(t, value)
|
|
|
|
require.False(t, moreResults)
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_QueryUnmarshal(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
var processes []win32Process
|
|
|
|
queryProcess, err := mi.NewQuery("select Name from win32_process where handle = 0")
|
|
require.NoError(t, err)
|
|
|
|
err = session.QueryUnmarshal(&processes, mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, queryProcess)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []win32Process{{Name: "System Idle Process"}}, processes)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_EmptyQuery(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operation, err := session.QueryInstances(mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, "SELECT Name, Status FROM win32_PrintJob")
|
|
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
instance, moreResults, err := operation.GetInstance()
|
|
require.NoError(t, err)
|
|
require.Empty(t, instance)
|
|
require.False(t, moreResults)
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_Query_Unmarshal(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
destinationOptions, err := application.NewDestinationOptions()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, destinationOptions)
|
|
|
|
err = destinationOptions.SetTimeout(1 * time.Second)
|
|
require.NoError(t, err)
|
|
|
|
err = destinationOptions.SetLocale(mi.LocaleEnglish)
|
|
require.NoError(t, err)
|
|
|
|
session, err := application.NewSession(destinationOptions)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
operation, err := session.QueryInstances(mi.OperationFlagsStandardRTTI, nil, mi.NamespaceRootCIMv2, mi.QueryDialectWQL, "SELECT Name FROM Win32_Process WHERE Handle = 0 OR Handle = 4")
|
|
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, operation)
|
|
|
|
var processes []win32Process
|
|
|
|
err = operation.Unmarshal(&processes)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []win32Process{{Name: "System Idle Process"}, {Name: "System"}}, processes)
|
|
|
|
err = operation.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_MI_FD_Leak(t *testing.T) {
|
|
application, err := mi.ApplicationInitialize()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, application)
|
|
|
|
session, err := application.NewSession(nil)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, session)
|
|
|
|
currentFileHandle, err := testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
|
|
queryPrinter, err := mi.NewQuery("SELECT Name FROM Win32_Process")
|
|
require.NoError(t, err)
|
|
|
|
for range 300 {
|
|
var processes []win32Process
|
|
|
|
err := session.Query(&processes, mi.NamespaceRootCIMv2, queryPrinter)
|
|
require.NoError(t, err)
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
}
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
|
|
err = session.Close()
|
|
require.NoError(t, err)
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
|
|
err = application.Close()
|
|
require.NoError(t, err)
|
|
|
|
currentFileHandle, err = testutils.GetProcessHandleCount(windows.CurrentProcess())
|
|
require.NoError(t, err)
|
|
|
|
t.Log("Current File Handle Count: ", currentFileHandle)
|
|
}
|