def test_create_rserver_3(self, mock_func): """Exception""" self.rs['parent_id'] = None cmd.create_rserver(self.ctx, self.rs) rollback_fn = self.ctx.add_rollback.call_args[0][0] rollback_fn(False) self.assertTrue(self.ctx.device.delete_real_server.called) self.assertTrue(mock_func.called)
def test_create_rserver_1(self, mock_func): """ parent_id is None """ self.rs['parent_id'] = None cmd.create_rserver(self.ctx, self.rs) self.assertTrue(self.ctx.device.create_real_server.called) self.assertTrue(mock_func.called) mock_func.assert_called_once_with(self.ctx.conf, self.rs['id'], self.rs)
def test_create_rserver_2(self, mock_func): """ parent_id is not None or 0, no exception """ self.rs['parent_id'] = 1 cmd.create_rserver(self.ctx, self.rs) self.assertFalse(self.ctx.device.create_real_server.called) self.assertFalse(mock_func.called)