Java tutorial
/* ############################################################################### # # # Copyright (C) 2011-2015 OpenMEAP, Inc. # # Credits to Jonathan Schang & Rob Thacher # # # # Released under the LGPLv3 # # # # OpenMEAP is free software: you can redistribute it and/or modify # # it under the terms of the GNU Lesser General Public License as published # # by the Free Software Foundation, either version 3 of the License, or # # (at your option) any later version. # # # # OpenMEAP is distributed in the hope that it will be useful, # # but WITHOUT ANY WARRANTY; without even the implied warranty of # # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # # GNU Lesser General Public License for more details. # # # # You should have received a copy of the GNU Lesser General Public License # # along with OpenMEAP. If not, see <http://www.gnu.org/licenses/>. # # # ############################################################################### */ package com.openmeap.services; import java.util.HashMap; import java.util.Map; import java.util.UUID; import com.openmeap.thirdparty.org.json.me.JSONObject; import org.junit.AfterClass; import org.junit.Assert; import org.junit.BeforeClass; import org.junit.Test; import org.springframework.mock.web.MockHttpServletRequest; import org.springframework.mock.web.MockHttpServletResponse; import com.openmeap.constants.UrlParamConstants; import com.openmeap.digest.DigestInputStreamFactory; import com.openmeap.json.JSONObjectBuilder; import com.openmeap.model.ModelManager; import com.openmeap.model.ModelTestUtils; import com.openmeap.model.dto.Application; import com.openmeap.model.dto.GlobalSettings; import com.openmeap.model.event.ModelEntityEventAction; import com.openmeap.model.event.handler.ModelServiceRefreshHandler; import com.openmeap.services.dto.Result; import com.openmeap.util.AuthTokenProvider; public class ServletManagementServletTest { static ModelManager modelManager = null; @BeforeClass static public void beforeClass() { if (modelManager == null) { ModelTestUtils.resetTestDb(); ModelTestUtils.createModel(null); modelManager = ModelTestUtils.createModelManager(); } } @AfterClass static public void afterClass() { ModelTestUtils.resetTestDb(); } class Request extends MockHttpServletRequest { private Map<String, String[]> parameters = new HashMap<String, String[]>(); public Map<String, String[]> getParameterMap() { return parameters; } }; @Test public void testRefreshApplication() throws Exception { MockHttpServletRequest request = new Request(); MockHttpServletResponse response = new MockHttpServletResponse(); String randomUuid = UUID.randomUUID().toString(); GlobalSettings settings = modelManager.getGlobalSettings(); ///////////////// // validate that finding the application, modifying it, and then finding it again // will return an object with the same modifications. Application app = modelManager.getModelService().findByPrimaryKey(Application.class, 1L); app.setName(randomUuid); Assert.assertTrue(modelManager.getModelService().findByPrimaryKey(Application.class, 1L).getName() .equals(randomUuid)); modelManager.refresh(app, null); app = modelManager.getModelService().findByPrimaryKey(Application.class, 1L); Assert.assertTrue(!modelManager.getModelService().findByPrimaryKey(Application.class, 1L).getName() .equals(randomUuid)); ServiceManagementServlet servlet = new ServiceManagementServlet(); servlet.setModelManager(modelManager); servlet.setModelServiceRefreshHandler(new ModelServiceRefreshHandler()); servlet.getModelServiceRefreshHandler().setModelManager(modelManager); //////////////////// // validate the happy path of providing all the required information String authSalt = servlet.getAuthSalt(); String authToken = AuthTokenProvider.newAuthToken(authSalt); request.setParameter(UrlParamConstants.REFRESH_TYPE, "Application"); request.setParameter(UrlParamConstants.REFRESH_OBJ_PKID, "1"); request.setParameter(UrlParamConstants.AUTH_TOKEN, authToken); request.setParameter(UrlParamConstants.ACTION, ModelEntityEventAction.MODEL_REFRESH.getActionName()); servlet.service(request, response); String contentString = response.getContentAsString(); JSONObjectBuilder job = new JSONObjectBuilder(); Result result = (Result) job.fromJSON(new JSONObject(contentString), new Result()); Assert.assertTrue(result.getStatus().equals(Result.Status.SUCCESS)); Assert.assertTrue(!modelManager.getModelService().findByPrimaryKey(Application.class, 1L).getName() .equals(randomUuid)); //////////////////// // validate that failing to provide auth token fails to refresh cache app = modelManager.getModelService().findByPrimaryKey(Application.class, 1L); app.setName(randomUuid); response = new MockHttpServletResponse(); request.removeParameter(UrlParamConstants.AUTH_TOKEN); request.setParameter(UrlParamConstants.ACTION, ModelEntityEventAction.MODEL_REFRESH.getActionName()); request.setParameter(UrlParamConstants.REFRESH_TYPE, "Application"); request.setParameter(UrlParamConstants.REFRESH_OBJ_PKID, "1"); servlet.service(request, response); contentString = response.getContentAsString(); result = (Result) job.fromJSON(new JSONObject(contentString), new Result()); Assert.assertTrue(result.getStatus().equals(Result.Status.FAILURE)); Assert.assertTrue(modelManager.getModelService().findByPrimaryKey(Application.class, 1L).getName() .equals(randomUuid)); } }