// Copyright 2026 Alibaba Group Holding Ltd.
//
// 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.

package telemetry

import (
	"context"
	"testing"

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

func TestRecordIsolatedRunWithoutInit(t *testing.T) {
	t.Parallel()

	// Must not panic when instruments aren't initialised.
	RecordIsolatedRun(context.Background(), "success", 42.0)
	RecordIsolatedRun(context.Background(), "error", 5.0)
}

func TestIsolationStatsProvider(t *testing.T) {
	orig := isolationStatsProvider
	t.Cleanup(func() { isolationStatsProvider = orig })

	assert.Nil(t, isolationStatsProvider)

	called := false
	SetIsolationStatsProvider(func() IsolationStats {
		called = true
		return IsolationStats{ActiveSessions: 3, UpperUsageBytes: 1024}
	})

	assert.NotNil(t, isolationStatsProvider)
	stats := isolationStatsProvider()
	assert.True(t, called)
	assert.Equal(t, int64(3), stats.ActiveSessions)
	assert.Equal(t, int64(1024), stats.UpperUsageBytes)
}