Adding unit tests for server