From 3c171cc92c2c62167b338b9eb79bb1ff973fd356 Mon Sep 17 00:00:00 2001
From: wn_ <invalid@email.com>
Date: Sat, 23 Dec 2023 19:52:56 +0000
Subject: Add some tests for UrlHelper::fetch()

---
 tests/UrlHelperTest.php | 68 ++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 64 insertions(+), 4 deletions(-)

(limited to 'tests')

diff --git a/tests/UrlHelperTest.php b/tests/UrlHelperTest.php
index fe4eb5db2..2170ed50b 100644
--- a/tests/UrlHelperTest.php
+++ b/tests/UrlHelperTest.php
@@ -1,5 +1,11 @@
 <?php
 
+use GuzzleHttp\Client;
+use GuzzleHttp\Handler\MockHandler;
+use GuzzleHttp\HandlerStack;
+use GuzzleHttp\Psr7\Response;
+use GuzzleHttp\Psr7\Request;
+use GuzzleHttp\Exception\RequestException;
 use PHPUnit\Framework\TestCase;
 
 final class UrlHelperTest extends TestCase {
@@ -13,16 +19,22 @@ final class UrlHelperTest extends TestCase {
 		// magnet allowed because it's a href attribute
 		$this->assertEquals(
 			'magnet:?xt=urn:btih:...',
-			UrlHelper::rewrite_relative('http://example.com/example/',
+			UrlHelper::rewrite_relative(
+				'http://example.com/example/',
 				'magnet:?xt=urn:btih:...',
-				"a", "href", "")
+				"a",
+				"href",
+				""
+			)
 		);
 
 		// disallowed magnet
 		$this->assertEquals(
 			'http://example.com?xt=urn:btih:...',
-			UrlHelper::rewrite_relative('http://example.com/example/',
-				'magnet:?xt=urn:btih:...')
+			UrlHelper::rewrite_relative(
+				'http://example.com/example/',
+				'magnet:?xt=urn:btih:...'
+			)
 		);
 
 		$this->assertEquals(
@@ -49,6 +61,54 @@ final class UrlHelperTest extends TestCase {
 			'http://www.example.com/test',
 			UrlHelper::rewrite_relative('http://www.example.com/test2 ', 'http://www.example.com/test')
 		);
+	}
+
+	public function test_fetch(): void {
+		$mock = new MockHandler();
+
+		UrlHelper::$client = new Client([
+			'handler' => HandlerStack::create($mock),
+		]);
+
+		$mock->append(new Response(200, [], 'Hello, World'));
+		$result = UrlHelper::fetch('https://www.example.com');
+		$this->assertEquals(200, UrlHelper::$fetch_last_error_code);
+		$this->assertEquals('Hello, World', $result);
+
+		foreach (['ftp://ftp.example.com', 'http://127.0.0.1', 'blah', '', 42, null] as $url) {
+			$result = UrlHelper::fetch($url);
+			$this->assertFalse($result);
+		}
+
+		$mock->append(new Response(200, ['Content-Length' => PHP_INT_MAX]));
+		$result = UrlHelper::fetch('https://www.example.com/very-large-content-length');
+		$this->assertFalse($result);
+
+		$mock->append(new Response(301, ['Location' => 'https://www.example.com']));
+		$result = UrlHelper::fetch(['url' => 'https://example.com', 'followlocation' => false]);
+		$this->assertFalse($result);
+
+		$mock->append(
+			new Response(301, ['Location' => 'http://127.0.0.1']),
+			new Response(200, [], 'Hello, World'),
+		);
+		$result = UrlHelper::fetch(['url' => 'https://example.com', 'followlocation' => true]);
+		$this->assertFalse($result);
+		$this->assertEquals('URL received after redirection failed extended validation.', UrlHelper::$fetch_last_error);
+		$this->assertEquals('http://127.0.0.1', UrlHelper::$fetch_effective_url);
+
+		$mock->append(new Response(200, [], ''));
+		$result = UrlHelper::fetch('https://www.example.com');
+		$this->assertFalse($result);
+		$this->assertEquals('Successful response, but no content was received.', UrlHelper::$fetch_last_error);
 
+		// Currently failing with `Error: Undefined constant "CURLOPT_HTTPAUTH"`.
+		// $mock->append(
+		// 	new Response(403, []),
+		// 	new Response(200, [], 'Hello, World'),
+		// );
+		// $result = UrlHelper::fetch(['url' => 'https://example.com/requires-credentials', 'login' => 'some_username', 'pass' => 'some_password']);
+		// $this->assertEquals(200, UrlHelper::$fetch_last_error_code);
+		// $this->assertEquals('Hello, World', $result);
 	}
 }
-- 
cgit v1.2.3-54-g00ecf