66use Mockery \Adapter \Phpunit \MockeryPHPUnitIntegration ;
77use Mockery \MockInterface ;
88use PHPUnit \Framework \TestCase ;
9+ use ScriptFUSION \Porter \Connector \AsyncDataSource ;
910use ScriptFUSION \Porter \Connector \Connector ;
1011use ScriptFUSION \Porter \Connector \DataSource ;
1112use ScriptFUSION \Porter \Connector \ImportConnector ;
@@ -28,11 +29,15 @@ final class ImportConnectorTest extends TestCase
2829 /** @var DataSource|MockInterface */
2930 private $ source ;
3031
32+ /** @var AsyncDataSource|MockInterface */
33+ private $ asyncSource ;
34+
3135 protected function setUp ()
3236 {
3337 parent ::setUp ();
3438
3539 $ this ->source = \Mockery::mock (DataSource::class);
40+ $ this ->asyncSource = \Mockery::mock (AsyncDataSource::class);
3641 }
3742
3843 /**
@@ -135,7 +140,7 @@ public function testAsyncUserRecoverableExceptionHandler(): void
135140 );
136141
137142 try {
138- wait ($ connector ->fetchAsync ($ this ->source ));
143+ wait ($ connector ->fetchAsync ($ this ->asyncSource ));
139144 } catch (FailingTooHardException $ exception ) {
140145 // This is fine.
141146 }
@@ -158,7 +163,7 @@ public function testAsyncResourceRecoverableExceptionHandler(): void
158163 $ connector ->setRecoverableExceptionHandler (self ::createAsyncRecoverableExceptionHandler ());
159164
160165 try {
161- wait ($ connector ->fetchAsync ($ this ->source ));
166+ wait ($ connector ->fetchAsync ($ this ->asyncSource ));
162167 } catch (FailingTooHardException $ exception ) {
163168 // This is fine.
164169 }
@@ -183,7 +188,7 @@ public function testAsyncUserAndResourceRecoverableExceptionHandlers(): void
183188 $ connector ->setRecoverableExceptionHandler (self ::createAsyncRecoverableExceptionHandler ());
184189
185190 try {
186- wait ($ connector ->fetchAsync ($ this ->source ));
191+ wait ($ connector ->fetchAsync ($ this ->asyncSource ));
187192 } catch (FailingTooHardException $ exception ) {
188193 // This is fine.
189194 }
0 commit comments