1 files changed, 46 insertions, 0 deletions

diff --git a/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/support/MockRepositoryConnectorAdapter.java b/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/support/MockRepositoryConnectorAdapter.java
index e4a26c25a..994438985 100644
--- a/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/support/MockRepositoryConnectorAdapter.java
+++ b/org.eclipse.mylyn.tasks.tests/src/org/eclipse/mylyn/tasks/tests/support/MockRepositoryConnectorAdapter.java
@@ -13,6 +13,8 @@ package org.eclipse.mylyn.tasks.tests.support;
 
 import java.io.IOException;
 import java.io.InputStream;
+import java.util.ArrayList;
+import java.util.List;
 
 import org.eclipse.core.runtime.IAdapterFactory;
 import org.eclipse.mylyn.commons.sdk.util.CommonTestUtil;
@@ -22,6 +24,8 @@ import org.eclipse.mylyn.tasks.tests.connector.MockRepositoryConnector;
 import org.eclipse.mylyn.tasks.tests.connector.MockRepositoryConnectorUi;
 import org.eclipse.mylyn.tasks.ui.AbstractRepositoryConnectorUi;
 
+import com.google.common.collect.ImmutableList;
+
 public class MockRepositoryConnectorAdapter implements IAdapterFactory {
 
 	public final static class DynamicMockRepositoryConnectorUi extends MockRepositoryConnectorUi {
@@ -58,6 +62,48 @@ public class MockRepositoryConnectorAdapter implements IAdapterFactory {
 					public InputStream getBrandingImageData() throws IOException {
 						return CommonTestUtil.getResource(this, "testdata/icons/mock-repository.gif");
 					}
+
+					@Override
+					public List<String> getBrands() {
+						ArrayList<String> brands = new ArrayList<String>(ImmutableList.of("org.mylyn", "org.eclipse",
+								"exceptional"));
+						brands.add(1, null);
+						return brands;
+					}
+
+					@Override
+					public String getConnectorLabel(String brand) {
+						if ("exceptional".equals(brand)) {
+							throw new NullPointerException();
+						} else if (getBrands().contains(brand)) {
+							return "Label for " + brand;
+						}
+						return super.getConnectorLabel(brand);
+					}
+
+					@Override
+					public InputStream getOverlayImageData(final String brand) throws IOException {
+						if ("org.mylyn".equals(brand)) {
+							return CommonTestUtil.getResource(this, "testdata/icons/mock-5x5.gif");
+						} else if ("org.eclipse".equals(brand)) {
+							return CommonTestUtil.getResource(this, "testdata/icons/mock-4x4.gif");
+						} else if ("exceptional".equals(brand)) {
+							throw new NullPointerException();
+						}
+						return super.getOverlayImageData(brand);
+					}
+
+					@Override
+					public InputStream getBrandingImageData(final String brand) throws IOException {
+						if ("org.mylyn".equals(brand)) {
+							return CommonTestUtil.getResource(this, "testdata/icons/mock-3x3.gif");
+						} else if ("org.eclipse".equals(brand)) {
+							return CommonTestUtil.getResource(this, "testdata/icons/mock-2x2.gif");
+						} else if ("exceptional".equals(brand)) {
+							throw new NullPointerException();
+						}
+						return super.getBrandingImageData(brand);
+					}
 				};
 			}
 		}