def test_delete_snapshot_files_internal(self, mock_os, mock_conn):
mock_os.path.isfile.return_value = True
mock_os.remove.return_value = True
snapshot_xml = '''<domainsnapshot>
<memory snapshot="internal"/>
<domain>
<cpu mode='host-model'/>
</domain>
</domainsnapshot>'''
snapshot = mock.Mock()
snapshot.getXMLDesc.return_value = snapshot_xml
dd = DevopsDriver()
dd._delete_snapshot_files(snapshot)
self.assertEqual(mock_os.remove.called, False)
def test_delete_snapshot_files(self, mock_os, mock_conn):
mock_os.path.isfile.return_value = True
mock_os.remove.return_value = True
memory_file = factories.fuzzy_string('/path/to/')
snapshot_xml = '''<domainsnapshot>
<memory file="{0}" snapshot="external"/>
<domain>
<cpu mode='host-model'/>
</domain>
</domainsnapshot>'''.format(memory_file)
snapshot = mock.Mock()
snapshot.getXMLDesc.return_value = snapshot_xml
dd = DevopsDriver()
dd._delete_snapshot_files(snapshot)
mock_os.remove.assert_called_with(memory_file)
