14
14
15
15
use Bazinga \GeocoderBundle \Plugin \FakeIpPlugin ;
16
16
use Geocoder \Query \GeocodeQuery ;
17
+ use Geocoder \Query \Query ;
17
18
use PHPUnit \Framework \TestCase ;
18
19
19
20
/**
@@ -26,7 +27,8 @@ public function testSimpleHandleQuery()
26
27
$ fakeIpPlugin = new FakeIpPlugin ('127.0.0.1 ' , '123.123.123.123 ' );
27
28
$ query = GeocodeQuery::create ('127.0.0.1 ' );
28
29
29
- $ fakeIpPlugin ->handleQuery ($ query , function () {}, function () {});
30
+ /** @var Query $query */
31
+ $ query = $ fakeIpPlugin ->handleQuery ($ query , function (Query $ query ) { return $ query ; }, function () {});
30
32
31
33
$ this ->assertSame ($ query ->getText (), '123.123.123.123 ' );
32
34
}
@@ -36,7 +38,8 @@ public function testHandleQueryUsingFaker()
36
38
$ fakeIpPlugin = new FakeIpPlugin ('127.0.0.1 ' , '192.168.1.1 ' , true );
37
39
$ query = GeocodeQuery::create ('127.0.0.1 ' );
38
40
39
- $ fakeIpPlugin ->handleQuery ($ query , function () {}, function () {});
41
+ /** @var Query $query */
42
+ $ query = $ fakeIpPlugin ->handleQuery ($ query , function (Query $ query ) { return $ query ; }, function () {});
40
43
41
44
$ this ->assertNotSame ($ query ->getText (), '192.168.1.1 ' );
42
45
}
0 commit comments