TEST_F(GarbageCollectorTest, Unschedule)
{
GarbageCollector gc;
// Attempt to unschedule a file that is not scheduled.
AWAIT_ASSERT_EQ(false, gc.unschedule("bogus"));
// Make some temporary files to gc.
const string& file1 = "file1";
const string& file2 = "file2";
const string& file3 = "file3";
ASSERT_SOME(os::touch(file1));
ASSERT_SOME(os::touch(file2));
ASSERT_SOME(os::touch(file3));
ASSERT_TRUE(os::exists(file1));
ASSERT_TRUE(os::exists(file2));
ASSERT_TRUE(os::exists(file3));
Clock::pause();
// Schedule the gc operations.
Future<Nothing> schedule1 = gc.schedule(Seconds(10), file1);
Future<Nothing> schedule2 = gc.schedule(Seconds(10), file2);
Future<Nothing> schedule3 = gc.schedule(Seconds(10), file3);
// Unschedule each operation.
AWAIT_ASSERT_EQ(true, gc.unschedule(file2));
AWAIT_ASSERT_EQ(true, gc.unschedule(file3));
AWAIT_ASSERT_EQ(true, gc.unschedule(file1));
// Advance the clock to ensure nothing was GCed.
Clock::advance(Seconds(10));
Clock::settle();
// The unscheduling will have discarded the GC futures.
AWAIT_DISCARDED(schedule1);
AWAIT_DISCARDED(schedule2);
AWAIT_DISCARDED(schedule3);
EXPECT_TRUE(os::exists(file1));
EXPECT_TRUE(os::exists(file2));
EXPECT_TRUE(os::exists(file3));
Clock::resume();
}
TEST_F(GarbageCollectorTest, Prune)
{
GarbageCollector gc;
// Make some temporary files to prune.
const string& file1 = "file1";
const string& file2 = "file2";
const string& file3 = "file3";
const string& file4 = "file4";
ASSERT_SOME(os::touch(file1));
ASSERT_SOME(os::touch(file2));
ASSERT_SOME(os::touch(file3));
ASSERT_SOME(os::touch(file4));
ASSERT_TRUE(os::exists(file1));
ASSERT_TRUE(os::exists(file2));
ASSERT_TRUE(os::exists(file3));
ASSERT_TRUE(os::exists(file4));
Clock::pause();
Future<Nothing> schedule1 = gc.schedule(Seconds(10), file1);
Future<Nothing> schedule2 = gc.schedule(Seconds(10), file2);
Future<Nothing> schedule3 = gc.schedule(Seconds(15), file3);
Future<Nothing> schedule4 = gc.schedule(Seconds(15), file4);
AWAIT_ASSERT_EQ(true, gc.unschedule(file3));
AWAIT_DISCARDED(schedule3);
// Prune file1 and file2.
gc.prune(Seconds(10));
AWAIT_READY(schedule1);
AWAIT_READY(schedule2);
ASSERT_TRUE(schedule4.isPending());
// Both file1 and file2 will have been removed.
EXPECT_FALSE(os::exists(file1));
EXPECT_FALSE(os::exists(file2));
EXPECT_TRUE(os::exists(file3));
EXPECT_TRUE(os::exists(file4));
// Prune file4.
gc.prune(Seconds(15));
AWAIT_READY(schedule4);
EXPECT_FALSE(os::exists(file4));
Clock::resume();
}
