UC in splitio_ios (#74)

Author: gthea

splitio_ios/example/ios/SplitTests/SplitMethodParserTests.swift

@@ -297,6 +297,28 @@ class SplitMethodParserTests: XCTestCase {
         XCTAssert(providerHelper.splitClientConfigValue?.impressionListener != nil)
         XCTAssert(providerHelper.splitClientConfigValue?.streamingEnabled == false)
     }
+
+    func testGetUserConsent() {
+        methodParser?.onMethodCall(methodName: "getUserConsent", arguments: [], result: { (_: Any?) in })
+
+        guard let wrapper = splitWrapper as? SplitWrapperStub else {
+            XCTFail()
+            return
+        }
+
+        XCTAssert(wrapper.userConsent == "unknown")
+    }
+
+    func testSetUserConsent() {
+        methodParser?.onMethodCall(methodName: "setUserConsent", arguments: [], result: { (_: Any?) in })
+
+        guard let wrapper = splitWrapper as? SplitWrapperStub else {
+            XCTFail()
+            return
+        }
+
+        XCTAssert(wrapper.userConsent == "declined")
+    }
 }
 
 class SplitWrapperStub: SplitWrapper {
@@ -316,6 +338,7 @@ class SplitWrapperStub: SplitWrapper {
     var attributeNameValue: String = ""
     var splitsCalled = false
     var splitNamesCalled = false
+    var userConsent = "unknown"
 
     func getClient(matchingKey: String, bucketingKey: String?) -> SplitClient? {
         matchingKeyValue = matchingKey
@@ -440,6 +463,18 @@ class SplitWrapperStub: SplitWrapper {
         splitNameValue = splitName
         return nil
     }
+
+    func getUserConsent() -> String {
+        return userConsent
+    }
+
+    func setUserConsent(enabled: Bool) {
+        if (enabled) {
+            userConsent = "granted"
+        } else {
+            userConsent = "declined"
+        }
+    }
 }
 
 class SplitProviderHelperStub: SplitProviderHelper {

splitio_ios/example/ios/SplitTests/SplitTests.swift

@@ -181,6 +181,26 @@ class SplitTests: XCTestCase {
         let split = splitWrapper.split(splitName: "my-split")
         XCTAssert(manager.splitNameValue == "my-split")
     }
+
+    func testGetUserConsent() {
+        let manager = SplitManagerStub()
+        let factoryProvider = SplitFactoryProviderStub(manager: manager)
+        splitWrapper = DefaultSplitWrapper(splitFactoryProvider: factoryProvider)
+        let userConsent = splitWrapper.getUserConsent()
+        XCTAssert(userConsent == "unknown")
+    }
+
+    func testSetUserConsent() {
+        let manager = SplitManagerStub()
+        let factoryProvider = SplitFactoryProviderStub(manager: manager)
+        splitWrapper = DefaultSplitWrapper(splitFactoryProvider: factoryProvider)
+        splitWrapper.setUserConsent(enabled: true)
+        let grantedUserConsent = splitWrapper.getUserConsent()
+        splitWrapper.setUserConsent(enabled: false)
+        let declinedUserConsent = splitWrapper.getUserConsent()
+        XCTAssert(grantedUserConsent == "granted")
+        XCTAssert(declinedUserConsent == "declined")
+    }
 }
 
 class SplitFactoryProviderStub: SplitFactoryProvider {

splitio_ios/ios/Classes/Constants.swift

@@ -24,6 +24,8 @@ enum Method: String {
     case splits = "splits"
     case split = "split"
     case impressionLog = "impressionLog"
+    case getUserConsent = "getUserConsent"
+    case setUserConsent = "setUserConsent"
 }
 
 enum Argument: String {

splitio_ios/ios/Classes/SplitMethodParser.swift

@@ -133,6 +133,12 @@ class DefaultSplitMethodParser: SplitMethodParser {
         case .split:
             result(SplitView.asMap(splitView: splitWrapper?.split(splitName: argumentParser.getStringArgument(argumentName: .splitName, arguments: arguments) ?? "")))
             break
+        case .getUserConsent:
+            result(splitWrapper?.getUserConsent())
+            break
+        case .setUserConsent:
+            splitWrapper?.setUserConsent(enabled: argumentParser.getBooleanArgument(argumentName: .value, arguments: arguments))
+            result(nil)
         default:
             result(FlutterMethodNotImplemented)
             break

splitio_ios/ios/Classes/SplitWrapper.swift

@@ -16,6 +16,10 @@ protocol SplitWrapper: EvaluationWrapper, AttributesWrapper {
     func splits() -> [SplitView]
 
     func split(splitName: String) -> SplitView?
+
+    func getUserConsent() -> String
+
+    func setUserConsent(enabled: Bool)
 }
 
 protocol EvaluationWrapper {
@@ -227,4 +231,26 @@ class DefaultSplitWrapper: SplitWrapper {
             return nil
         }
     }
+
+    func getUserConsent() -> String {
+        if let splitFactory = splitFactory {
+            let userConsent: UserConsent = splitFactory.userConsent
+            
+            if (userConsent == .granted) {
+                return "granted"
+            } else if (userConsent == .declined) {
+                return "declined"
+            } else {
+                return "unknown"
+            }
+        } else {
+            return "unknown"
+        }
+    }
+
+    func setUserConsent(enabled: Bool) {
+        if let splitFactory = splitFactory {
+            splitFactory.setUserConsent(enabled: enabled)
+        }
+    }
 }