Add tests with gen_server