22
33require "test_helper"
44require "active_support/rescuable"
5- require "active_agent/generation_provider/base"
6- require "active_agent/generation_provider/error_handling"
5+ require "active_agent/generation_provider/_base_provider"
76
87class ErrorHandlingTest < ActiveSupport ::TestCase
98 class TestError < StandardError ; end
@@ -42,7 +41,7 @@ def operation_that_succeeds_on_retry
4241
4342 test "with_error_handling uses rescue_with_handler" do
4443 # The default rescue_from StandardError handler will convert to GenerationProviderError
45- error = assert_raises ( ActiveAgent ::GenerationProvider ::Base ::GenerationProviderError ) do
44+ error = assert_raises ( ActiveAgent ::GenerationProvider ::BaseProvider ::GenerationProviderError ) do
4645 @provider . with_error_handling do
4746 @provider . operation_that_fails
4847 end
@@ -53,7 +52,7 @@ def operation_that_succeeds_on_retry
5352
5453 test "preserves original backtrace" do
5554 original_line = nil
56- error = assert_raises ( ActiveAgent ::GenerationProvider ::Base ::GenerationProviderError ) do
55+ error = assert_raises ( ActiveAgent ::GenerationProvider ::BaseProvider ::GenerationProviderError ) do
5756 @provider . with_error_handling do
5857 original_line = __LINE__ + 1
5958 raise TestError , "With backtrace"
@@ -79,7 +78,7 @@ def operation_that_succeeds_on_retry
7978 @provider . class . retry_on_errors = [ TestError ]
8079 @provider . class . max_retries = 2
8180
82- assert_raises ( ActiveAgent ::GenerationProvider ::Base ::GenerationProviderError ) do
81+ assert_raises ( ActiveAgent ::GenerationProvider ::BaseProvider ::GenerationProviderError ) do
8382 @provider . with_error_handling do
8483 @provider . operation_that_fails
8584 end
@@ -243,7 +242,7 @@ class VerboseProvider < TestProvider
243242 @provider . class . retry_on_errors = [ ]
244243 @provider . class . max_retries = 3
245244
246- assert_raises ( ActiveAgent ::GenerationProvider ::Base ::GenerationProviderError ) do
245+ assert_raises ( ActiveAgent ::GenerationProvider ::BaseProvider ::GenerationProviderError ) do
247246 @provider . with_error_handling do
248247 @provider . operation_that_fails
249248 end
@@ -293,7 +292,7 @@ class VerboseProvider < TestProvider
293292 @provider . config = { "logger" => logger }
294293
295294 error = TestError . new ( "Original error" )
296- assert_raises ( ActiveAgent ::GenerationProvider ::Base ::GenerationProviderError ) do
295+ assert_raises ( ActiveAgent ::GenerationProvider ::BaseProvider ::GenerationProviderError ) do
297296 @provider . send ( :handle_generation_error , error )
298297 end
299298
@@ -304,7 +303,7 @@ class VerboseProvider < TestProvider
304303 # This test would require mocking ActiveSupport::Notifications
305304 # For now, just ensure the method doesn't error
306305 error = TestError . new ( "Test" )
307- wrapped = ActiveAgent ::GenerationProvider ::Base ::GenerationProviderError . new ( "Wrapped" )
306+ wrapped = ActiveAgent ::GenerationProvider ::BaseProvider ::GenerationProviderError . new ( "Wrapped" )
308307
309308 assert_nothing_raised do
310309 @provider . send ( :instrument_error , error , wrapped )
0 commit comments