bpf/tests: Use migrate disable instead of preempt disable