proxy/src/test/java/org/fedoraproject/candlepin/resource/test/EntitlementMatcherTest.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

/**
 * Copyright (c) 2009 Red Hat, Inc.
 *
 * This software is licensed to you under the GNU General Public License,
 * version 2 (GPLv2). There is NO WARRANTY for this software, express or
 * implied, including the implied warranties of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. You should have received a copy of GPLv2
 * along with this software; if not, see
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
 *
 * Red Hat trademarks are not licensed under GPLv2. No permission is
 * granted to use or replicate Red Hat trademarks that are incorporated
 * in this software or its documentation.
 */
package org.fedoraproject.candlepin.resource.test;

import org.fedoraproject.candlepin.model.Consumer;
import org.fedoraproject.candlepin.model.ConsumerType;
import org.fedoraproject.candlepin.model.ObjectFactory;
import org.fedoraproject.candlepin.model.Product;
import org.fedoraproject.candlepin.model.ProductFactory;
import org.fedoraproject.candlepin.model.test.TestUtil;
import org.fedoraproject.candlepin.resource.EntitlementMatcher;

import java.util.List;

import org.junit.Test;
import static org.junit.Assert.*;

/**
 * EntitlementMatcherTest
 * @version $Rev$
 */
public class EntitlementMatcherTest {

    @Test
    public void testIsCompatable() throws Exception {
        Consumer consumer = TestUtil.createConsumer();
        ConsumerType typeSystem = ProductFactory.get().lookupConsumerTypeByLabel("system");
        consumer.setType(typeSystem);
        
        List f = ObjectFactory.get().listObjectsByClass(Product.class);
        Product rhel = (Product) ObjectFactory.get().lookupByFieldName(
                Product.class, "label", "rhel");
        Product rhelvirt = (Product) ObjectFactory.get().lookupByFieldName(
                Product.class, "label", "rhel-virt");

        EntitlementMatcher m = new EntitlementMatcher();
        
        assertTrue(m.isCompatible(consumer, rhel));
        
        ConsumerType vmwarehost = 
            ProductFactory.get().lookupConsumerTypeByLabel("vmwarehost");
        consumer.setType(vmwarehost);
        
        // Check that you can't use rhelvirt on a vmware host
        assertFalse(m.isCompatible(consumer, rhelvirt));
    }
}