frankenphp/metrics_test.go

package frankenphp

import (
	"github.com/prometheus/client_golang/prometheus"
	"github.com/stretchr/testify/require"
	"sync"
	"testing"
	"time"

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

func TestGetWorkerNameForMetrics(t *testing.T) {
	tests := []struct {
		input    string
		expected string
	}{
		{"worker-1", "worker_1"},
		{"worker@name", "worker_name"},
		{"worker name", "worker_name"},
		{"worker/name", "worker_name"},
		{"worker.name", "worker_name"},
		{"////worker////name...//worker", "worker_name_worker"},
	}

	for _, test := range tests {
		result := getWorkerNameForMetrics(test.input)
		assert.Equal(t, test.expected, result)
	}
}

func createPrometheusMetrics() *PrometheusMetrics {
	return &PrometheusMetrics{
		registry:           prometheus.NewRegistry(),
		totalThreads:       prometheus.NewCounter(prometheus.CounterOpts{Name: "total_threads"}),
		busyThreads:        prometheus.NewGauge(prometheus.GaugeOpts{Name: "busy_threads"}),
		totalWorkers:       make(map[string]prometheus.Gauge),
		busyWorkers:        make(map[string]prometheus.Gauge),
		workerRequestTime:  make(map[string]prometheus.Counter),
		workerRequestCount: make(map[string]prometheus.Counter),
		workerCrashes:      make(map[string]prometheus.Counter),
		workerRestarts:     make(map[string]prometheus.Counter),
		readyWorkers:       make(map[string]prometheus.Gauge),
		mu:                 sync.Mutex{},
	}
}

func TestPrometheusMetrics_TotalWorkers(t *testing.T) {
	m := createPrometheusMetrics()

	tests := []struct {
		name   string
		worker string
		num    int
	}{
		{"SetWorkers", "test_worker", 5},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			m.TotalWorkers(tt.worker, tt.num)
			actualName, _ := m.getIdentity(tt.worker)
			_, ok := m.totalWorkers[actualName]
			require.True(t, ok)
		})
	}
}

func TestPrometheusMetrics_StopWorkerRequest(t *testing.T) {
	m := createPrometheusMetrics()
	m.StopWorkerRequest("test_worker", 2*time.Second)

	name := "test_worker"
	_, ok := m.workerRequestTime[name]
	require.False(t, ok)
}

func TestPrometheusMetrics_StartWorkerRequest(t *testing.T) {
	m := createPrometheusMetrics()
	m.StartWorkerRequest("test_worker")

	name := "test_worker"
	_, ok := m.workerRequestCount[name]
	require.False(t, ok)
}