5976 loadWCPayCharge function for CoreData access

Author: joshheald

Storage/Storage/Tools/StorageType+Extensions.swift

@@ -584,6 +584,13 @@ public extension StorageType {
         return firstObject(ofType: PaymentGatewayAccount.self, matching: predicate)
     }
 
+    /// Returns a charge with a specified `siteID` and `chargeID`
+    ///
+    func loadWCPayCharge(siteID: Int64, chargeID: String) -> WCPayCharge? {
+        let predicate = \WCPayCharge.siteID == siteID && \WCPayCharge.id == chargeID
+        return firstObject(ofType: WCPayCharge.self, matching: predicate)
+    }
+
     // MARK: - System plugins
 
     /// Returns all stored system plugins for a provided `siteID`.

Storage/StorageTests/Tools/StorageTypeExtensionsTests.swift

@@ -1170,4 +1170,21 @@ final class StorageTypeExtensionsTests: XCTestCase {
         // Then
         XCTAssertEqual(foundSystemPlugin, systemPlugin2)
     }
+
+    func test_load_WCPayCharge_by_siteID_and_chargeID() throws {
+        // Given
+        let charge1 = storage.insertNewObject(ofType: WCPayCharge.self)
+        charge1.id = "ch_1"
+        charge1.siteID = sampleSiteID
+
+        let charge2 = storage.insertNewObject(ofType: WCPayCharge.self)
+        charge2.id = "ch_2"
+        charge2.siteID = sampleSiteID
+
+        // When
+        let foundCharge = try XCTUnwrap(storage.loadWCPayCharge(siteID: sampleSiteID, chargeID: "ch_2"))
+
+        // Then
+        XCTAssertEqual(foundCharge, charge2)
+    }
 }