Add server unit tests