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
import XCTest
@testable import SiteImageView
final class SiteImageViewTests: XCTestCase {
private var imageFetcher: MockSiteImageHandler!
override func setUp() {
super.setUp()
self.imageFetcher = MockSiteImageHandler()
}
override func tearDown() {
super.tearDown()
self.imageFetcher = nil
}
func testFaviconSetup() {
let expectation = expectation(description: "Completed image setup")
let viewModel = FaviconImageViewModel(siteURLString: url,
faviconCornerRadius: 8)
let subject = FaviconImageView(frame: .zero, imageFetcher: imageFetcher) {
expectation.fulfill()
XCTAssertEqual(self.imageFetcher.getImageCalled, 1, "get image should be called")
}
subject.setFavicon(viewModel)
waitForExpectations(timeout: 0.1)
}
func testHeroImageSetup() {
let expectation = expectation(description: "Completed image setup")
let viewModel = DefaultHeroImageViewModel(urlStringRequest: url,
generalCornerRadius: 8,
faviconCornerRadius: 4,
faviconBorderWidth: 0.5,
heroImageSize: CGSize(),
fallbackFaviconSize: CGSize())
let subject = HeroImageView(frame: .zero, imageFetcher: imageFetcher) {
expectation.fulfill()
XCTAssertEqual(self.imageFetcher.getImageCalled, 1, "get image should be called")
}
subject.setHeroImage(viewModel)
waitForExpectations(timeout: 0.1)
}
func testCanMakeRequest_firstTime_true() {
let subject = FaviconImageView(frame: .zero, imageFetcher: imageFetcher) {}
let canMakeRequestFirstTime = subject.canMakeRequest(with: url)
XCTAssertTrue(canMakeRequestFirstTime)
}
func testCanMakeRequest_secondTime_false() {
let subject = FaviconImageView(frame: .zero, imageFetcher: imageFetcher) {}
_ = subject.canMakeRequest(with: url)
let canMakeRequestSecondTime = subject.canMakeRequest(with: url)
XCTAssertFalse(canMakeRequestSecondTime)
}
func testCanMakeRequest_secondTime_newURL_true() {
let subject = FaviconImageView(frame: .zero, imageFetcher: imageFetcher) {}
_ = subject.canMakeRequest(with: url)
let canMakeRequestSecondTime = subject.canMakeRequest(with: newURL)
XCTAssertTrue(canMakeRequestSecondTime)
}
}
class MockSiteImageHandler: SiteImageHandler {
var image = UIImage()
var siteURL: URL?
var faviconURL: URL?
var getImageCalled = 0
var cacheFaviconURLCalled = 0
var clearAllCachesCalled = 0
func getImage(model: SiteImageModel) async -> UIImage {
getImageCalled += 1
return image
}
func cacheFaviconURL(siteURL: URL, faviconURL: URL) {
self.siteURL = siteURL
self.faviconURL = faviconURL
cacheFaviconURLCalled += 1
}
func clearAllCaches() {
clearAllCachesCalled += 1
}
}