def test_delete_mvs_seen_error(self): mock_pm = MagicMock(powermanager) mock_pm._mvs_seen = { 'test_name_0': '0', 'test_name_1': '1', 'test_name_2': '2'} powermanager._delete_mvs_seen(mock_pm, 'test_name_1') self.assertIn( "Error trying to save INDIGO orchestrator plugin data", self.log.getvalue())
def test_delete_mvs_seen(self): mock_pm = MagicMock(powermanager) mock_pm._mvs_seen = { 'test_name_0': '0', 'test_name_1': '1', 'test_name_2': '2'} mock_pm._db = MagicMock() mock_pm._db.sql_query.return_value = (True, "", "") powermanager._delete_mvs_seen(mock_pm, 'test_name_1') self.assertEquals( mock_pm._mvs_seen, {'test_name_0': '0', 'test_name_2': '2'}) self.assertEquals(mock_pm._db.sql_query.call_args_list, [call("DELETE FROM orchestrator_vms WHERE node_name = 'test_name_1'", True)])