Added more test cases for LogoLoader

ghostery · Sep 21, 2018 · 234c277e41e8e5845602037fbdb097ca8e4b465e · 234c277
1 parent 3455058
commit 234c277e41e8e5845602037fbdb097ca8e4b465e
Unified Split

Showing with 31 additions and 0 deletions.

+31 −0 UnitTests/LogoLoaderTests.swift
diff --git a/UnitTests/LogoLoaderTests.swift b/UnitTests/LogoLoaderTests.swift
@@ -24,17 +24,36 @@ class LogoLoaderTests: XCTestCase {
     func testDomains() {
         XCTAssertTrue(LogoLoader.domainName("http://abc.go.com") == "go", "domain for http://abc.go.com != go")
         XCTAssertTrue(LogoLoader.domainName("http://m.facebook.com") == "facebook", "domain for http://m.facebook.com != facebook")
+        XCTAssertTrue(LogoLoader.domainName("http://de-de.facebook.com") == "facebook", "domain for http://de-de.facebook.com != facebook")
         XCTAssertTrue(LogoLoader.domainName("http://direct.gov.uk") == "direct", "domain for http://direct.gov.uk != direct")
         XCTAssertTrue(LogoLoader.domainName("http://images.google.com") == "google", "domain for http://images.google.com != google")
         XCTAssertTrue(LogoLoader.domainName("http://images.google.com.eg") == "google", "domain for http://images.google.com.eg != google")
         XCTAssertTrue(LogoLoader.domainName("http://bbc.co.uk") == "bbc", "domain for http://bbc.co.uk != bbc")
+        XCTAssertTrue(LogoLoader.domainName("http://en.wikipedia.org") == "wikipedia", "domain for en.wikipedia.org != wikipedia")
+        XCTAssertTrue(LogoLoader.domainName("http://amp.welt.de") == "welt", "domain for http://amp.welt.de != welt")
+        XCTAssertTrue(LogoLoader.domainName("https://www.fci.cu.edu.eg") == "cu", "domain for https://www.fci.cu.edu.eg != cu")
     }
 
     func testFacebookMobileLogoDetails() {
         let logoDetails = LogoLoader.fetchLogoDetails("http://m.facebook.com")
         XCTAssertTrue(logoDetails.hostName == "facebook", "Host name for http://m.facebook.com != facebook")
     }
 
+    func testFacebookDELogoDetails() {
+        let logoDetails = LogoLoader.fetchLogoDetails("http://de-de.facebook.com")
+        XCTAssertTrue(logoDetails.hostName == "facebook", "Host name for http://de-de.facebook.com != facebook")
+    }
+
+    func testFciLogoDetails() {
+        let logoDetails = LogoLoader.fetchLogoDetails("https://www.fci.cu.edu.eg")
+        XCTAssertTrue(logoDetails.hostName == "cu", "Host name for https://www.fci.cu.edu.eg != cu")
+    }
+
+    func testDirectGovLogoDetails() {
+        let logoDetails = LogoLoader.fetchLogoDetails("http://direct.gov.uk")
+        XCTAssertTrue(logoDetails.hostName == "direct", "Host name for http://direct.gov.uk != direct")
+    }
+
     func testGoogleImagesLogoDetails() {
         let logoDetails = LogoLoader.fetchLogoDetails("http://images.google.com")
         XCTAssertTrue(logoDetails.hostName == "images.google", "Host name for http://images.google.com != images.google")
@@ -51,4 +70,16 @@ class LogoLoaderTests: XCTestCase {
         XCTAssertTrue(logoDetails.hostName == "bbc", "Host name for http://bbc.co.uk != bbc")
     }
 
+
+    func testAmpWeltLogoDetails() {
+        let logoDetails = LogoLoader.fetchLogoDetails("http://amp.welt.de")
+        XCTAssertTrue(logoDetails.hostName == "welt", "Host name for http://amp.welt.de != welt")
+    }
+
+
+    func testWikipediaLogoDetails() {
+        let logoDetails = LogoLoader.fetchLogoDetails("http://en.wikipedia.org")
+        XCTAssertTrue(logoDetails.hostName == "wikipedia", "Host name for http://en.wikipedia.org != wikipedia")
+    }
+
 }