diff --git a/Tests/FlareTests/UnitTests/TestHelpers/Mocks/StoreTransactionMock.swift b/Tests/FlareTests/UnitTests/TestHelpers/Mocks/StoreTransactionMock.swift index 144e17cdb..cab029486 100644 --- a/Tests/FlareTests/UnitTests/TestHelpers/Mocks/StoreTransactionMock.swift +++ b/Tests/FlareTests/UnitTests/TestHelpers/Mocks/StoreTransactionMock.swift @@ -106,4 +106,13 @@ final class StoreTransactionMock: IStoreTransaction, @unchecked Sendable { invokedCurrencyGetterCount += 1 return stubbedCurrency } + + var invokedOriginalID = false + var invokedOriginalIDCount = 0 + var stubbedOriginalID: UInt64? + var originalID: UInt64? { + invokedOriginalID = true + invokedOriginalIDCount += 1 + return stubbedOriginalID + } } diff --git a/Tests/FlareTests/UnitTests/TestHelpers/Stubs/StoreTransactionStub.swift b/Tests/FlareTests/UnitTests/TestHelpers/Stubs/StoreTransactionStub.swift index 6d2e9cb44..e187a4cf9 100644 --- a/Tests/FlareTests/UnitTests/TestHelpers/Stubs/StoreTransactionStub.swift +++ b/Tests/FlareTests/UnitTests/TestHelpers/Stubs/StoreTransactionStub.swift @@ -66,4 +66,9 @@ final class StoreTransactionStub: IStoreTransaction, @unchecked Sendable { var currency: String? { stubbedCurrency } + + var stubbedOriginalID: UInt64? + var originalID: UInt64? { + stubbedOriginalID + } }