Add more test cases to server_test.go