MAGETWO-15874: Edit button is present when order is already captured

Author: Kostiantyn Poida

app/code/Magento/Sales/Model/Order.php

@@ -801,6 +801,10 @@ public function canEdit()
             return false;
         }
 
+        if ($this->hasInvoices()) {
+            return false;
+        }
+
         if (!$this->getPayment()->getMethodInstance()->canEdit()) {
             return false;
         }
@@ -1820,6 +1824,18 @@ public function getInvoiceCollection()
         return $this->_invoices;
     }
 
+    /**
+     * Set order invoices collection
+     *
+     * @param InvoiceCollection $invoices
+     * @return $this
+     */
+    public function setInvoiceCollection(InvoiceCollection $invoices)
+    {
+        $this->_invoices = $invoices;
+        return $this;
+    }
+
     /**
      * Retrieve order shipments collection
      *

dev/tests/unit/testsuite/Magento/Sales/Model/OrderTest.php

@@ -191,6 +191,23 @@ public function testCanCancelIsPaymentReview()
         $this->assertFalse($this->order->canCancel());
     }
 
+    public function testCanEditIfHasInvoices()
+    {
+        $invoiceCollection = $this->getMockBuilder('Magento\Sales\Model\Resource\Order\Invoice\Collection')
+            ->disableOriginalConstructor()
+            ->setMethods(['count'])
+            ->getMock();
+
+        $invoiceCollection->expects($this->once())
+            ->method('count')
+            ->willReturn(2);
+
+        $this->order->setInvoiceCollection($invoiceCollection);
+        $this->order->setState(\Magento\Sales\Model\Order::STATE_PROCESSING);
+
+        $this->assertFalse($this->order->canEdit());
+    }
+
     public function testCanCancelCanReviewPayment()
     {
         $paymentMock = $this->getMockBuilder('Magento\Sales\Model\Resource\Order\Payment')