Track Memory Leaks Unit Test with XCTestCase Extension

test.swift

import XCTest

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