ServletManagementServletTest.java

/*
 ###############################################################################
 #                                                                             #
 #    Copyright (C) 2011-2016 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) );

	}
}