mirror of https://github.com/oxen-io/session-ios
				
				
				
			
			You cannot select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
	
	
		
			62 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Swift
		
	
			
		
		
	
	
			62 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Swift
		
	
//
 | 
						|
//  Copyright (c) 2017 Open Whisper Systems. All rights reserved.
 | 
						|
//
 | 
						|
 | 
						|
import XCTest
 | 
						|
@testable import Signal
 | 
						|
 | 
						|
class ImageCacheTest: XCTestCase {
 | 
						|
 | 
						|
    var imageCache: ImageCache!
 | 
						|
 | 
						|
    let firstVariation = UIImage()
 | 
						|
    let secondVariation = UIImage()
 | 
						|
    let otherImage = UIImage()
 | 
						|
 | 
						|
    let cacheKey1 = "cache-key-1" as NSString
 | 
						|
    let cacheKey2 = "cache-key-2" as NSString
 | 
						|
 | 
						|
    override func setUp() {
 | 
						|
        super.setUp()
 | 
						|
         self.imageCache = ImageCache()
 | 
						|
        imageCache.setImage(firstVariation, forKey:cacheKey1, diameter:100)
 | 
						|
        imageCache.setImage(secondVariation, forKey:cacheKey1, diameter:200)
 | 
						|
        imageCache.setImage(otherImage, forKey:cacheKey2, diameter:100)
 | 
						|
    }
 | 
						|
 | 
						|
    override func tearDown() {
 | 
						|
        // Put teardown code here. This method is called after the invocation of each test method in the class.
 | 
						|
        super.tearDown()
 | 
						|
    }
 | 
						|
 | 
						|
    func testSetGet() {
 | 
						|
        XCTAssertEqual(firstVariation, imageCache.image(forKey:cacheKey1, diameter: 100))
 | 
						|
        XCTAssertEqual(secondVariation, imageCache.image(forKey:cacheKey1, diameter: 200))
 | 
						|
        XCTAssertNotEqual(secondVariation, imageCache.image(forKey:cacheKey1, diameter: 100))
 | 
						|
        XCTAssertEqual(otherImage, imageCache.image(forKey:cacheKey2, diameter: 100))
 | 
						|
        XCTAssertNil(imageCache.image(forKey:cacheKey2, diameter: 200))
 | 
						|
    }
 | 
						|
 | 
						|
    func testRemoveAllForKey() {
 | 
						|
        // sanity check
 | 
						|
        XCTAssertEqual(firstVariation, imageCache.image(forKey:cacheKey1, diameter: 100))
 | 
						|
        XCTAssertEqual(otherImage, imageCache.image(forKey:cacheKey2, diameter: 100))
 | 
						|
 | 
						|
        imageCache.removeAllImages(forKey:cacheKey1)
 | 
						|
 | 
						|
        XCTAssertNil(imageCache.image(forKey:cacheKey1, diameter: 100))
 | 
						|
        XCTAssertNil(imageCache.image(forKey:cacheKey1, diameter: 200))
 | 
						|
        XCTAssertEqual(otherImage, imageCache.image(forKey:cacheKey2, diameter: 100))
 | 
						|
    }
 | 
						|
 | 
						|
    func testRemoveAll() {
 | 
						|
        XCTAssertEqual(firstVariation, imageCache.image(forKey:cacheKey1, diameter: 100))
 | 
						|
 | 
						|
        imageCache.removeAllImages()
 | 
						|
 | 
						|
        XCTAssertNil(imageCache.image(forKey:cacheKey1, diameter: 100))
 | 
						|
        XCTAssertNil(imageCache.image(forKey:cacheKey1, diameter: 200))
 | 
						|
        XCTAssertNil(imageCache.image(forKey:cacheKey2, diameter: 100))
 | 
						|
    }
 | 
						|
}
 |