clean up network test

sansecio · Feb 10, 2025 · 58be777 · 58be777
1 parent 62bf16c
commit 58be777
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/Test/Model/RuleTest.php b/Test/Model/RuleTest.php
@@ -30,8 +30,13 @@ public function testRuleRegex()
 
     public function testRuleNetwork()
     {
-        $_SERVER['REMOTE_ADDR'] = '123.123.123.123';
-        $rule = new Rule(new IP(), 'block', [
+        $ipMock = $this->getMockBuilder(IP::class)
+            ->onlyMethods(['collectRequestIPs'])
+            ->getMock();
+
+        $ipMock->method('collectRequestIPs')->willReturn(['123.123.123.123']);
+
+        $rule = new Rule($ipMock, 'block', [
             new Condition('req.ip', 'network', '123.123.123.0/24')
         ]);
         $this->assertTrue($rule->matches($this->createMock(Http::class)));