Add UIPickerView data source extensions tests

qazyn951230 · Jul 14, 2017 · de8de1d · de8de1d
1 parent 3127ef4
commit de8de1d
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 1 deletion.
diff --git a/Tests/RxCocoaTests/DelegateProxyTest+UIKit.swift b/Tests/RxCocoaTests/DelegateProxyTest+UIKit.swift
@@ -74,6 +74,10 @@ extension DelegateProxyTest {
     func test_UIPickerViewExtension() {
         performDelegateTest(UIPickerViewSubclass(frame: CGRect.zero))
     }
+
+    func test_UIPickerViewDataSourceExtension() {
+        performDelegateTest(UIPickerViewSubclass2(frame: CGRect.zero))
+    }
 }
 #endif
 
@@ -390,6 +394,37 @@ final class UIPickerViewSubclass
                                                                 onProxyForObject: self)
     }
 }
+
+final class ExtendPickerViewDataSourceProxy
+    : RxPickerViewDataSourceProxy
+    , TestDelegateProtocol {
+    weak fileprivate(set) var control: UIPickerViewSubclass2?
+
+    required init(parentObject: AnyObject) {
+        self.control = (parentObject as! UIPickerViewSubclass2)
+        super.init(parentObject: parentObject)
+    }
+}
+
+final class UIPickerViewSubclass2: UIPickerView, TestDelegateControl {
+    override func createRxDataSourceProxy() -> RxPickerViewDataSourceProxy {
+        return ExtendPickerViewDataSourceProxy(parentObject: self)
+    }
+
+    func doThatTest(_ value: Int) {
+        if dataSource != nil {
+            (dataSource as! TestDelegateProtocol).testEventHappened?(value)
+        }
+    }
+
+    var delegateProxy: DelegateProxy {
+        return self.rx.dataSource
+    }
+
+    func setMineForwardDelegate(_ testDelegate: TestDelegateProtocol) -> Disposable {
+        return RxPickerViewDataSourceProxy.installForwardDelegate(testDelegate, retainDelegate: false, onProxyForObject: self)
+    }
+}
 
 final class ExtendWebViewDelegateProxy
     : RxWebViewDelegateProxy

diff --git a/Tests/RxCocoaTests/DelegateProxyTest.swift b/Tests/RxCocoaTests/DelegateProxyTest.swift
@@ -508,7 +508,8 @@ extension MockTestDelegateProtocol
     : UICollectionViewDataSource
     , UIScrollViewDelegate
     , UITableViewDataSource
-    , UITableViewDelegate {
+    , UITableViewDelegate
+    , UIPickerViewDataSource {
 
     func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
         fatalError()
@@ -525,5 +526,13 @@ extension MockTestDelegateProtocol
     func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
         fatalError()
     }
+
+    func numberOfComponents(in pickerView: UIPickerView) -> Int {
+        fatalError()
+    }
+
+    func pickerView(_ pickerView: UIPickerView, numberOfRowsInComponent component: Int) -> Int {
+        fatalError()
+    }
 }
 #endif