Add more basic test for udp