Full coverage after refactor

coddin-web · Apr 6, 2022 · b2de9ae · b2de9ae
1 parent 390d3ee
commit b2de9ae
Showing 1 changed file with 40 additions and 6 deletions.
diff --git a/src/Test/Unit/Service/CustomerQuoteTest.php b/src/Test/Unit/Service/CustomerQuoteTest.php
@@ -103,15 +103,32 @@ public function testGetQuoteGuestWithExistingCart(): void
             ->method('isLoggedIn')
             ->willReturn(false);
 
-        $quoteMock = $this->createQuote();
-        $quoteMock
+        $quoteMockNotFullyLoaded = $this->createQuote();
+        $quoteMockNotFullyLoaded
             ->expects(self::exactly(2))
             ->method('getId')
-            ->willReturn(1);
+            ->willReturn(null, 456);
 
         $this->checkoutSession
             ->expects(self::once())
             ->method('getQuote')
+            ->willReturn($quoteMockNotFullyLoaded);
+
+        $this->quoteRepository
+            ->expects(self::once())
+            ->method('save')
+            ->with($quoteMockNotFullyLoaded);
+
+        $quoteMock = $this->createQuote();
+        $quoteMock
+            ->expects(self::once())
+            ->method('getId')
+            ->willReturn(456);
+
+        $this->quoteRepository
+            ->expects(self::once())
+            ->method('get')
+            ->with(456)
             ->willReturn($quoteMock);
 
         $quote = $customerQuote->getQuote();
@@ -177,15 +194,32 @@ public function testGetQuoteCustomerWithExistingCart(): void
             ->method('isLoggedIn')
             ->willReturn(true);
 
-        $quoteMock = $this->createQuote();
-        $quoteMock
+        $quoteMockNotFullyLoaded = $this->createQuote();
+        $quoteMockNotFullyLoaded
             ->expects(self::exactly(2))
             ->method('getId')
-            ->willReturn(1);
+            ->willReturn(null, 456);
 
         $this->checkoutSession
             ->expects(self::once())
             ->method('getQuote')
+            ->willReturn($quoteMockNotFullyLoaded);
+
+        $this->quoteRepository
+            ->expects(self::once())
+            ->method('save')
+            ->with($quoteMockNotFullyLoaded);
+
+        $quoteMock = $this->createQuote();
+        $quoteMock
+            ->expects(self::once())
+            ->method('getId')
+            ->willReturn(456);
+
+        $this->quoteRepository
+            ->expects(self::once())
+            ->method('get')
+            ->with(456)
             ->willReturn($quoteMock);
 
         $quote = $customerQuote->getQuote();