mozilla-mobile/firefox-ios/BrowserKit/Tests/CommonTests/Extensions/URLExtensionTests.swift

Revision control

Copy as Markdown

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/
import XCTest
@testable import Common
final class URLExtensionTests: XCTestCase {
private var webServerPort = 1234
func testNormalBaseDomainWithSingleSubdomain() {
// TLD Entry: co.uk
let url = URL(string: "http://a.bbc.co.uk")!
let expected = url.publicSuffix!
XCTAssertEqual("co.uk", expected)
}
func testCanadaComputers() {
let url = URL(string: "http://m.canadacomputers.com")!
let actual = url.baseDomain!
XCTAssertEqual("canadacomputers.com", actual)
}
func testMultipleSuffixesInsideURL() {
let url = URL(string: "http://com:org@m.canadacomputers.co.uk")!
let actual = url.baseDomain!
XCTAssertEqual("canadacomputers.co.uk", actual)
}
func testNormalBaseDomainWithManySubdomains() {
// TLD Entry: co.uk
let url = URL(string: "http://a.b.c.d.bbc.co.uk")!
let expected = url.publicSuffix!
XCTAssertEqual("co.uk", expected)
}
func testWildCardDomainWithSingleSubdomain() {
// TLD Entry: *.kawasaki.jp
let url = URL(string: "http://a.kawasaki.jp")!
let expected = url.publicSuffix!
XCTAssertEqual("a.kawasaki.jp", expected)
}
func testWildCardDomainWithManySubdomains() {
// TLD Entry: *.kawasaki.jp
let url = URL(string: "http://a.b.c.d.kawasaki.jp")!
let expected = url.publicSuffix!
XCTAssertEqual("d.kawasaki.jp", expected)
}
func testExceptionDomain() {
// TLD Entry: !city.kawasaki.jp
let url = URL(string: "http://city.kawasaki.jp")!
let expected = url.publicSuffix!
XCTAssertEqual("kawasaki.jp", expected)
}
// MARK: Base Domain
func testNormalBaseSubdomain() {
// TLD Entry: co.uk
let url = URL(string: "http://bbc.co.uk")!
let expected = url.baseDomain!
XCTAssertEqual("bbc.co.uk", expected)
}
func testNormalBaseSubdomainWithAdditionalSubdomain() {
// TLD Entry: co.uk
let url = URL(string: "http://a.bbc.co.uk")!
let expected = url.baseDomain!
XCTAssertEqual("bbc.co.uk", expected)
}
func testBaseDomainForWildcardDomain() {
// TLD Entry: *.kawasaki.jp
let url = URL(string: "http://a.b.kawasaki.jp")!
let expected = url.baseDomain!
XCTAssertEqual("a.b.kawasaki.jp", expected)
}
func testBaseDomainForWildcardDomainWithAdditionalSubdomain() {
// TLD Entry: *.kawasaki.jp
let url = URL(string: "http://a.b.c.kawasaki.jp")!
let expected = url.baseDomain!
XCTAssertEqual("b.c.kawasaki.jp", expected)
}
func testBaseDomainForExceptionDomain() {
// TLD Entry: !city.kawasaki.jp
let url = URL(string: "http://city.kawasaki.jp")!
let expected = url.baseDomain!
XCTAssertEqual("city.kawasaki.jp", expected)
}
func testBaseDomainForExceptionDomainWithAdditionalSubdomain() {
// TLD Entry: !city.kawasaki.jp
let url = URL(string: "http://a.city.kawasaki.jp")!
let expected = url.baseDomain!
XCTAssertEqual("city.kawasaki.jp", expected)
}
func testBugzillaURLDomain() {
XCTAssertNotNil(url, "URL parses.")
let host = url!.normalizedHost
XCTAssertEqual(host!, "bugzilla.mozilla.org")
XCTAssertEqual(url!.fragment!, "h=dupes%7CData%20%26%20BI%20Services%20Team%7C")
}
func testNormalizedHostReturnsOriginalHost() {
let url = URL(string: "https://mobile.co.uk")!
let host = url.normalizedHost
XCTAssertEqual(host, "mobile.co.uk")
}
func testIPv6Domain() {
let url = URL(string: "http://[::1]/foo/bar")!
XCTAssertTrue(url.isIPv6)
XCTAssertNil(url.baseDomain)
XCTAssertEqual(url.normalizedHost!, "[::1]")
}
func testDomainURL() {
let urls = [
]
urls.forEach { XCTAssertEqual(URL(string: $0.0)!.domainURL.absoluteString, $0.1) }
}
func testnormalizedHostAndPath() {
let goodurls = [
("https://www.example.com/index.html", "example.com/index.html"),
("https://mail.example.com/index.html", "mail.example.com/index.html"),
("https://mail.example.co.uk/index.html", "mail.example.co.uk/index.html"),
("https://m.example.co.uk/index.html", "example.co.uk/index.html")
]
let badurls = [
]
goodurls.forEach { XCTAssertEqual(URL(string: $0.0)!.normalizedHostAndPath, $0.1) }
badurls.forEach { XCTAssertNil(URL(string: $0)!.normalizedHostAndPath) }
}
func testGetSubdomainAndHost() {
let testCases = [
("https://www.google.com", (nil, "google.com")),
("https://blog.engineering.company.com", ("blog.engineering.", "blog.engineering.company.com")),
("https://long-extended-subdomain-name-containing-many-letters-and-dashes.badssl.com", ("long-extended-subdomain-name-containing-many-letters-and-dashes.", "long-extended-subdomain-name-containing-many-letters-and-dashes.badssl.com")),
("http://com:org@m.canadacomputers.co.uk", (nil, "canadacomputers.co.uk")),
("nothing", (nil, "nothing")),
("https://super-long-url-with-dashes-and-things.badssl.com/xyz-something", ("super-long-url-with-dashes-and-things.", "super-long-url-with-dashes-and-things.badssl.com")),
("https://accounts.firefox.com", ("accounts.", "accounts.firefox.com")),
("http://username:password@subdomain.example.com:8080", ("subdomain.", "subdomain.example.com")),
("https://example.com:8080#fragment", (nil, "example.com")),
("http://username:password@subdomain.example.com:8080#fragment", ("subdomain.", "subdomain.example.com")),
("https://www.amazon.co.uk", (nil, "amazon.co.uk")),
("https://mobile.co.uk", (nil, "mobile.co.uk"))
]
for testCase in testCases {
let (urlString, expected) = testCase
let result = URL.getSubdomainAndHost(from: urlString)
XCTAssertEqual(result.subdomain, expected.0, "Unexpected subdomain for URL: \(urlString)")
XCTAssertEqual(result.normalizedHost, expected.1, "Unexpected normalized host for URL: \(urlString)")
}
}
func testShortDisplayString() {
let urls = [
]
urls.forEach { XCTAssertEqual(URL(string: $0.0)!.shortDisplayString, $0.1) }
}
func testBlobURLGetHost() {
let url = URL(string: "blob:https://example.blob.com")!
XCTAssertNil(url.host)
}
func testRemoveBlobFromUrl_WithBlob() {
let url = URL(string: "blob:https://example.blob.com")!
let originalURL = url.removeBlobFromUrl()
XCTAssertEqual(originalURL, URL(string: "https://example.blob.com"))
}
func testRemoveBlobFromUrl_WithoutBlob() {
let url = URL(string: "https://example.blob.com")!
let originalURL = url.removeBlobFromUrl()
XCTAssertEqual(originalURL, URL(string: "https://example.blob.com"))
}
// MARK: getQuery tests
func testGetQueryWhenTheresParametersThenGetQueryReturnsTheRightParameters() {
let url = URL(string: "http://example.com/path?a=1&b=2&c=3")!
let urlParams = url.getQuery()
let expectedParams = ["a": "1", "b": "2", "c": "3"]
XCTAssertEqual(urlParams["a"], expectedParams["a"])
XCTAssertEqual(urlParams["b"], expectedParams["b"])
XCTAssertEqual(urlParams["c"], expectedParams["c"])
}
func testGetQueryWhenPercentEncodedParamsThenGetQueryReturnsTheRightParameters() {
let url = URL(string: "http://example.com/path?a=%20")!
let urlParams = url.getQuery()
XCTAssertEqual(urlParams["a"], "%20")
}
// MARK: isWebPage
func testIsWebPageGivenReaderModeURLThenisWebPage() {
XCTAssertTrue(url.isWebPage())
}
func testIsWebPageGivenSessionRestoreHTMLThenisWebPage() {
XCTAssertTrue(url.isWebPage())
}
func testIsWebPageGivenDataSessionRestoreThenisWebPage() {
XCTAssertTrue(url.isWebPage())
}
func testIsWebPageGivenAboutURLThenisNotWebPage() {
let url = URL(string: "about://google.com")!
XCTAssertFalse(url.isWebPage())
}
func testIsWebPageGivenTelURLThenisNotWebPage() {
let url = URL(string: "tel:6044044004")!
XCTAssertFalse(url.isWebPage())
}
func testIsWebPageGivenLocalHostURLThenisNotWebPage() {
let url = URL(string: "hax://localhost:\(webServerPort)/about")!
XCTAssertFalse(url.isWebPage())
}
// MARK: Host port
func testHostPortGivenExampleHostThenIsEqual() {
let givenURL = URL(string: "https://www.example.com")!
XCTAssertEqual(givenURL.hostPort, "www.example.com")
}
func testHostPortGivenUserPassHostThenIsEqual() {
let givenURL = URL(string: "https://user:pass@www.example.com")!
XCTAssertEqual(givenURL.hostPort, "www.example.com")
}
func testHostPortGivenLocalHostThenIsEqual() {
let givenURL = URL(string: "http://localhost:6000/blah")!
XCTAssertEqual(givenURL.hostPort, "localhost:6000")
}
func testHostPortGivenBlahURLThenIsNil() {
let givenURL = URL(string: "blah")!
XCTAssertNil(givenURL.hostPort)
}
func testHostPortGivenEmptyURLThenIsNil() {
let givenURL = URL(string: "http://")!
XCTAssertNil(givenURL.hostPort)
}
// MARK: Origin
func testOriginGivenExampleIndexURLThenOriginIsExample() {
let givenURL = URL(string: "https://www.example.com/index.html")!
XCTAssertEqual(givenURL.origin, "https://www.example.com")
let badurls = [
]
badurls.forEach { XCTAssertNil(URL(string: $0)!.origin) }
}
func testOriginGivenUserPassURLThenOriginIsFoo() {
let givenURL = URL(string: "https://user:pass@m.foo.com/bar/baz?noo=abc#123")!
XCTAssertEqual(givenURL.origin, "https://m.foo.com")
}
func testOriginGivenDataURLThenOriginIsNil() {
let givenURL = URL(string: "data://google.com")!
XCTAssertNil(givenURL.origin)
}
}