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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
|
/**
* 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.model.test;
import org.fedoraproject.candlepin.model.Consumer;
import org.fedoraproject.candlepin.model.ConsumerType;
import org.fedoraproject.candlepin.model.EntitlementPool;
import org.fedoraproject.candlepin.model.Owner;
import org.fedoraproject.candlepin.model.Product;
import org.fedoraproject.candlepin.model.User;
import org.fedoraproject.candlepin.test.DatabaseTestFixture;
import org.fedoraproject.candlepin.test.TestUtil;
import java.util.List;
import org.junit.Test;
import static org.junit.Assert.*;
public class OwnerTest extends DatabaseTestFixture {
@Test
public void testCreate() throws Exception {
String ownerName = "Example Corporation";
Owner o = new Owner(ownerName);
persistAndCommit(o);
Owner result = (Owner)em.createQuery(
"select o from Owner o where o.name = :name")
.setParameter("name", ownerName).getSingleResult();
assertNotNull(result);
assertEquals(ownerName, result.getName());
assertTrue(result.getId() > 0);
assertEquals(o.getId(), result.getId());
}
@Test
public void testList() throws Exception {
beginTransaction();
List<Owner> orgs = em.createQuery("select o from Owner as o")
.getResultList();
int beforeCount = orgs.size();
for (int i = 0; i < 10; i++) {
em.persist(new Owner("Corp " + i));
}
commitTransaction();
orgs = em.createQuery("select o from Owner as o")
.getResultList();
int afterCount = orgs.size();
assertEquals(10, afterCount - beforeCount);
}
@Test
public void testObjectRelationships() throws Exception {
Owner owner = new Owner("test-owner");
// Product
Product rhel = new Product();
rhel.setName("Red Hat Enterprise Linux");
// User
User u = new User();
u.setLogin("test-login");
u.setPassword("redhat");
owner.addUser(u);
assertEquals(1, owner.getUsers().size());
// Consumer
Consumer c = new Consumer();
c.setOwner(owner);
owner.addConsumer(c);
c.addConsumedProduct(rhel);
assertEquals(1, owner.getConsumers().size());
assertEquals(1, c.getConsumedProducts().size());
// EntitlementPool
EntitlementPool pool = new EntitlementPool();
owner.addEntitlementPool(pool);
pool.setProduct(rhel);
assertEquals(1, owner.getEntitlementPools().size());
}
@Test
public void bidirectionalConsumers() throws Exception {
beginTransaction();
Owner o = TestUtil.createOwner();
ConsumerType consumerType = TestUtil.createConsumerType();
Consumer c1 = TestUtil.createConsumer(consumerType, o);
Consumer c2 = TestUtil.createConsumer(consumerType, o);
o.addConsumer(c1);
o.addConsumer(c2);
em.persist(o);
em.persist(consumerType);
em.persist(c1);
em.persist(c2);
commitTransaction();
assertEquals(2, o.getConsumers().size());
em.clear();
Owner lookedUp = (Owner)em.find(Owner.class, o.getId());
assertEquals(2, lookedUp.getConsumers().size());
}
@Test
public void bidirectionalUsers() throws Exception {
beginTransaction();
Owner o = TestUtil.createOwner();
User u1 = TestUtil.createUser(o);
User u2 = TestUtil.createUser(o);
o.addUser(u1);
o.addUser(u2);
em.persist(o);
em.persist(u1);
em.persist(u2);
commitTransaction();
assertEquals(2, o.getUsers().size());
em.clear();
Owner lookedUp = (Owner)em.find(Owner.class, o.getId());
assertEquals(2, lookedUp.getUsers().size());
}
}
|
