如何正确地对根据MediaType返回不同内容的路由进行单元测试？

import org.junit.Test;
import org.springframework.http.MediaType;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;

public class MyControllerTest {

    @Test
    public void testJsonRoute() {
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setMethod("GET");
        request.setRequestURI("/route");
        request.addHeader("Accept", MediaType.APPLICATION_JSON_VALUE);

        MockHttpServletResponse response = new MockHttpServletResponse();

        MyController controller = new MyController();
        controller.handleRoute(request, response);

        // 验证响应结果
        String expectedContent = "{\"message\":\"Hello, JSON!\"}";
        assertEquals(expectedContent, response.getContentAsString());
        assertEquals(MediaType.APPLICATION_JSON_VALUE, response.getContentType());
    }

    @Test
    public void testXmlRoute() {
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setMethod("GET");
        request.setRequestURI("/route");
        request.addHeader("Accept", MediaType.APPLICATION_XML_VALUE);

        MockHttpServletResponse response = new MockHttpServletResponse();

        MyController controller = new MyController();
        controller.handleRoute(request, response);

        // 验证响应结果
        String expectedContent = "<message>Hello, XML!</message>";
        assertEquals(expectedContent, response.getContentAsString());
        assertEquals(MediaType.APPLICATION_XML_VALUE, response.getContentType());
    }
}