Add tests for server.rb