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)