[tests] Check that BaseVM tracks snapshot totals

canonical · Nov 24, 2023 · f583acf · f583acf
1 parent cf21c8b
commit f583acf
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/tests/test_base_virtual_machine.cpp b/tests/test_base_virtual_machine.cpp
@@ -346,4 +346,36 @@ TEST(BaseVMSnapshots, countsCurrentSnapshots)
     vm.take_snapshot(specs, "s4", "");
     EXPECT_EQ(vm.get_num_snapshots(), 1);
 }
+
+TEST(BaseVMSnapshots, countsTotalSnapshots)
+{
+    const mp::VMSpecs specs{};
+    MockBaseVirtualMachine vm{"mock-vm"};
+    EXPECT_EQ(vm.get_num_snapshots(), 0);
+
+    auto snapshot = std::make_shared<MockSnapshot>();
+    EXPECT_CALL(vm, make_specific_snapshot(_, _, _, _)).WillRepeatedly(Return(snapshot));
+    EXPECT_CALL(*snapshot, capture).Times(AnyNumber());
+    EXPECT_CALL(*snapshot, erase).Times(AnyNumber());
+
+    vm.take_snapshot(specs, "s1", "");
+    vm.take_snapshot(specs, "s2", "");
+    vm.take_snapshot(specs, "s3", "");
+    EXPECT_EQ(vm.get_snapshot_count(), 3);
+
+    vm.take_snapshot(specs, "s4", "");
+    vm.take_snapshot(specs, "s5", "");
+    EXPECT_EQ(vm.get_snapshot_count(), 5);
+
+    vm.delete_snapshot("s1");
+    vm.delete_snapshot("s2");
+    EXPECT_EQ(vm.get_snapshot_count(), 5);
+
+    vm.delete_snapshot("s4");
+    EXPECT_EQ(vm.get_snapshot_count(), 5);
+
+    vm.take_snapshot(specs, "s6", "");
+    EXPECT_EQ(vm.get_snapshot_count(), 6);
+}
+
 } // namespace