package aibridged_test

import (
	"net/http"
	"sync/atomic"

	"go.opentelemetry.io/otel"
)

var testTracer = otel.Tracer("aibridged_test")

var _ http.Handler = &mockAIUpstreamServer{}

type mockAIUpstreamServer struct {
	hitCounter atomic.Int32
}

func (m *mockAIUpstreamServer) ServeHTTP(rw http.ResponseWriter, _ *http.Request) {
	m.hitCounter.Add(1)

	rw.WriteHeader(http.StatusTeapot)
	_, _ = rw.Write([]byte(`i am a teapot`))
}

func (m *mockAIUpstreamServer) Hits() int32 {
	return m.hitCounter.Load()
}