Update unit tests for new drag callback firing pattern