1414
1515use Bazinga \GeocoderBundle \Plugin \FakeIpPlugin ;
1616use Geocoder \Query \GeocodeQuery ;
17+ use Geocoder \Query \Query ;
1718use PHPUnit \Framework \TestCase ;
1819
1920/**
@@ -26,7 +27,8 @@ public function testSimpleHandleQuery()
2627 $ fakeIpPlugin = new FakeIpPlugin ('127.0.0.1 ' , '123.123.123.123 ' );
2728 $ query = GeocodeQuery::create ('127.0.0.1 ' );
2829
29- $ fakeIpPlugin ->handleQuery ($ query , function () {}, function () {});
30+ /** @var Query $query */
31+ $ query = $ fakeIpPlugin ->handleQuery ($ query , function (Query $ query ) { return $ query ; }, function () {});
3032
3133 $ this ->assertSame ($ query ->getText (), '123.123.123.123 ' );
3234 }
@@ -36,7 +38,8 @@ public function testHandleQueryUsingFaker()
3638 $ fakeIpPlugin = new FakeIpPlugin ('127.0.0.1 ' , '192.168.1.1 ' , true );
3739 $ query = GeocodeQuery::create ('127.0.0.1 ' );
3840
39- $ fakeIpPlugin ->handleQuery ($ query , function () {}, function () {});
41+ /** @var Query $query */
42+ $ query = $ fakeIpPlugin ->handleQuery ($ query , function (Query $ query ) { return $ query ; }, function () {});
4043
4144 $ this ->assertNotSame ($ query ->getText (), '192.168.1.1 ' );
4245 }
0 commit comments