Test

src/BackupMasterRecoveryTest.cc

 /* Copyright (c) 2012 Stanford University
  *
  * Permission to use, copy, modify, and distribute this software for any
  * purpose with or without fee is hereby granted, provided that the above
  * copyright notice and this permission notice appear in all copies.
  *
  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR(S) DISCLAIM ALL WARRANTIES
  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL AUTHORS BE LIABLE FOR
  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
@@ skipping 229 matching lines @@
 }
 
 TEST_F(BackupMasterRecoveryTest, getRecoverySegment) {
     mockMetadata(88); // secondary
     mockMetadata(89, true, true); // primary
     recovery->testingExtractDigest = &mockExtractDigest;
     recovery->testingSkipBuild = true;
     recovery->start(frames, NULL, NULL);
     recovery->setPartitionsAndSchedule(partitions);
 
-    Status status = recovery->getRecoverySegment(456, 89, 0, NULL, NULL);
-    EXPECT_EQ(STATUS_RETRY, status);
+    EXPECT_THROW(recovery->getRecoverySegment(456, 89, 0, NULL, NULL),
+                 RetryException);
 
     taskQueue.performTask();
 
-    status = recovery->getRecoverySegment(456, 88, 0, NULL, NULL);
+    Status status = recovery->getRecoverySegment(456, 88, 0, NULL, NULL);
     EXPECT_EQ(STATUS_OK, status);
     Buffer buffer;
     buffer.append("important", 10);
     ASSERT_TRUE(recovery->replicas[1].recoverySegments[0].append(
         LOG_ENTRY_TYPE_OBJ, buffer));
     buffer.reset();
     Segment::Certificate certificate;
     memset(&certificate, 0xff, sizeof(certificate));
     status = recovery->getRecoverySegment(456, 88, 0, &buffer, &certificate);
     EXPECT_EQ(STATUS_OK, status);
@@ skipping 93 matching lines @@
     EXPECT_TRUE(StringUtil::startsWith(TestLog::get(),
         "buildRecoverySegments: Couldn't build recovery segments for "
         "<99.0,88>: RAMCloud::SegmentIteratorException: cannot iterate: "
         "corrupt segment thrown "));
     EXPECT_TRUE(recovery->replicas.at(0).recoveryException);
     EXPECT_FALSE(recovery->replicas.at(0).recoverySegments);
     EXPECT_TRUE(recovery->replicas.at(0).built);
 }
 
 } // namespace RAMCloud