Add basic WFI test