Move tests into mod and add UDP smoke test.