invoiceninja/tests/Unit/UrlTest.php

<?php
/**
 * Invoice Ninja (https://invoiceninja.com).
 *
 * @link https://github.com/invoiceninja/invoiceninja source repository
 *
 * @copyright Copyright (c) 2021. Invoice Ninja LLC (https://invoiceninja.com)
 *
 * @license https://www.elastic.co/licensing/elastic-license
 */

namespace Tests\Unit;

use Tests\TestCase;

/**
 * @test
 */
class UrlTest extends TestCase
{
    protected function setUp() :void
    {
        parent::setUp();
    }

    public function testNoScheme()
    {
        $url = 'google.com';

        $this->assertEquals('https://google.com', $this->addScheme($url));
    }

    public function testNoSchemeAndTrailingSlash()
    {
        $url = 'google.com/';

        $this->assertEquals('https://google.com', $this->addScheme($url));
    }

    public function testNoSchemeAndTrailingSlashAndHttp()
    {
        $url = 'http://google.com/';

        $this->assertEquals('https://google.com', $this->addScheme($url));
    }

    private function addScheme($url, $scheme = 'https://')
    {
        $url = str_replace('http://', '', $url);

        $url = parse_url($url, PHP_URL_SCHEME) === null ? $scheme.$url : $url;

        return rtrim($url, '/');
    }
}