improving tests to include udp and tcp