Add a test for the server class and make it work.