basecontrollertest.java

anewssystem新闻发布系统集成使用了spring hibernate freemarker EXTJS等开源框架 可以作为学习参考

        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/removeAll.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/removeAll.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameterValues("itemlist"))
            .andReturn(new String[] {"1"});
        dao.removeById(1L);
        expect(session.getAttribute("messages")).andReturn(null);
        session.setAttribute(same("messages"), isA(List.class));

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public void testRemove() throws Exception {
        EntityDao<Object> dao = createMock(EntityDao.class);
        HttpServletRequest request = createMock(HttpServletRequest.class);
        HttpServletResponse response = createMock(HttpServletResponse.class);
        HttpSession session = createMock(HttpSession.class);
        ServletContext context = createMock(ServletContext.class);

        expect(request.getSession()).andReturn(session).anyTimes();
        expect(session.getServletContext()).andReturn(context).anyTimes();
        expect(request.getAttribute("javax.servlet.include.context_path"))
            .andReturn("/").anyTimes();
        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/remove.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/remove.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameter("id")).andReturn("1").anyTimes();
        dao.removeById(1L);
        expect(session.getAttribute("messages")).andReturn(null);
        session.setAttribute(same("messages"), isA(List.class));

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public void testRemoveAll2() throws Exception {
        EntityDao<Object> dao = createMock(EntityDao.class);
        HttpServletRequest request = createMock(HttpServletRequest.class);
        HttpServletResponse response = createMock(HttpServletResponse.class);
        HttpSession session = createMock(HttpSession.class);
        ServletContext context = createMock(ServletContext.class);

        expect(request.getSession()).andReturn(session).anyTimes();
        expect(session.getServletContext()).andReturn(context).anyTimes();
        expect(request.getAttribute("javax.servlet.include.context_path"))
            .andReturn("/").anyTimes();
        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/removeAll.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/removeAll.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameterValues("itemlist")).andReturn(null);
        dao.removeById(1L);
        expect(session.getAttribute("messages")).andReturn(null);
        session.setAttribute(same("messages"), isA(List.class));

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public void testRemove2() throws Exception {
        EntityDao<Object> dao = createMock(EntityDao.class);
        HttpServletRequest request = createMock(HttpServletRequest.class);
        HttpServletResponse response = createMock(HttpServletResponse.class);
        HttpSession session = createMock(HttpSession.class);
        ServletContext context = createMock(ServletContext.class);

        expect(request.getSession()).andReturn(session).anyTimes();
        expect(session.getServletContext()).andReturn(context).anyTimes();
        expect(request.getAttribute("javax.servlet.include.context_path"))
            .andReturn("/").anyTimes();
        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/remove.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/remove.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameter("id")).andReturn(null).anyTimes();
        dao.removeById(-1L);
        expect(session.getAttribute("messages")).andReturn(null);
        session.setAttribute(same("messages"), isA(List.class));

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public void testRemoveAll3() throws Exception {
        EntityDao<Object> dao = createMock(EntityDao.class);
        HttpServletRequest request = createMock(HttpServletRequest.class);
        HttpServletResponse response = createMock(HttpServletResponse.class);
        HttpSession session = createMock(HttpSession.class);
        ServletContext context = createMock(ServletContext.class);

        expect(request.getSession()).andReturn(session).anyTimes();
        expect(session.getServletContext()).andReturn(context).anyTimes();
        expect(request.getAttribute("javax.servlet.include.context_path"))
            .andReturn("/").anyTimes();
        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/removeAll.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/removeAll.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameterValues("itemlist"))
            .andReturn(new String[] {"1"});
        dao.removeById(1L);
        expectLastCall()
            .andThrow(new DataIntegrityViolationException("test"));
        expect(session.getAttribute("messages")).andReturn(null).anyTimes();
        session.setAttribute(same("messages"), isA(List.class));
        expectLastCall().anyTimes();
        expect(dao.get(1L)).andReturn("test test");

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public void testRemove3() throws Exception {
        EntityDao<Object> dao = createMock(EntityDao.class);
        HttpServletRequest request = createMock(HttpServletRequest.class);
        HttpServletResponse response = createMock(HttpServletResponse.class);
        HttpSession session = createMock(HttpSession.class);
        ServletContext context = createMock(ServletContext.class);

        expect(request.getSession()).andReturn(session).anyTimes();
        expect(session.getServletContext()).andReturn(context).anyTimes();
        expect(request.getAttribute("javax.servlet.include.context_path"))
            .andReturn("/").anyTimes();
        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/remove.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/remove.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameter("id")).andReturn("1").anyTimes();
        dao.removeById(1L);
        expectLastCall()
            .andThrow(new DataIntegrityViolationException("test"));
        expect(session.getAttribute("messages")).andReturn(null).anyTimes();
        session.setAttribute(same("messages"), isA(List.class));
        expectLastCall().anyTimes();
        expect(dao.get(1L)).andReturn("test test");

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public void testUploadImageToFile() throws Exception {
        controller = new TestBaseController();

        EntityDao<Object> dao = createMock(EntityDao.class);
        HttpServletRequest request = createMock(HttpServletRequest.class);
        HttpServletResponse response = createMock(HttpServletResponse.class);
        HttpSession session = createMock(HttpSession.class);
        ServletContext context = createMock(ServletContext.class);

        expect(request.getSession()).andReturn(session).anyTimes();
        expect(session.getServletContext()).andReturn(context).anyTimes();
        expect(request.getAttribute("javax.servlet.include.context_path"))
            .andReturn("/").anyTimes();
        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/upload.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/upload.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameter("id")).andReturn("1").anyTimes();
        dao.removeById(1L);
        expectLastCall()
            .andThrow(new DataIntegrityViolationException("test"));
        expect(session.getAttribute("messages")).andReturn(null).anyTimes();
        session.setAttribute(same("messages"), isA(List.class));
        expectLastCall().anyTimes();
        expect(dao.get(1L)).andReturn("test test");
        response.sendError(404);

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public void testGetStrParam() throws Exception {
        controller = new TestBaseController();

        EntityDao<Object> dao = createMock(EntityDao.class);
        HttpServletRequest request = createMock(HttpServletRequest.class);
        HttpServletResponse response = createMock(HttpServletResponse.class);
        HttpSession session = createMock(HttpSession.class);
        ServletContext context = createMock(ServletContext.class);

        expect(request.getSession()).andReturn(session).anyTimes();
        expect(session.getServletContext()).andReturn(context).anyTimes();
        expect(request.getAttribute("javax.servlet.include.context_path"))
            .andReturn("/").anyTimes();
        expect(request.getContextPath()).andReturn("/").anyTimes();
        expect(request.getAttribute("javax.servlet.include.request_uri"))
            .andReturn("/getStr.htm").anyTimes();
        expect(request.getRequestURI()).andReturn("/index.htm").anyTimes();
        expect(request.getAttribute("javax.servlet.include.servlet_path"))
            .andReturn("/getStr.htm").anyTimes();
        expect(request.getServletPath()).andReturn("/index.htm").anyTimes();
        expect(context.getInitParameter(
                "extremecomponentsPreferencesLocation")).andReturn("/");
        expect(request.getParameter("id")).andReturn("1").anyTimes();
        dao.removeById(1L);
        expectLastCall()
            .andThrow(new DataIntegrityViolationException("test"));
        expect(session.getAttribute("messages")).andReturn(null).anyTimes();
        session.setAttribute(same("messages"), isA(List.class));
        expectLastCall().anyTimes();
        expect(dao.get(1L)).andReturn("test test");
        response.sendError(404);
        expect(request.getParameter("test")).andReturn(null);

        replay(dao);
        replay(request);
        replay(response);
        replay(session);
        replay(context);

        controller.setEntityDao(dao);
        //controller.setControllerClassNameHandlerMapping
        controller.handleRequestInternal(request, response);
        verify();
    }

    public class TestBaseController extends BaseController<Object, EntityDao<Object>> {
        public void upload() throws Exception {
            uploadImageToFile("/", "test");
        }

        public void getStr() throws Exception {
            getStrParam("test", null);
        }
    }
}