/*******************************************************************************
* Copyright (c) 2007, 2008 IBM Corporation and others.
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*
* Contributors:
* IBM Corporation - initial API and implementation
*******************************************************************************/
package org.eclipse.equinox.p2.tests.metadata;
import java.io.*;
import java.util.*;
import javax.xml.parsers.ParserConfigurationException;
import org.eclipse.equinox.internal.p2.core.helpers.OrderedProperties;
import org.eclipse.equinox.internal.p2.metadata.InstallableUnit;
import org.eclipse.equinox.internal.p2.metadata.InstallableUnitPatch;
import org.eclipse.equinox.internal.p2.metadata.repository.io.MetadataParser;
import org.eclipse.equinox.internal.p2.metadata.repository.io.MetadataWriter;
import org.eclipse.equinox.internal.provisional.p2.metadata.*;
import org.eclipse.equinox.p2.tests.AbstractProvisioningTest;
import org.eclipse.equinox.p2.tests.TestActivator;
import org.eclipse.equinox.internal.provisional.p2.core.VersionRange;
import org.osgi.framework.BundleContext;
import org.eclipse.equinox.internal.provisional.p2.core.Version;
import org.xml.sax.*;
public class IUPatchPersistenceTest extends AbstractProvisioningTest {
// Randomly chose org.eclipse.osgi.services as the IU for testing persistence
// but 'enhanced' it for better coverage.
private static String PackagesNS = "osgi.packages";
private static String id = "org.eclipse.osgi.services";
private static Version version = new Version("3.1.200.v20070605");
private static String filter = "(& (osgi.ws=win32) (osgi.os=win32) (osgi.arch=x86))"; // not really
private static String[][] properties = new String[][] {new String[] {"equinox.p2.name", "OSGi Release 4.0.1 Services"}, //
new String[] {"equinox.p2.description", "OSGi Service Platform Release 4.0.1 Service Interfaces and Classes"}, //
new String[] {"equinox.p2.provider", "Eclipse.org"}, //
new String[] {"equinox.p2.contact", "www.eclipse.org"}, //
new String[] {"equinox.p2.copyright", "Copyright (c) 2003, 2004 IBM Corporation and others. A long-winded copyright notice."}};
private static String[][] provides = new String[][] {new String[] {PackagesNS, "org.osgi.service.cm", "1.2.0"}, //
new String[] {PackagesNS, "org.osgi.service.component", "1.0.0"}, //
new String[] {PackagesNS, "org.osgi.service.device", "1.1.0"}, //
new String[] {PackagesNS, "org.osgi.service.event", "1.1.0"}, //
new String[] {PackagesNS, "org.osgi.service.http", "1.2.0"}, //
new String[] {PackagesNS, "org.osgi.service.io", "1.0.0"}, //
new String[] {PackagesNS, "org.osgi.service.log", "1.3.0"}, //
new String[] {PackagesNS, "org.osgi.service.metatype", "1.1.0"}, //
new String[] {PackagesNS, "org.osgi.service.provisioning", "1.1.0"}, //
new String[] {PackagesNS, "org.osgi.service.upnp", "1.1.0"}, //
new String[] {PackagesNS, "org.osgi.service.useradmin", "1.1.0"}, //
new String[] {PackagesNS, "org.osgi.service.wireadmin", "1.0.0"}}; //
private static String[][] requires = new String[][] {new String[] {PackagesNS, "javax.servlet", "0.0.0", "true"}, //
new String[] {PackagesNS, "javax.servlet.http", "0.0.0", "true"}, //
new String[] {PackagesNS, "org.osgi.framework", "1.2.0", "false"}}; //
private static String[][] instructions = new String[][] {new String[] {"manifest", "Manifest-Version: 1.0\\Bundle-Vendor: Eclipse.org\\Bundle-ContactAddress: www.eclipse.org\\...a whole bunch of other manifest content..."}, new String[] {"zipped", "true"}, //
new String[] {"configure", "addProgramArg(programArg:-startup);addProgramArg(programArg:@artifact);"}}; //
public static IInstallableUnit createPersistenceTestIU() {
Map propertyMap = createProperties(properties);
ProvidedCapability[] additionalProvides = createProvided(provides);
RequiredCapability[] requirements = createRequired(requires);
TouchpointData tpData = createTouchpointData(instructions);
IUpdateDescriptor update = createUpdateDescriptor();
boolean singleton = false;
IInstallableUnit iu = createIU(id, version, filter, requirements, additionalProvides, propertyMap, TOUCHPOINT_OSGI, tpData, singleton, update);
return iu;
}
private static IUpdateDescriptor createUpdateDescriptor() {
return MetadataFactory.createUpdateDescriptor(id, new VersionRange(IU_TEST_VERSION, true, IU_TEST_VERSION, true), IUpdateDescriptor.HIGH, "desc");
}
private static Map createProperties(String[][] keyValuePairs) {
OrderedProperties props = new OrderedProperties(keyValuePairs.length);
for (int i = 0; i < keyValuePairs.length; i++) {
String[] nextPair = keyValuePairs[i];
props.put(nextPair[0], nextPair[1]);
}
return props;
}
private static ProvidedCapability[] createProvided(String[][] provideTuples) {
ProvidedCapability[] provided = new ProvidedCapability[provideTuples.length];
for (int i = 0; i < provideTuples.length; i++) {
String[] nextTuple = provideTuples[i];
provided[i] = MetadataFactory.createProvidedCapability(nextTuple[0], nextTuple[1], new Version(nextTuple[2]));
}
// provided[provideTuples.length] = BUNDLE_CAPABILITY;
return provided;
}
private static RequiredCapability[] createRequired(String[][] requireTuples) {
RequiredCapability[] required = new RequiredCapability[requireTuples.length];
for (int i = 0; i < requireTuples.length; i++) {
String[] nextTuple = requireTuples[i];
required[i] = MetadataFactory.createRequiredCapability(nextTuple[0], nextTuple[1], new VersionRange(nextTuple[2]), null, Boolean.valueOf(nextTuple[3]).booleanValue(), false);
}
return required;
}
private static TouchpointData createTouchpointData(String[][] instructionData) {
Map map = new LinkedHashMap(instructionData.length);
for (int i = 0; i < instructionData.length; i++) {
String[] nextInstruction = instructionData[i];
map.put(nextInstruction[0], nextInstruction[1]);
}
return MetadataFactory.createTouchpointData(map);
}
private static String IU_TEST_TARGET = "installableUnitTest";
private static Version IU_TEST_VERSION = new Version("0.0.1");
private static String IU_TEST_ELEMENT = "test";
class IUStringWriter extends MetadataWriter {
public IUStringWriter(ByteArrayOutputStream stream) throws IOException {
super(stream, new ProcessingInstruction[] {ProcessingInstruction.makeClassVersionInstruction(IU_TEST_TARGET, InstallableUnit.class, IU_TEST_VERSION)});
}
public void writeTest(IInstallableUnit iu) {
start(IU_TEST_ELEMENT);
writeInstallableUnit(iu);
end(IU_TEST_ELEMENT);
flush();
}
}
class IUStringParser extends MetadataParser {
public IUStringParser(BundleContext context, String bundleId) {
super(context, bundleId);
}
public void parse(String profileString) throws IOException {
this.status = null;
try {
getParser();
TestHandler testHandler = new TestHandler();
xmlReader.setContentHandler(new IUDocHandler(IU_TEST_ELEMENT, testHandler));
xmlReader.parse(new InputSource(new StringReader(profileString)));
if (isValidXML()) {
theIU = testHandler.getIU();
}
} catch (SAXException e) {
throw new IOException(e.getMessage());
} catch (ParserConfigurationException e) {
fail();
}
}
private IInstallableUnit theIU = null;
private final class IUDocHandler extends DocHandler {
public IUDocHandler(String rootName, RootHandler rootHandler) {
super(rootName, rootHandler);
}
public void processingInstruction(String target, String data) throws SAXException {
if (IU_TEST_TARGET.equals(target)) {
String clazz = extractPIClass(data);
try {
if (!Class.forName(clazz).equals(InstallableUnit.class)) {
throw new SAXException("Wrong class '" + clazz + "' in processing instruction"); //$NON-NLS-1$//$NON-NLS-2$
}
} catch (ClassNotFoundException e) {
throw new SAXException("InstallableUnit class '" + clazz + "' not found"); //$NON-NLS-1$//$NON-NLS-2$
}
Version iuTestVersion = extractPIVersion(target, data);
if (!IU_TEST_VERSION.equals(iuTestVersion)) {
throw new SAXException("Bad iu test version.");
}
}
}
}
private final class TestHandler extends RootHandler {
private InstallableUnitHandler iuHandler = null;
private InstallableUnit iu = null;
private List singleton = new ArrayList(1);
public TestHandler() {
super();
}
public InstallableUnit getIU() {
return iu;
}
protected void handleRootAttributes(Attributes attributes) {
parseAttributes(attributes, noAttributes, noAttributes);
}
public void startElement(String name, Attributes attributes) {
if (INSTALLABLE_UNIT_ELEMENT.equals(name)) {
if (iuHandler == null) {
iuHandler = new InstallableUnitHandler(this, attributes, singleton);
} else {
duplicateElement(this, name, attributes);
}
} else {
invalidElement(name, attributes);
}
}
protected void finished() {
if (isValidXML()) {
if (iuHandler != null && singleton.size() == 1) {
iu = (InstallableUnit) iuHandler.getInstallableUnit();
}
}
}
}
protected String getErrorMessage() {
return "Error parsing installable unit string";
}
protected Object getRootObject() {
return theIU;
}
}
public void testIUPersistence() throws IOException {
IInstallableUnit iu0 = createPersistenceTestIU();
validateIU(iu0);
ByteArrayOutputStream output0 = new ByteArrayOutputStream(3072);
IUStringWriter writer0 = new IUStringWriter(output0);
writer0.writeTest(iu0);
String iuText0 = output0.toString();
output0.close();
IUStringParser parser = new IUStringParser(TestActivator.context, TestActivator.PI_PROV_TESTS);
parser.parse(iuText0);
assertTrue("Error parsing test iu: " + parser.getStatus().getMessage(), parser.getStatus().isOK());
InstallableUnit iu1 = (InstallableUnit) parser.getRootObject();
validateIU(iu1);
ByteArrayOutputStream output1 = new ByteArrayOutputStream(1492);
IUStringWriter writer = new IUStringWriter(output1);
writer.writeTest(iu1);
String iuText1 = output1.toString();
output1.close();
assertTrue("Installable unit write after read after write produced different XML", iuText1.equals(iuText0));
}
public void testIUPatchPersistence() throws IOException {
IInstallableUnit iu0 = createPatchIU();
validateIU(iu0);
ByteArrayOutputStream output0 = new ByteArrayOutputStream(3072);
IUStringWriter writer0 = new IUStringWriter(output0);
writer0.writeTest(iu0);
String iuText0 = output0.toString();
output0.close();
IUStringParser parser = new IUStringParser(TestActivator.context, TestActivator.PI_PROV_TESTS);
parser.parse(iuText0);
assertTrue("Error parsing test iu: " + parser.getStatus().getMessage(), parser.getStatus().isOK());
InstallableUnitPatch iu1 = (InstallableUnitPatch) parser.getRootObject();
validateIU(iu1);
validateIUPatch(iu1);
ByteArrayOutputStream output1 = new ByteArrayOutputStream(1492);
IUStringWriter writer = new IUStringWriter(output1);
writer.writeTest(iu1);
String iuText1 = output1.toString();
output1.close();
assertTrue("Installable unit write after read after write produced different XML", iuText1.equals(iuText0));
}
private IInstallableUnitPatch createPatchIU() {
Map propertyMap = createProperties(properties);
ProvidedCapability[] additionalProvides = createProvided(provides);
RequiredCapability[] requirements = createRequired(requires);
TouchpointData tpData = createTouchpointData(instructions);
IUpdateDescriptor update = createUpdateDescriptor();
boolean singleton = false;
RequirementChange change1 = new RequirementChange(MetadataFactory.createRequiredCapability(IInstallableUnit.NAMESPACE_IU_ID, "B", VersionRange.emptyRange, null, false, false, false), MetadataFactory.createRequiredCapability(IInstallableUnit.NAMESPACE_IU_ID, "B", new VersionRange("[1.1.0, 1.3.0)"), null, false, false, true));
RequirementChange change2 = new RequirementChange(null, MetadataFactory.createRequiredCapability(IInstallableUnit.NAMESPACE_IU_ID, "B", new VersionRange("[1.1.0, 1.3.0)"), null, false, false, true));
RequirementChange change3 = new RequirementChange(MetadataFactory.createRequiredCapability(IInstallableUnit.NAMESPACE_IU_ID, "B", VersionRange.emptyRange, null, false, false, false), null);
RequiredCapability[][] scope = new RequiredCapability[][] { {MetadataFactory.createRequiredCapability("foo", "bar", null, null, true, true), MetadataFactory.createRequiredCapability("foo", "bar", null, null, true, true)}, {MetadataFactory.createRequiredCapability("zoo", "far", null, null, true, true)}};
RequiredCapability lifeCycle = MetadataFactory.createRequiredCapability("zoo", "x", null, null, false, false, false);
IInstallableUnitPatch iu = createIUPatch(id, version, filter, requirements, additionalProvides, propertyMap, TOUCHPOINT_OSGI, tpData, singleton, update, new RequirementChange[] {change1, change2, change3}, scope, lifeCycle);
return iu;
}
private static void validateIUPatch(IInstallableUnitPatch iu) {
validateIU(iu);
assertTrue(iu.getApplicabilityScope() != null);
assertTrue(iu.getRequiredCapabilities() != null);
assertEquals(3, iu.getRequirementsChange().length);
assertEquals(null, iu.getRequirementsChange()[1].applyOn());
assertNotNull(iu.getRequirementsChange()[1].newValue());
assertEquals(null, iu.getRequirementsChange()[2].newValue());
assertNotNull(iu.getRequirementsChange()[2].applyOn());
assertEquals(2, iu.getApplicabilityScope().length);
assertEquals(2, iu.getApplicabilityScope()[0].length);
assertEquals(1, iu.getApplicabilityScope()[1].length);
assertNotNull(iu.getLifeCycle());
}
private static void validateIU(IInstallableUnit iu) {
assertTrue("Installable unit id is not correct", id.equals(iu.getId()));
assertTrue("Installable unit version is not correct", version.equals(iu.getVersion()));
assertTrue("Installable unit filter is not correct", filter.equals(iu.getFilter()));
// assertTrue("Installable unit properties are not correct", Arrays.equals(properties, extractProperties(iu)));
assertTrue("Installable unit properties are not correct", equal(properties, extractProperties(iu)));
assertTrue("Installable unit provided capabilities are not correct", equal(addSelfCapability(iu, provides), extractProvides(iu)));
// assertTrue("Installable unit required capabilities are not correct", equal(requires, extractRequires(iu))); The lifecycle is added as a requirement for now to make things easier
assertTrue("Installable unit update descriptor are not correct", id.equals(iu.getUpdateDescriptor().getId()));
assertTrue("Installable unit update descriptor are not correct", IUpdateDescriptor.HIGH == iu.getUpdateDescriptor().getSeverity());
assertTrue("Installable unit update descriptor are not correct", "desc".equals(iu.getUpdateDescriptor().getDescription()));
assertTrue("Installable unit update descriptor are not correct", new VersionRange(IU_TEST_VERSION, true, IU_TEST_VERSION, true).equals(iu.getUpdateDescriptor().getRange()));
}
private static String[][] extractProperties(IInstallableUnit iu) {
Map props = iu.getProperties();
Set keys = props.keySet();
String[][] pairs = new String[keys.size()][2];
int index = 0;
for (Iterator iter = keys.iterator(); iter.hasNext();) {
String nextKey = (String) iter.next();
String nextValue = (String) props.get(nextKey);
pairs[index] = new String[] {nextKey, nextValue};
index++;
}
return pairs;
}
private static String[][] addSelfCapability(IInstallableUnit iu, String[][] provideTuples) {
String[][] augmentedProvides = new String[provideTuples.length + 1][3];
ProvidedCapability self = getSelfCapability(iu);
augmentedProvides[0] = new String[] {self.getNamespace(), self.getName(), self.getVersion().toString()};
for (int i = 0; i < provideTuples.length; i++) {
augmentedProvides[i + 1] = provideTuples[i];
}
return augmentedProvides;
}
private static String[][] extractProvides(IInstallableUnit iu) {
ProvidedCapability[] provydes = iu.getProvidedCapabilities();
String[][] tuples = new String[provydes.length][3];
for (int i = 0; i < provydes.length; i++) {
ProvidedCapability next = provydes[i];
tuples[i] = new String[] {next.getNamespace(), next.getName(), next.getVersion().toString()};
}
return tuples;
}
// private static String[][] extractRequires(IInstallableUnit iu) {
// RequiredCapability[] requyres = iu.getRequiredCapabilities();
// String[][] tuples = new String[requyres.length][4];
// for (int i = 0; i < requyres.length; i++) {
// RequiredCapability next = requyres[i];
// tuples[i] = new String[] {next.getNamespace(), next.getName(), next.getRange().toString(), Boolean.valueOf(next.isOptional()).toString()};
// }
// return tuples;
// }
}
