Move TransmissionWait test to separate package