enable UDP tests in socket tests

Somehow during my last round of socket work I used a different define in the test runner / source files. I've corrected that, and made a few minor fixes for the tests when UDP mode is enabled.

* check correct define for UDP mode
* minor fixes related to UDP mode