Unit tests

Author: brunoroeder

app/code/Magento/Quote/Test/Unit/Observer/Frontend/Quote/Address/CollectTotalsObserverTest.php

@@ -314,6 +314,43 @@ public function testDispatchWithCustomerCountryInEU()
         $this->model->execute($this->observerMock);
     }
 
+    public function testDispatchWithAddressCustomerVatIdAndCountryId()
+    {
+        $customerCountryCode = "BE";
+        $customerVat = "123123123";
+        $defaultShipping = 1;
+
+        $customerAddress = $this->createMock(\Magento\Quote\Model\Quote\Address::class);
+        $customerAddress->expects($this->any())
+            ->method("getVatId")
+            ->willReturn($customerVat);
+
+        $customerAddress->expects($this->any())
+            ->method("getCountryId")
+            ->willReturn($customerCountryCode);
+
+        $this->addressRepository->expects($this->once())
+            ->method("getById")
+            ->with($defaultShipping)
+            ->willReturn($customerAddress);
+
+        $this->customerMock->expects($this->atLeastOnce())
+            ->method("getDefaultShipping")
+            ->willReturn($defaultShipping);
+
+        $this->vatValidatorMock->expects($this->once())
+            ->method('isEnabled')
+            ->with($this->quoteAddressMock, $this->storeId)
+            ->will($this->returnValue(true));
+
+        $this->customerVatMock->expects($this->once())
+            ->method('isCountryInEU')
+            ->with($customerCountryCode)
+            ->willReturn(true);
+
+        $this->model->execute($this->observerMock);
+    }
+
     public function testDispatchWithEmptyShippingAddress()
     {
         $customerCountryCode = "DE";