XCTestCase+teardown

XCTestCase+teardown.swift

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

private func getSUT(file: StaticString = #filePath, line: UInt = #line) -> Something {
    let sut = Something()
    trackForMemoryLeaks(instance: sut, file: file, line: line)
    return sut
}