XCTestCase+trackForMemoryLeak

XCTestCase+trackForMemoryLeak.swift

import XCTest

extension XCTestCase {
    func assertDeallocated(
        _ instance: AnyObject,
        file: StaticString = #filePath,
        line: UInt = #line
    ) {
        addTeardownBlock { [weak instance] in
            XCTAssertNil(
                instance,
                "Instance not deallocated. Potential memory leak.",
                file: file,
                line: line
            )
        }
    }
}