package victoriametrics

import (
	"bytes"
	"context"
	"testing"

	"go.unistack.org/micro/v4/client"
	"go.unistack.org/micro/v4/codec"
	"go.unistack.org/micro/v4/meter"
)

func BenchmarkBuildName(b *testing.B) {
	m := NewMeter(meter.Labels("pod", "xxx"))
	if err := m.Init(); err != nil {
		b.Fatal(err)
	}
	im := m.(*victoriametricsMeter)
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		name := im.buildName("micro_foo", "bar", "baz", "aaa", "b", "ccc", "d")
		_ = name
	}
}

func TestBuildName(t *testing.T) {
	m := NewMeter()
	im := m.(*victoriametricsMeter)
	check := `micro_foo{bar="baz",aaa="b",ccc="d"}`
	name := im.buildName("micro_foo", "bar", "baz", "aaa", "b", "ccc", "d")
	if name != check {
		t.Fatalf("metric name error: %s != %s", name, check)
	}

	cnt := m.Counter("counter", "key", "val")
	cnt.Inc()
}

func TestWrapper(t *testing.T) {
	m := NewMeter()
	if err := m.Init(); err != nil {
		t.Fatal(err)
	}
	ctx := context.Background()

	c := client.NewClient(client.Meter(m))
	if err := c.Init(); err != nil {
		t.Fatal(err)
	}
	rsp := &codec.Frame{}
	req := &codec.Frame{}
	err := c.Call(ctx, c.NewRequest("svc2", "Service.Method", req), rsp)
	_, _ = rsp, err
	buf := bytes.NewBuffer(nil)
	_ = m.Write(buf, meter.WriteProcessMetrics(false), meter.WriteFDMetrics(false))
	if !bytes.Contains(buf.Bytes(), []byte(`micro_client_request_total{endpoint="Service.Method",status="failure",code="500"} 1`)) {
		t.Fatalf("invalid metrics output: %s", buf.Bytes())
	}
}