mirror of https://github.com/oxen-io/session-ios
Michael Kirk
8 years ago
parent
b579ea5915
commit
f6720f9afa
@ -0,0 +1,53 @@
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
import Foundation
|
||||
import UIKit
|
||||
|
||||
class ImageCacheRecord: NSObject {
|
||||
var variations: [CGFloat: UIImage]
|
||||
init(variations: [CGFloat: UIImage]) {
|
||||
self.variations = variations
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* A two dimensional hash, allowing you to store variations under a single key.
|
||||
* This is useful because we generate multiple diameters of an image, but when we
|
||||
* want to clear out the images for a key we want to clear out *all* variations.
|
||||
*/
|
||||
@objc
|
||||
class ImageCache: NSObject {
|
||||
|
||||
let backingCache: NSCache<AnyObject, ImageCacheRecord>
|
||||
|
||||
override init() {
|
||||
self.backingCache = NSCache()
|
||||
}
|
||||
|
||||
func image(forKey key: AnyObject, diameter: CGFloat) -> UIImage? {
|
||||
guard let record = backingCache.object(forKey: key) else {
|
||||
return nil
|
||||
}
|
||||
return record.variations[diameter]
|
||||
}
|
||||
|
||||
func setImage(_ image: UIImage, forKey key: AnyObject, diameter: CGFloat) {
|
||||
if let existingRecord = backingCache.object(forKey: key) {
|
||||
existingRecord.variations[diameter] = image
|
||||
backingCache.setObject(existingRecord, forKey: key)
|
||||
} else {
|
||||
let newRecord = ImageCacheRecord(variations: [diameter: image])
|
||||
backingCache.setObject(newRecord, forKey: key)
|
||||
}
|
||||
}
|
||||
|
||||
func removeAllImages() {
|
||||
backingCache.removeAllObjects()
|
||||
}
|
||||
|
||||
func removeAllImages(forKey key: AnyObject) {
|
||||
backingCache.removeObject(forKey: key)
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,60 @@
|
||||
//
|
||||
// Copyright (c) 2017 Open Whisper Systems. All rights reserved.
|
||||
//
|
||||
|
||||
import XCTest
|
||||
|
||||
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))
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue