commit 497317167feb5c480c0bd54b3e622153a3dece7b
author Patrick Venture <venture@google.com> Mon Jun 17 10:04:02 2019 -0700
committer Patrick Venture <venture@google.com> Mon Jun 17 10:05:45 2019 -0700
tree 5ba4fdebf0b2199bfc2a869ceea93148e91616b2
parent d574102016b41802c32a61c4255a88584705c2c6

bmc: abort from updateStarted

Signed-off-by: Patrick Venture <venture@google.com>
Change-Id: Ib5cd094415af4e0c6b2bb3067fadef6b1a5af757

test/firmware_state_updatestarted_unittest.cpp

diff --git a/test/firmware_state_updatestarted_unittest.cpp b/test/firmware_state_updatestarted_unittest.cpp
index 420d7a7..51a3c20 100644
--- a/test/firmware_state_updatestarted_unittest.cpp
+++ b/test/firmware_state_updatestarted_unittest.cpp
@@ -212,8 +212,22 @@
 }
 
 /*
- * TODO: close(session) - this will abort.
+ * close(session) - this will abort.
  */
+TEST_F(FirmwareHandlerUpdateStartedTest, CloseOnUpdateDuringUpdateAbortsProcess)
+{
+    getToUpdateStarted();
+    EXPECT_CALL(*updateMockPtr, abort()).Times(1);
+
+    EXPECT_TRUE(handler->close(session));
+
+    std::vector<std::string> expectedBlobs = {staticLayoutBlobId, hashBlobId};
+
+    EXPECT_THAT(handler->getBlobIds(),
+                UnorderedElementsAreArray(expectedBlobs));
+
+    expectedState(FirmwareBlobHandler::UpdateState::notYetStarted);
+}
 
 } // namespace
 } // namespace ipmi_flash