def testGetPartitionIdentifiers(self, mock_volumesystem): """Test the GetPartitionIdentifiers function.""" mediator = UnattendedVolumeScannerMediator() type(mock_volumesystem.return_value).number_of_volumes = 2 volume_system = dfvfs_volume_system.VolumeSystem() volume_identifiers = ['p1', 'p2'] result = mediator.GetPartitionIdentifiers(volume_system, volume_identifiers) self.assertEqual(result, volume_identifiers)
def testGetVSSStoreIdentifiers(self, mock_volumesystem): """Test the GetVSSStoreIdentifiers function. VSS support is not yet implemented. """ mediator = UnattendedVolumeScannerMediator() type(mock_volumesystem.return_value).number_of_volumes = 3 volume_system = dfvfs_volume_system.VolumeSystem() volume_identifiers = ['vss1', 'vss2', 'vss3'] result = mediator.GetVSSStoreIdentifiers(volume_system, volume_identifiers) self.assertEqual(result, [])