Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Core/Extensions/UriExtensionsTests.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using System;
using NUnit.Framework;
using Umbraco.Extensions;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Core.Extensions;

[TestFixture]
public class UriExtensionsTests
{
    [TestCase("http://www.domain.com/foo/bar", "/", "http://www.domain.com/")]
    [TestCase("http://www.domain.com/foo/bar#hop", "/", "http://www.domain.com/")]
    [TestCase("http://www.domain.com/foo/bar?q=2#hop", "/", "http://www.domain.com/?q=2")]
    [TestCase("http://www.domain.com/foo/bar", "/path/to/page", "http://www.domain.com/path/to/page")]
    [TestCase("http://www.domain.com/foo/bar", "/path/to/page/", "http://www.domain.com/path/to/page/")]
    [TestCase("http://www.domain.com/", "/path/to/page/", "http://www.domain.com/path/to/page/")]
    [TestCase("http://www.domain.com", "/path/to/page/", "http://www.domain.com/path/to/page/")]
    [TestCase("http://www.domain.com/foo?q=3", "/path/to/page/", "http://www.domain.com/path/to/page/?q=3")]
    [TestCase("http://www.domain.com/foo#bang", "/path/to/page/", "http://www.domain.com/path/to/page/")]
    [TestCase("http://www.domain.com/foo?q=3#bang", "/path/to/page/", "http://www.domain.com/path/to/page/?q=3")]
    public void RewritePath(string input, string path, string expected)
    {
        var source = new Uri(input);
        var output = source.Rewrite(path);
        Assert.AreEqual(expected, output.ToString());
    }

    [TestCase("http://www.domain.com/", "path/to/page/", typeof(ArgumentException))]
    [TestCase("http://www.domain.com", "path/to/page/", typeof(ArgumentException))]
    public void RewritePath_Exceptions(string input, string path, Type exception)
    {
        var source = new Uri(input);
        Assert.Throws(exception, () =>
        {
            var output = source.Rewrite(path);
        });
    }

    [TestCase("http://www.domain.com/foo/bar", "/path/to/page", "", "http://www.domain.com/path/to/page")]
    [TestCase("http://www.domain.com/foo/bar?k=3", "/path/to/page", "", "http://www.domain.com/path/to/page")]
    [TestCase("http://www.domain.com/foo/bar?k=3#hop", "/path/to/page", "", "http://www.domain.com/path/to/page")]
    [TestCase("http://www.domain.com/foo/bar", "/path/to/page", "?x=12", "http://www.domain.com/path/to/page?x=12")]
    [TestCase("http://www.domain.com/foo/bar#hop", "/path/to/page", "?x=12", "http://www.domain.com/path/to/page?x=12")]
    [TestCase("http://www.domain.com/foo/bar?k=3", "/path/to/page", "?x=12", "http://www.domain.com/path/to/page?x=12")]
    [TestCase("http://www.domain.com/foo/bar?k=3#hop", "/path/to/page", "?x=12", "http://www.domain.com/path/to/page?x=12")]
    public void RewritePathAndQuery(string input, string path, string query, string expected)
    {
        var source = new Uri(input);
        var output = source.Rewrite(path, query);
        Assert.AreEqual(expected, output.ToString());
    }

    [TestCase("http://www.domain.com/", "path/to/page/", "", typeof(ArgumentException))]
    [TestCase("http://www.domain.com/", "/path/to/page/", "x=27", typeof(ArgumentException))]
    public void RewritePathAndQuery_Exceptions(string input, string path, string query, Type exception)
    {
        var source = new Uri(input);
        Assert.Throws(exception, () =>
        {
            var output = source.Rewrite(path, query);
        });
    }

    [TestCase("http://www.domain.com", "/")]
    [TestCase("http://www.domain.com/", "/")]
    [TestCase("http://www.domain.com/foo", "/foo")]
    [TestCase("http://www.domain.com/foo/", "/foo/")]
    [TestCase("http://www.domain.com/foo/bar", "/foo/bar")]
    [TestCase("http://www.domain.com/foo/bar%20nix", "/foo/bar%20nix")]
    [TestCase("http://www.domain.com/foo/bar?q=7#hop", "/foo/bar")]
    [TestCase("/", "/")]
    [TestCase("/foo", "/foo")]
    [TestCase("/foo/", "/foo/")]
    [TestCase("/foo/bar", "/foo/bar")]
    [TestCase("/foo/bar?q=7#hop", "/foo/bar")]
    [TestCase("/foo%20bar/pof", "/foo%20bar/pof")]
    public void GetSafeAbsolutePath(string input, string expected)
    {
        var source = new Uri(input, UriKind.RelativeOrAbsolute);
        var output = source.GetSafeAbsolutePath();
        Assert.AreEqual(expected, output);
    }

