Add server_test.go