Function TestExecute_PerModelIsolation

aibridge/circuitbreaker/circuitbreaker_test.go:18–65

(t *testing.T)

)

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

	sonnetCalls := atomic.Int32{}
	haikuCalls := atomic.Int32{}

	cbs := circuitbreaker.NewProviderCircuitBreakers("test", &config.CircuitBreaker{
		FailureThreshold: 1,
		Interval:         time.Minute,
		Timeout:          time.Minute,
		MaxRequests:      1,
	}, func(endpoint, model string, from, to gobreaker.State) {}, nil)

	endpoint := "/v1/messages"
	sonnetModel := "claude-sonnet-4-20250514"
	haikuModel := "claude-3-5-haiku-20241022"

	// Trip circuit on sonnet model (returns 503)
	w := httptest.NewRecorder()
	err := cbs.Execute(endpoint, sonnetModel, w, func(rw http.ResponseWriter) error {
		sonnetCalls.Add(1)
		rw.WriteHeader(http.StatusServiceUnavailable)
		return nil
	})
	assert.NoError(t, err)
	assert.Equal(t, int32(1), sonnetCalls.Load())

	// Second sonnet request should be blocked by circuit breaker
	w = httptest.NewRecorder()
	err = cbs.Execute(endpoint, sonnetModel, w, func(rw http.ResponseWriter) error {
		sonnetCalls.Add(1)
		rw.WriteHeader(http.StatusOK)
		return nil
	})
	assert.True(t, errors.Is(err, circuitbreaker.ErrCircuitOpen))
	assert.Equal(t, int32(1), sonnetCalls.Load()) // No new call
	assert.Equal(t, http.StatusServiceUnavailable, w.Code)

	// Haiku model on same endpoint should still work (independent circuit)
	w = httptest.NewRecorder()
	err = cbs.Execute(endpoint, haikuModel, w, func(rw http.ResponseWriter) error {
		haikuCalls.Add(1)
		rw.WriteHeader(http.StatusOK)
		return nil
	})
	assert.NoError(t, err)
	assert.Equal(t, int32(1), haikuCalls.Load())
}

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