mirror of
https://github.com/prometheus-community/windows_exporter.git
synced 2026-02-08 05:56:37 +00:00
295 lines
7.9 KiB
Go
295 lines
7.9 KiB
Go
//go:build windows
|
|
|
|
package mi_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/prometheus-community/windows_exporter/internal/mi"
|
|
"github.com/prometheus-community/windows_exporter/internal/testutils"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/sys/windows"
|
|
)
|
|
|
|
type win32Process struct {
|
|
Name string `mi:"Name"`
|
|
}
|
|
|
|
type wmiPrinter struct {
|
|
Name string `mi:"Name"`
|
|
Default bool `mi:"Default"`
|
|
PrinterStatus uint16 `mi:"PrinterStatus"`
|
|
JobCountSinceLastReset uint32 `mi:"JobCountSinceLastReset"`
|
|
}
|
|
|
|
type wmiPrintJob struct {
|
|
Name string `mi:"Name"`
|
|
Status string `mi:"Status"`
|
|
}
|
|
|
|
func Test_MI_Application_Initialize(t *testing.T) {
|
|
application, err := mi.Application_Initialize()
|
|
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.Application_Initialize()
|
|
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.Application_Initialize()
|
|
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.Application_Initialize()
|
|
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.Application_Initialize()
|
|
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.Application_Initialize()
|
|
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) {
|
|
t.Skip("This test is disabled because it is not deterministic and may fail on some systems.")
|
|
|
|
application, err := mi.Application_Initialize()
|
|
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, Default, PrinterStatus, JobCountSinceLastReset FROM win32_Printer")
|
|
require.NoError(t, err)
|
|
|
|
queryPrinterJob, err := mi.NewQuery("SELECT Name, Status FROM win32_PrintJob")
|
|
require.NoError(t, err)
|
|
|
|
for range 1000 {
|
|
var wmiPrinters []wmiPrinter
|
|
err := session.Query(&wmiPrinters, mi.NamespaceRootCIMv2, queryPrinter)
|
|
require.NoError(t, err)
|
|
|
|
var wmiPrintJobs []wmiPrintJob
|
|
err = session.Query(&wmiPrintJobs, mi.NamespaceRootCIMv2, queryPrinterJob)
|
|
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)
|
|
}
|