From 0ab036a59f7b7e526891021a962296be1eee4ace Mon Sep 17 00:00:00 2001
From: nikita tarasov <nikita@mygento.ru>
Date: Wed, 15 Nov 2017 17:28:35 +0300
Subject: [PATCH] Transaction Type Verification

---
 Model/Payment/Transaction.php | 46 +++++++++++++++++++++++++++++++++++
 etc/di.xml                    |  4 +--
 2 files changed, 48 insertions(+), 2 deletions(-)
 create mode 100644 Model/Payment/Transaction.php

diff --git a/Model/Payment/Transaction.php b/Model/Payment/Transaction.php
new file mode 100644
index 0000000..91eeba2
--- /dev/null
+++ b/Model/Payment/Transaction.php
@@ -0,0 +1,46 @@
+<?php
+/**
+ * @author Mygento
+ * @copyright See COPYING.txt for license details.
+ * @package Mygento_Base
+ */
+
+namespace Mygento\Base\Model\Payment;
+
+use Magento\Framework\Exception\LocalizedException;
+
+class Transaction extends \Magento\Sales\Model\Order\Payment\Transaction
+{
+    const TYPE_FISCAL = 'fiscal_receipt';
+    const TYPE_FISCAL_REFUND = 'fiscal_receipt_refund';
+
+    /**
+     * Check whether specified or set transaction type is supported
+     *
+     * @param string $txnType
+     * @return void
+     * @throws \Magento\Framework\Exception\LocalizedException
+     * @SuppressWarnings(PHPMD.CyclomaticComplexity)
+     */
+    protected function _verifyTxnType($txnType = null)
+    {
+        if (null === $txnType) {
+            $txnType = $this->getTxnType();
+        }
+        switch ($txnType) {
+            case self::TYPE_PAYMENT:
+            case self::TYPE_ORDER:
+            case self::TYPE_AUTH:
+            case self::TYPE_CAPTURE:
+            case self::TYPE_VOID:
+            case self::TYPE_REFUND:
+            case self::TYPE_FISCAL:
+            case self::TYPE_FISCAL_REFUND:
+                break;
+            default:
+                throw new LocalizedException(
+                    __('We found an unsupported transaction type "%1".', $txnType)
+                );
+        }
+    }
+}
diff --git a/etc/di.xml b/etc/di.xml
index e84f93d..318a422 100644
--- a/etc/di.xml
+++ b/etc/di.xml
@@ -6,9 +6,9 @@
  * @package Mygento_Base
  */
 -->
-<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
-        xsi:noNamespaceSchemaLocation="urn:magento:framework:ObjectManager/etc/config.xsd">
+<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:ObjectManager/etc/config.xsd">
     <type name="Magento\Sales\Model\Order\Payment\Transaction">
         <plugin name="mygento_base_transaction_receipt" type="Mygento\Base\Plugin\Transaction"/>
     </type>
+    <preference for="Magento\Sales\Model\Order\Payment\Transaction" type="Mygento\Base\Model\Payment\Transaction" />
 </config>