marshroute

1
import UIKit
2

3
final class TestablePeekAndPopSourceViewController: UIViewController {
4
    var shouldStartSpying = false
5
    
6
    var registerForPreviewingCalledCount = 0 
7
    
8
    override func registerForPreviewing(
9
        with delegate: UIViewControllerPreviewingDelegate,
10
        sourceView: UIView)
11
        -> UIViewControllerPreviewing
12
    {
13
        if shouldStartSpying {
14
            registerForPreviewingCalledCount += 1
15
        }
16
        
17
        return super.registerForPreviewing(
18
            with: delegate,
19
            sourceView: sourceView
20
        )
21
    }
22
    
23
    var unregisterForPreviewingCalledCount = 0
24
    
25
    override func unregisterForPreviewing(
26
        withContext previewing: UIViewControllerPreviewing)
27
    {
28
        if shouldStartSpying {
29
            unregisterForPreviewingCalledCount += 1
30
        }
31
        
32
        super.unregisterForPreviewing(
33
            withContext: previewing
34
        )
35
    }
36
}
37