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
|
/*******************************************************************************
* Copyright (c) 2004, 2007 Boeing.
* 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:
* Boeing - initial API and implementation
*******************************************************************************/
package org.eclipse.osee.orcs.core.internal;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import junit.framework.Assert;
import org.eclipse.osee.framework.core.enums.CoreAttributeTypes;
import org.eclipse.osee.framework.core.enums.CoreBranches;
import org.eclipse.osee.framework.core.enums.LoadLevel;
import org.eclipse.osee.framework.core.exception.OseeCoreException;
import org.eclipse.osee.framework.core.model.cache.BranchCache;
import org.eclipse.osee.orcs.ApplicationContext;
import org.eclipse.osee.orcs.OrcsApi;
import org.eclipse.osee.orcs.data.ReadableArtifact;
import org.eclipse.osee.orcs.db.mock.OseeDatabase;
import org.eclipse.osee.orcs.db.mock.OsgiUtil;
import org.eclipse.osee.orcs.search.Operator;
import org.eclipse.osee.orcs.search.QueryBuilder;
import org.eclipse.osee.orcs.search.QueryFactory;
import org.eclipse.osee.orcs.search.ResultSet;
import org.junit.Rule;
/**
* @author Jeff C. Phillips
*/
public class OrcsAttributeSearchTest {
@Rule
public OseeDatabase db1 = new OseeDatabase("osee.demo.h2");
@org.junit.Test
public void runGodMethod() throws OseeCoreException {
OrcsApi orcsApi = OsgiUtil.getService(OrcsApi.class);
ApplicationContext context = null; // TODO use real application context
QueryFactory queryFactory = orcsApi.getQueryFactory(context);
testNameAttributeNotEqualSearch(queryFactory);
testNameAttributeEqualSearch(queryFactory);
testBooleanAttributeSearch(queryFactory);
testWTCAttributeEqualSearch(queryFactory, orcsApi.getBranchCache());
}
public void testNameAttributeNotEqualSearch(QueryFactory queryFactory) throws OseeCoreException {
QueryBuilder builder =
queryFactory.fromBranch(CoreBranches.COMMON).and(CoreAttributeTypes.Name, Operator.NOT_EQUAL, "User Groups");
ResultSet<ReadableArtifact> resultSet = builder.build(LoadLevel.FULL);
List<ReadableArtifact> moreArts = resultSet.getList();
for (ReadableArtifact artifact : moreArts) {
Assert.assertTrue(artifact.getId() != 8);
}
}
public void testNameAttributeEqualSearch(QueryFactory queryFactory) throws OseeCoreException {
QueryBuilder builder =
queryFactory.fromBranch(CoreBranches.COMMON).and(CoreAttributeTypes.Name, Operator.EQUAL, "User Groups");
ResultSet<ReadableArtifact> resultSet = builder.build(LoadLevel.FULL);
List<ReadableArtifact> moreArts = resultSet.getList();
Assert.assertEquals(1, moreArts.size());
Assert.assertEquals(1, builder.getCount());
Map<Integer, ReadableArtifact> lookup = creatLookup(moreArts);
ReadableArtifact art8 = lookup.get(8);
//Test loading name attributes
Assert.assertEquals(art8.getSoleAttributeAsString(CoreAttributeTypes.Name), "User Groups");
}
public void testWTCAttributeEqualSearch(QueryFactory queryFactory, BranchCache branchCache) throws OseeCoreException {
QueryBuilder builder =
queryFactory.fromBranch(branchCache.getByName("SAW_Bld_1").iterator().next()).and(
CoreAttributeTypes.WholeWordContent, Operator.EQUAL, "commands");
ResultSet<ReadableArtifact> resultSet = builder.build(LoadLevel.FULL);
List<ReadableArtifact> moreArts = resultSet.getList();
Assert.assertFalse(moreArts.isEmpty());
Assert.assertTrue(builder.getCount() > 0);
}
public void testBooleanAttributeSearch(QueryFactory queryFactory) throws OseeCoreException {
QueryBuilder builder =
queryFactory.fromBranch(CoreBranches.COMMON).and(CoreAttributeTypes.DefaultGroup, Operator.EQUAL, "yes");
ResultSet<ReadableArtifact> resultSet = builder.build(LoadLevel.FULL);
List<ReadableArtifact> moreArts = resultSet.getList();
Assert.assertEquals(1, moreArts.size());
Assert.assertEquals(1, builder.getCount());
Map<Integer, ReadableArtifact> lookup = creatLookup(moreArts);
ReadableArtifact art9 = lookup.get(9);
Assert.assertEquals(art9.getSoleAttributeAsString(CoreAttributeTypes.Name), "Everyone");
}
Map<Integer, ReadableArtifact> creatLookup(List<ReadableArtifact> arts) {
Map<Integer, ReadableArtifact> lookup = new HashMap<Integer, ReadableArtifact>();
for (ReadableArtifact artifact : arts) {
lookup.put(artifact.getId(), artifact);
}
return lookup;
}
}
|
