1 files changed, 57 insertions, 0 deletions
diff --git a/plugins/org.eclipse.osee.framework.skynet.core.test/src/org/eclipse/osee/framework/skynet/core/artifact/search/AttributeCriteriaTest.java b/plugins/org.eclipse.osee.framework.skynet.core.test/src/org/eclipse/osee/framework/skynet/core/artifact/search/AttributeCriteriaTest.java
new file mode 100644
index 00000000000..49df447bb23
--- /dev/null
+++ b/plugins/org.eclipse.osee.framework.skynet.core.test/src/org/eclipse/osee/framework/skynet/core/artifact/search/AttributeCriteriaTest.java
@@ -0,0 +1,57 @@
+/*******************************************************************************
+ * Copyright (c) 2012 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.framework.skynet.core.artifact.search;
+
+import static org.mockito.Mockito.*;
+import java.util.Arrays;
+import java.util.List;
+import org.eclipse.osee.framework.core.enums.CoreAttributeTypes;
+import org.eclipse.osee.framework.core.enums.Operator;
+import org.eclipse.osee.framework.core.enums.TokenOrderType;
+import org.eclipse.osee.framework.core.exception.OseeCoreException;
+import org.eclipse.osee.orcs.rest.client.QueryBuilder;
+import org.junit.Test;
+
+/**
+ * @author John Misinco
+ */
+public class AttributeCriteriaTest {
+
+   @Test
+   public void testAddToQueryBuilder() throws OseeCoreException {
+      AttributeCriteria criteria = new AttributeCriteria(CoreAttributeTypes.Active);
+      QueryBuilder builder = mock(QueryBuilder.class);
+      criteria.addToQueryBuilder(builder);
+      verify(builder).andExists(CoreAttributeTypes.Active);
+
+      reset(builder);
+      List<String> values = Arrays.asList("true", "false");
+      criteria = new AttributeCriteria(CoreAttributeTypes.Active, values);
+      criteria.addToQueryBuilder(builder);
+      verify(builder).and(CoreAttributeTypes.Active, Operator.EQUAL, values);
+
+      reset(builder);
+      criteria = new AttributeCriteria(CoreAttributeTypes.Active, "true", Operator.LESS_THAN);
+      criteria.addToQueryBuilder(builder);
+      verify(builder).and(CoreAttributeTypes.Active, Operator.LESS_THAN, "true");
+
+      reset(builder);
+      criteria = new AttributeCriteria(CoreAttributeTypes.Active, "true", TokenOrderType.ANY_ORDER);
+      criteria.addToQueryBuilder(builder);
+      verify(builder).and(CoreAttributeTypes.Active, "true", TokenOrderType.ANY_ORDER);
+
+      reset(builder);
+      criteria = new AttributeCriteria(CoreAttributeTypes.Active, "true");
+      criteria.addToQueryBuilder(builder);
+      verify(builder).and(CoreAttributeTypes.Active, "true", QueryOptions.EXACT_MATCH_OPTIONS);
+
+   }
+}