diff --git a/test/firmware_unittest.hpp b/test/firmware_unittest.hpp
index d077818..947c2ba 100644
--- a/test/firmware_unittest.hpp
+++ b/test/firmware_unittest.hpp
@@ -103,6 +103,16 @@
         expectedState(FirmwareBlobHandler::UpdateState::updateStarted);
     }
 
+    void getToUpdateCompleted(UpdateStatus result)
+    {
+        getToUpdateStarted();
+        EXPECT_CALL(*updateMockPtr, status()).WillOnce(Return(result));
+
+        blobs::BlobMeta meta;
+        EXPECT_TRUE(handler->stat(session, &meta));
+        expectedState(FirmwareBlobHandler::UpdateState::updateCompleted);
+    }
+
     ImageHandlerMock imageMock;
     std::vector<HandlerPack> blobs;
     std::vector<DataHandlerPack> data = {