free-turn-proxy/internal/stats/stats_test.go

package stats

import (
	"strings"
	"testing"
	"time"
)

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

	cases := []struct {
		bytes    uint64
		interval time.Duration
		want     string
	}{
		{0, time.Second, "0 bit/s"},
		{125, time.Second, "1.0 kbit/s"},      // 1000 bit/s
		{125_000, time.Second, "1.00 Mbit/s"}, // 1_000_000 bit/s
		{50, time.Second, "400 bit/s"},
		{1_000_000, 5 * time.Second, "1.60 Mbit/s"},
	}
	for _, c := range cases {
		got := FormatBitsPerSecond(c.bytes, c.interval)
		if got != c.want {
			t.Errorf("FormatBitsPerSecond(%d, %s) = %q, want %q", c.bytes, c.interval, got, c.want)
		}
	}

	// interval <= 0 should not panic and should default to 1s.
	if got := FormatBitsPerSecond(125, 0); !strings.Contains(got, "kbit/s") {
		t.Errorf("zero interval fallback: got %q", got)
	}
}

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

	cases := []struct {
		bytes uint64
		want  string
	}{
		{0, "0 B"},
		{1023, "1023 B"},
		{1024, "1.0 KiB"},
		{1024 * 1024, "1.00 MiB"},
		{2 * 1024 * 1024, "2.00 MiB"},
	}
	for _, c := range cases {
		got := FormatByteCount(c.bytes)
		if got != c.want {
			t.Errorf("FormatByteCount(%d) = %q, want %q", c.bytes, got, c.want)
		}
	}
}

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

	s := New(false)
	s.AddTx(1234)
	s.AddRx(5678)
	if got := s.tx.Load(); got != 0 {
		t.Errorf("disabled AddTx leaked: tx=%d", got)
	}
	if got := s.rx.Load(); got != 0 {
		t.Errorf("disabled AddRx leaked: rx=%d", got)
	}
}

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

	s := New(true)
	s.AddTx(100)
	s.AddTx(50)
	s.AddRx(7)
	s.AddTx(-5) // negative: ignored
	s.AddRx(0)  // zero: ignored
	if got := s.tx.Load(); got != 150 {
		t.Errorf("tx=%d, want 150", got)
	}
	if got := s.rx.Load(); got != 7 {
		t.Errorf("rx=%d, want 7", got)
	}
}