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, [])