    [TestCase("http://www.domain.com/foo/bar%20nix", "/foo/bar nix")]
    public void GetAbsolutePathDecoded(string input, string expected)
    {
        var source = new Uri(input, UriKind.RelativeOrAbsolute);
        var output = source.GetAbsolutePathDecoded();
        Assert.AreEqual(expected, output);
    }

    [TestCase("http://www.domain.com/foo/bar%20nix", "/foo/bar nix")]
    [TestCase("/foo%20bar/pof", "/foo bar/pof")]
    public void GetSafeAbsolutePathDecoded(string input, string expected)
    {
        var source = new Uri(input, UriKind.RelativeOrAbsolute);
        var output = source.GetSafeAbsolutePathDecoded();
        Assert.AreEqual(expected, output);
    }

    [TestCase("http://www.domain.com/path/to/page", "http://www.domain.com/path/to/page/")]
    [TestCase("http://www.domain.com/path/to/", "http://www.domain.com/path/to/")]
    [TestCase("http://www.domain.com/", "http://www.domain.com/")]
    [TestCase("http://www.domain.com", "http://www.domain.com/")]
    [TestCase("http://www.domain.com/path/to?q=3#yop", "http://www.domain.com/path/to/?q=3")]
    public void EndPathWithSlash(string input, string expected)
    {
        var source = new Uri(input);
        var output = source.EndPathWithSlash();
        Assert.AreEqual(expected, output.ToString());
    }

    [TestCase("http://www.domain.com/path/to/page", "http://www.domain.com/path/to/page")]
    [TestCase("http://www.domain.com/path/to/", "http://www.domain.com/path/to")]
    [TestCase("http://www.domain.com/", "http://www.domain.com/")]
    [TestCase("http://www.domain.com", "http://www.domain.com/")]
    [TestCase("http://www.domain.com/path/to/?q=3#yop", "http://www.domain.com/path/to?q=3")]
    public void TrimPathEndSlash(string input, string expected)
    {
        var source = new Uri(input);
        var output = source.TrimPathEndSlash();
        Assert.AreEqual(expected, output.ToString());
    }

    [TestCase("/foo/bar", "http://www.domain.com", "http://www.domain.com/foo/bar")]
    [TestCase("/foo/bar", "http://www.domain.com/dang/dang", "http://www.domain.com/foo/bar")]
    [TestCase("/", "http://www.domain.com/dang/dang", "http://www.domain.com/")]
    [TestCase("/foo/bar", "http://www.domain.com/dang/dang?q=3#dang", "http://www.domain.com/foo/bar")]
    [TestCase("/foo/bar?k=6#yop", "http://www.domain.com/dang/dang?q=3#dang", "http://www.domain.com/foo/bar?k=6")]
    public void MakeAbsolute(string input, string reference, string expected)
    {
        var source = new Uri(input, UriKind.Relative);
        var absolute = new Uri(reference);
        var output = source.MakeAbsolute(absolute);
        Assert.AreEqual(expected, output.ToString());
    }

    [TestCase("http://www.domain.com/path/to/page", "http://www.domain.com/path/to/page")]
    [TestCase("http://www.domain.com/path/to/page/", "http://www.domain.com/path/to/page/")]
    [TestCase("http://www.domain.com", "http://www.domain.com/")]
    [TestCase("http://www.domain.com/", "http://www.domain.com/")]
    [TestCase("http://www.domain.com/path/to?q=3#yop", "http://www.domain.com/path/to?q=3#yop")]
    [TestCase("http://www.domain.com/path/to/?q=3#yop", "http://www.domain.com/path/to/?q=3#yop")]
    [TestCase("http://www.domain.com:666/path/to/page", "http://www.domain.com/path/to/page")]
    [TestCase("http://www.domain.com:666/path/to/page/", "http://www.domain.com/path/to/page/")]
    [TestCase("http://www.domain.com:666", "http://www.domain.com/")]
    [TestCase("http://www.domain.com:666/", "http://www.domain.com/")]
    [TestCase("http://www.domain.com:666/path/to?q=3#yop", "http://www.domain.com/path/to?q=3#yop")]
    [TestCase("http://www.domain.com:666/path/to/?q=3#yop", "http://www.domain.com/path/to/?q=3#yop")]
    public void WithoutPort(string input, string expected)
    {
        var source = new Uri(input);
        var output = source.WithoutPort();
        Assert.AreEqual(expected, output.ToString());
    }
}