1 files changed, 41 insertions, 0 deletions
diff --git a/bundles/org.eclipse.equinox.p2.tests.discovery/src/org/eclipse/equinox/p2/discovery/tests/core/mock/MockBundleDiscoveryStrategy.java b/bundles/org.eclipse.equinox.p2.tests.discovery/src/org/eclipse/equinox/p2/discovery/tests/core/mock/MockBundleDiscoveryStrategy.java
new file mode 100644
index 000000000..ac6069da1
--- /dev/null
+++ b/bundles/org.eclipse.equinox.p2.tests.discovery/src/org/eclipse/equinox/p2/discovery/tests/core/mock/MockBundleDiscoveryStrategy.java
@@ -0,0 +1,41 @@
+/*******************************************************************************
+ * Copyright (c) 2009 Tasktop Technologies 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:
+ *     Tasktop Technologies - initial API and implementation
+ *******************************************************************************/
+
+package org.eclipse.equinox.p2.discovery.tests.core.mock;
+
+import org.eclipse.core.runtime.IContributor;
+import org.eclipse.equinox.internal.p2.discovery.AbstractCatalogSource;
+import org.eclipse.equinox.internal.p2.discovery.Policy;
+import org.eclipse.equinox.internal.p2.discovery.compatibility.BundleDiscoveryStrategy;
+
+/**
+ * a discovery strategy for bundles where the policy can be arbitrarily set
+ * 
+ * @author David Green
+ */
+public class MockBundleDiscoveryStrategy extends BundleDiscoveryStrategy {
+	private Policy policy = Policy.defaultPolicy();
+
+	@Override
+	protected AbstractCatalogSource computeDiscoverySource(IContributor contributor) {
+		AbstractCatalogSource discoverySource = super.computeDiscoverySource(contributor);
+		discoverySource.setPolicy(policy);
+		return discoverySource;
+	}
+
+	public Policy getPolicy() {
+		return policy;
+	}
+
+	public void setPolicy(Policy policy) {
+		this.policy = policy;
+	}
+}