Author: mwringe Date: 2010-08-16 18:44:50 -0400 (Mon, 16 Aug 2010) New Revision: 3837 Added: components/wsrp/trunk/wsrp-producer-war/src/test/java/org/gatein/wsrp/support/TestMockExportPersistenceManager.java Log: GTNWSRP-55: add missing classes from last commit Added: components/wsrp/trunk/wsrp-producer-war/src/test/java/org/gatein/wsrp/support/TestMockExportPersistenceManager.java =================================================================== --- components/wsrp/trunk/wsrp-producer-war/src/test/java/org/gatein/wsrp/support/TestMockExportPersistenceManager.java (rev 0) +++ components/wsrp/trunk/wsrp-producer-war/src/test/java/org/gatein/wsrp/support/TestMockExportPersistenceManager.java 2010-08-16 22:44:50 UTC (rev 3837) @@ -0,0 +1,159 @@ +/****************************************************************************** + * JBoss, a division of Red Hat * + * Copyright 2010, Red Hat Middleware, LLC, and individual * + * contributors as indicated by the @authors tag. See the * + * copyright.txt in the distribution for a full listing of * + * individual contributors. * + * * + * This 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 2.1 of * + * the License, or (at your option) any later version. * + * * + * This software 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 this software; if not, write to the Free * + * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA * + * 02110-1301 USA, or see the FSF site: http://www.fsf.org. * + ******************************************************************************/ +package org.gatein.wsrp.support; + +import java.io.IOException; +import java.io.UnsupportedEncodingException; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.UUID; + +import org.gatein.exports.ExportPersistenceManager; +import org.gatein.exports.data.ExportContext; +import org.gatein.exports.data.ExportPortletData; +import org.gatein.exports.data.PersistedExportData; +import org.jboss.messaging.util.NotYetImplementedException; +import org.oasis.wsrp.v2.Lifetime; + +/** + * @author <a href="mailto:mwringe@redhat.com">Matt Wringe</a> + * @version $Revision$ + */ +public class TestMockExportPersistenceManager implements ExportPersistenceManager +{ + + public static final String PEC_TYPE = "P_EC"; + public static final double PEC_VERSION = 1.0; + + Map<String,ExportContext> exportContexts = new HashMap<String,ExportContext>(); + + //For testing purposes only + public Map<String, ExportContext> getExportContexts() + { + return exportContexts; + } + + public ExportContext getExportContext(String type, double version, byte[] bytes) throws UnsupportedEncodingException + { + if (supports(type, version)) + { + PersistedExportData persistedExportData = PersistedExportData.create(bytes); + String refId = persistedExportData.getRefId(); + return exportContexts.get(refId); + } + else + { + return null; + } + } + + public void releaseExport(byte[] bytes) + { + try + { + PersistedExportData persistedExportData = PersistedExportData.create(bytes); + String refId = persistedExportData.getRefId(); + if (exportContexts.containsKey(refId)) + { + exportContexts.remove(refId); + } + } + catch (Exception e) + { + System.out.println("ERROR When trying to release exports"); + e.printStackTrace(); + } + } + + public ExportContext retrieveExportContextData(String refid) + { + if (exportContexts.containsKey(refid)) + { + return exportContexts.get(refid); + } + else + { + return null; + } + } + + public ExportPortletData retrieveExportPortletData(String refid) + { + throw new NotYetImplementedException(); + } + + public String storeExportContextData(ExportContext exportContextData) + { + if (exportContextData != null) + { + String refId = UUID.randomUUID().toString(); + exportContexts.put(refId, exportContextData); + return refId; + } + else + { + return null; + } + } + + public String storeExportPortletData(ExportPortletData exportPortletData) + { + throw new NotYetImplementedException(); + } + + public boolean supports(String type, double version) + { + if (type.equals(PEC_TYPE) && (version == PEC_VERSION)) + { + return true; + } + else + { + return false; + } + } + + @Override + public Lifetime updateExportLifetime(ExportContext exportContext, Lifetime lifetime) + { + throw new NotYetImplementedException(); + } + + @Override + public byte[] encodeExportContextData(ExportContext exportContext) throws IOException + { + String refId = storeExportContextData(exportContext); + PersistedExportData persistedExportData = new PersistedExportData(this.PEC_TYPE, refId); + return persistedExportData.encodeAsBytes(); + } + + @Override + public byte[] encodeExportPortletData(ExportContext exportContext, ExportPortletData exportPortlet) + { + // FIXME encodeExportPortletData + return null; + } +} +