SiteSettingsRemote: add setFeature for updating feature switch in WC settings.

Author: jaclync

Modules/Sources/Networking/Remote/SiteSettingsRemote.swift

@@ -114,16 +114,49 @@ public class SiteSettingsRemote: Remote {
         let mapper = SiteSettingMapper(siteID: siteID, settingsGroup: .advanced)
         let response = try await enqueue(request, mapper: mapper)
         switch response.value {
-        case "yes":
+        case Constants.featureEnabledValue:
             return true
-        case "no":
+        case Constants.featureDisabledValue:
+            return false
+        default:
+            throw SiteSettingsRemoteError.invalidResponse
+        }
+    }
+
+    /// Enables or disables a specific feature in the site WC settings.
+    ///
+    /// - Parameters:
+    ///   - siteID: Site for which we'll update the feature status.
+    ///   - feature: The feature to enable or disable.
+    ///   - enabled: Whether the feature should be enabled (true) or disabled (false).
+    /// - Returns: A boolean indicating the updated feature status.
+    /// - Throws: Error if the request fails.
+    ///
+    public func setFeature(for siteID: Int64, feature: SiteSettingsFeature, enabled: Bool) async throws -> Bool {
+        let value = enabled ? Constants.featureEnabledValue : Constants.featureDisabledValue
+        let parameters: [String: Any] = [Constants.valueParameter: value]
+        let path = Constants.siteSettingsPath + Constants.advancedSettingsGroup + "/woocommerce_feature_\(feature.rawValue)_enabled"
+        let request = JetpackRequest(wooApiVersion: .mark3,
+                                     method: .put,
+                                     siteID: siteID,
+                                     path: path,
+                                     parameters: parameters,
+                                     availableAsRESTRequest: true)
+        let mapper = SiteSettingMapper(siteID: siteID, settingsGroup: .advanced)
+        let response = try await enqueue(request, mapper: mapper)
+        switch response.value {
+        case Constants.featureEnabledValue:
+            return true
+        case Constants.featureDisabledValue:
             return false
         default:
             throw SiteSettingsRemoteError.invalidResponse
         }
     }
 }
 
+extension SiteSettingsRemote: SiteSettingsRemoteProtocol {}
+
 public extension SiteSettingsFeature {
     var rawValue: String {
         switch self {
@@ -142,6 +175,8 @@ private extension SiteSettingsRemote {
         static let productSettingsGroup: String   = "products"
         static let advancedSettingsGroup: String   = "advanced"
         static let valueParameter: String = "value"
+        static let featureEnabledValue: String = "yes"
+        static let featureDisabledValue: String = "no"
     }
 }
 

Modules/Tests/NetworkingTests/Remote/SiteSettingsRemoteTests.swift

@@ -228,4 +228,48 @@ final class SiteSettingsRemoteTests: XCTestCase {
             XCTAssertEqual(error as? NetworkError, .unacceptableStatusCode(statusCode: 500))
         }
     }
+
+    // MARK: - `setFeature`
+
+    func test_setFeature_enables_feature_when_enabled_is_true() async throws {
+        // Given
+        let remote = SiteSettingsRemote(network: network)
+        network.simulateResponse(requestUrlSuffix: "settings/advanced/woocommerce_feature_point_of_sale_enabled",
+                                 filename: "settings-advanced-feature-pos-enabled")
+
+        // When
+        let isEnabled = try await remote.setFeature(for: sampleSiteID, feature: .pointOfSale, enabled: true)
+
+        // Then
+        XCTAssertTrue(isEnabled)
+    }
+
+    func test_setFeature_disables_feature_when_enabled_is_false() async throws {
+        // Given
+        let remote = SiteSettingsRemote(network: network)
+        network.simulateResponse(requestUrlSuffix: "settings/advanced/woocommerce_feature_point_of_sale_enabled",
+                                 filename: "settings-advanced-feature-pos-disabled")
+
+        // When
+        let isEnabled = try await remote.setFeature(for: sampleSiteID, feature: .pointOfSale, enabled: false)
+
+        // Then
+        XCTAssertFalse(isEnabled)
+    }
+
+    func test_setFeature_throws_error_when_network_fails() async {
+        // Given
+        let remote = SiteSettingsRemote(network: network)
+        let error = NetworkError.unacceptableStatusCode(statusCode: 500)
+        network.simulateError(requestUrlSuffix: "settings/advanced/woocommerce_feature_point_of_sale_enabled",
+                              error: error)
+
+        // When/Then
+        do {
+            _ = try await remote.setFeature(for: sampleSiteID, feature: .pointOfSale, enabled: true)
+            XCTFail("Expected error to be thrown")
+        } catch {
+            XCTAssertEqual(error as? NetworkError, .unacceptableStatusCode(statusCode: 500))
+        }
+    }
 }