Files
windows_exporter/internal/headers/clusapi/objects_test.go
2026-10-10 17:52:05 +02:00

110 lines
3.1 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 clusapi
import (
"context"
"errors"
"testing"
"time"
"golang.org/x/sys/windows"
)
func TestControlCodes(t *testing.T) {
// Values from clusapi.h (CLUSCTL_GROUP_CODES).
for code, want := range map[uint32]uint32{
objectGroup<<24 | ctlGetCharacteristics: 0x03000005,
objectGroup<<24 | ctlGetFlags: 0x03000009,
objectGroup<<24 | ctlGetROCommonProperties: 0x03000055,
objectGroup<<24 | ctlGetCommonProperties: 0x03000059,
// CLUSCTL_NODE_CODES.
objectNode<<24 | ctlGetCharacteristics: 0x04000005,
objectNode<<24 | ctlGetFlags: 0x04000009,
objectNode<<24 | ctlGetROCommonProperties: 0x04000055,
objectNode<<24 | ctlGetCommonProperties: 0x04000059,
// CLUSCTL_NETWORK_CODES.
objectNetwork<<24 | ctlGetCharacteristics: 0x05000005,
objectNetwork<<24 | ctlGetFlags: 0x05000009,
objectNetwork<<24 | ctlGetROCommonProperties: 0x05000055,
objectNetwork<<24 | ctlGetCommonProperties: 0x05000059,
} {
if code != want {
t.Errorf("control code %#x, want %#x", code, want)
}
}
if resourceGetROCommonProperties != 1<<24|ctlGetROCommonProperties || resourceGetFlags != 1<<24|ctlGetFlags {
t.Error("object control codes diverge from the resource control codes")
}
}
func TestGroupStateBuffers(t *testing.T) {
calls := 0
state, node, _, err := stateBuffers(time.Time{}, func(node, _ []uint16) (uint32, uint32, uint32, error) {
calls++
if calls == 1 {
return ^uint32(0), 300, 0, windows.ERROR_MORE_DATA
}
copy(node, []uint16{'N', 'B', 0})
return 0, 2, 0, nil
})
if err != nil || state != 0 || node != "NB" || calls != 2 {
t.Fatalf("state=%d node=%q calls=%d err=%v", state, node, calls, err)
}
}
func TestClusterClosedObjects(t *testing.T) {
c := &Cluster{}
if err := c.Close(); err != nil {
t.Fatal(err)
}
if _, err := c.Groups(time.Time{}); err == nil {
t.Fatal("read closed cluster")
}
if _, err := c.Nodes(time.Time{}); err == nil {
t.Fatal("read closed cluster")
}
if _, err := c.Networks(time.Time{}); err == nil {
t.Fatal("read closed cluster")
}
}
func TestUnopenedClusterObjectsExpiredBudget(t *testing.T) {
c := &Cluster{}
_, err := c.Groups(time.Now().Add(-time.Second))
if errors.Is(err, context.DeadlineExceeded) {
if c.handle != 0 || c.closed {
t.Fatal("expired scrape changed unopened cluster lifecycle")
}
return
}
// Hosts without the Failover Clustering feature lack clusapi.dll.
if err == nil {
t.Fatal("expired budget accepted")
}
}