bundles/org.eclipse.equinox.p2.tests.discovery/src/org/eclipse/equinox/p2/discovery/tests/core/util/CatalogConnectorComparatorTest.java

/*******************************************************************************
 * Copyright (c) 2004, 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.util;

import junit.framework.TestCase;
import org.eclipse.equinox.internal.p2.discovery.model.*;
import org.eclipse.equinox.internal.p2.discovery.util.CatalogItemComparator;

public class CatalogConnectorComparatorTest extends TestCase {

	private CatalogCategory category;

	private CatalogItemComparator comparator;

	@Override
	protected void setUp() throws Exception {
		super.setUp();
		category = new CatalogCategory();
		comparator = new CatalogItemComparator();
	}

	private Group addGroup(String id) {
		Group group = new Group();
		group.setId(id);
		category.getGroup().add(group);
		return group;
	}

	private CatalogItem addConnectorDescriptor(String id, String name, String groupId) {
		CatalogItem connector = new CatalogItem();
		connector.setId(id);
		connector.setName(name);
		connector.setGroupId(groupId);
		connector.setCategory(category);
		category.getItems().add(connector);
		return connector;
	}

	public void testOrderByGroup() {
		addGroup("1");
		addGroup("2");
		CatalogItem t1 = addConnectorDescriptor("b", "btest", "2");
		CatalogItem t2 = addConnectorDescriptor("a", "atest", "2");
		CatalogItem t3 = addConnectorDescriptor("c", "ctest", "1");
		CatalogItem t4 = addConnectorDescriptor("d", "dtest", "1");
		CatalogItem t5 = addConnectorDescriptor("0", "0test", null);

		assertEquals(-1, comparator.compare(t2, t1));
		assertEquals(1, comparator.compare(t1, t2));
		assertEquals(-1, comparator.compare(t3, t4));
		assertEquals(1, comparator.compare(t4, t3));

		assertEquals(-1, comparator.compare(t1, t5));
		assertEquals(1, comparator.compare(t5, t1));
		assertEquals(-1, comparator.compare(t2, t5));
		assertEquals(1, comparator.compare(t5, t2));
		assertEquals(-1, comparator.compare(t3, t5));
		assertEquals(1, comparator.compare(t5, t3));
		assertEquals(-1, comparator.compare(t4, t5));
		assertEquals(1, comparator.compare(t5, t4));

		assertEquals(-1, comparator.compare(t3, t1));
		assertEquals(1, comparator.compare(t1, t3));
		assertEquals(-1, comparator.compare(t3, t2));
		assertEquals(1, comparator.compare(t2, t3));

		assertEquals(-1, comparator.compare(t4, t1));
		assertEquals(1, comparator.compare(t1, t4));
		assertEquals(-1, comparator.compare(t4, t2));
		assertEquals(1, comparator.compare(t2, t4));
	}
}