Add security code to transaction commit

uphold · Feb 20, 2018 · 772618a · 772618a
1 parent f9e1265
commit 772618a
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 0 deletions.
diff --git a/Source/Model/Transaction/TransactionCommitRequest.swift b/Source/Model/Transaction/TransactionCommitRequest.swift
@@ -10,6 +10,9 @@ open class TransactionCommitRequest: Mappable {
     /// The transaction message.
     public private(set) final var message: String?
 
+    /// The transaction security code.
+    public private(set) final var securityCode: String?
+
     /**
       Constructor.
 
@@ -28,6 +31,15 @@ open class TransactionCommitRequest: Mappable {
         self.message = message
     }
 
+    /**
+     Constructor.
+
+     - parameter securityCode: The transaction security code.
+     */
+    public init(securityCode: String) {
+        self.securityCode = securityCode
+    }
+
     /**
       Constructor.
 
@@ -39,6 +51,17 @@ open class TransactionCommitRequest: Mappable {
         self.message = message
     }
 
+    /**
+      Constructor.
+
+      - parameter message: The transaction message.
+      - parameter securityCode: The transaction security code.
+    */
+    public init(message: String?, securityCode: String) {
+        self.message = message
+        self.securityCode = securityCode
+    }
+
     // MARK: Required by the ObjectMapper.
 
     /**

diff --git a/Tests/IntegrationTests/ModelTests/TransactionTest.swift b/Tests/IntegrationTests/ModelTests/TransactionTest.swift
@@ -262,6 +262,23 @@ class TransactionTest: UpholdTestCase {
         wait()
     }
 
+    func testCommitWithTransactionCommitRequestWithSecurityCodeShouldReturnTheTransaction() {
+        let testExpectation = expectation(description: "Transaction test: commit transaction.")
+
+        let transaction: Transaction = Fixtures.loadTransaction(fields: ["transactionId": "foobar", "transactionStatus": "pending"])
+        transaction.adapter = MockRestAdapter(body: Mapper().toJSONString(transaction)!)
+
+        transaction.commit(transactionCommit: TransactionCommitRequest(securityCode: "1234")).then { (transaction: Transaction) -> Void in
+            XCTAssertEqual(transaction.id, "foobar", "Failed: Wrong transaction object.")
+
+            testExpectation.fulfill()
+        }.catch(execute: { (_: Error) in
+            XCTFail("User test: create transaction transfer error.")
+        })
+
+        wait()
+    }
+
     func testCommitWithTransactionCommitRequestWithBeneficiaryAndMessageShouldReturnTheTransaction() {
         let testExpectation = expectation(description: "Transaction test: commit transaction.")
 
@@ -279,6 +296,23 @@ class TransactionTest: UpholdTestCase {
         wait()
     }
 
+    func testCommitWithTransactionCommitRequestWithMessageAndSecurityCodeShouldReturnTheTransaction() {
+        let testExpectation = expectation(description: "Transaction test: commit transaction.")
+
+        let transaction: Transaction = Fixtures.loadTransaction(fields: ["transactionId": "foobar", "transactionStatus": "pending"])
+        transaction.adapter = MockRestAdapter(body: Mapper().toJSONString(transaction)!)
+
+        transaction.commit(transactionCommit: TransactionCommitRequest(message: "foobar", securityCode: "1234")).then { (transaction: Transaction) -> Void in
+            XCTAssertEqual(transaction.id, "foobar", "Failed: Wrong transaction object.")
+
+            testExpectation.fulfill()
+        }.catch(execute: { (_: Error) in
+            XCTFail("User test: create transaction transfer error.")
+        })
+
+        wait()
+    }
+
     func testCommitShouldReturnUnexpectedResponseErrorIfAccountIdIsNil() {
         let testExpectation = expectation(description: "Transaction test: commit transaction.")