Я тоже столкнулся с этой проблемой; и основываясь на том факте, что дерево не восстановилось визуально при срабатывании точки останова onDragComplete, я поместил в метод обработчика onDragComplete следующее:
callLater(function():void{
treeGrid.selectedItem = draggedItem;
});
Этот обработанный выбранный элемент был выделен.