1 files changed, 219 insertions, 0 deletions
diff --git a/plugins/org.eclipse.emf.cdo.tests.objectivity/src/org/eclipse/emf/cdo/tests/objectivity/ObjyStoreQueryTest.java b/plugins/org.eclipse.emf.cdo.tests.objectivity/src/org/eclipse/emf/cdo/tests/objectivity/ObjyStoreQueryTest.java
new file mode 100644
index 0000000000..b90c9776ea
--- /dev/null
+++ b/plugins/org.eclipse.emf.cdo.tests.objectivity/src/org/eclipse/emf/cdo/tests/objectivity/ObjyStoreQueryTest.java
@@ -0,0 +1,219 @@
+/**
+ * Copyright (c) 2004 - 2009 Eike Stepper (Berlin, Germany) 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:
+ *    Simon McDuff - initial API and implementation
+ *    Eike Stepper - maintenance
+ */
+package org.eclipse.emf.cdo.tests.objectivity;
+
+import org.eclipse.emf.cdo.eresource.CDOResource;
+import org.eclipse.emf.cdo.internal.server.Repository;
+import org.eclipse.emf.cdo.session.CDOSession;
+import org.eclipse.emf.cdo.tests.AbstractCDOTest;
+import org.eclipse.emf.cdo.tests.model1.Category;
+import org.eclipse.emf.cdo.tests.model1.Company;
+import org.eclipse.emf.cdo.transaction.CDOTransaction;
+import org.eclipse.emf.cdo.view.CDOQuery;
+
+import org.eclipse.emf.internal.cdo.query.CDOQueryResultIteratorImpl;
+
+import org.eclipse.net4j.util.collection.CloseableIterator;
+import org.eclipse.net4j.util.tests.AbstractOMTest.PollingTimeOuter;
+
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+/**
+ * @author Simon McDuff
+ */
+public class ObjyStoreQueryTest extends AbstractCDOTest
+{
+  public void testObjyStoreBasicQuery() throws Exception
+  {
+    //skipUnlessConfig(MEM);
+
+    Set<Object> objects = new HashSet<Object>();
+    CDOSession session = openSession();
+    CDOTransaction transaction = session.openTransaction();
+
+    CDOResource resource1 = transaction.createResource("/test1");
+    objects.add(resource1);
+    Company company1 = getModel1Factory().createCompany();
+    Category category1 = getModel1Factory().createCategory();
+
+    resource1.getContents().add(company1);
+    company1.getCategories().add(category1);
+
+    objects.add(company1);
+    objects.add(category1);
+    objects.add(transaction.getRootResource());
+    company1.setName("TEST");
+
+    transaction.commit();
+
+    CDOQuery query = transaction.createQuery("TEST", "QUERYSTRING");
+    List<Object> result = query.getResult(Object.class);
+    assertEquals(4, result.size());
+    for (Object object : result)
+    {
+      assertEquals(true, objects.contains(object));
+    }
+
+    transaction.close();
+    session.close();
+  }
+
+  public void testObjyStoreBasicQuery_EClassParameter() throws Exception
+  {
+    //skipUnlessConfig(MEM);
+
+    Set<Object> objects = new HashSet<Object>();
+    CDOSession session = openSession();
+    CDOTransaction transaction = session.openTransaction();
+
+    CDOResource resource1 = transaction.createResource("/test1");
+    objects.add(resource1);
+    Company company1 = getModel1Factory().createCompany();
+    Category category1 = getModel1Factory().createCategory();
+
+    resource1.getContents().add(company1);
+    company1.getCategories().add(category1);
+
+    objects.add(company1);
+    objects.add(category1);
+
+    company1.setName("TEST");
+
+    transaction.commit();
+    System.out.println(category1.eClass().getEPackage().getNsURI());
+
+    CDOQuery query = transaction.createQuery("TEST", "QUERYSTRING");
+    query.setParameter("context", getModel1Package().getCategory());
+
+    List<Category> result = query.getResult(Category.class);
+    assertEquals(1, result.size());
+    assertEquals(category1, result.get(0));
+
+    transaction.close();
+    session.close();
+  }
+
+  public void testObjyStoreQueryCancel_successful() throws Exception
+  {
+    //skipUnlessConfig(MEM);
+
+	    CDOTransaction transaction = initialize(500);
+	    CDOQuery query = transaction.createQuery("TEST", "QUERYSTRING");
+	    query.setParameter("sleep", 1000L);
+	    final CloseableIterator<Object> result = query.getResultAsync(Object.class);
+	    result.close();
+
+	    new PollingTimeOuter()
+	    {
+	      @Override
+	      protected boolean successful()
+	      {
+	        return !getRepository().getQueryManager().isRunning(((CDOQueryResultIteratorImpl<?>)result).getQueryID());
+	      }
+	    }.assertNoTimeOut();
+
+	    CDOSession session = transaction.getSession();
+	    transaction.close();
+	    session.close();
+  }
+
+  public void testObjyStoreQueryCancel_ViewClose() throws Exception
+  {
+    //skipUnlessConfig(MEM);
+
+	    CDOTransaction transaction = initialize(500);
+	    CDOQuery query = transaction.createQuery("TEST", "QUERYSTRING");
+	    query.setParameter("sleep", 1000L);
+	    final CloseableIterator<Object> result = query.getResultAsync(Object.class);
+	    CDOSession session = transaction.getSession();
+	    transaction.close();
+	    new PollingTimeOuter()
+	    {
+	      @Override
+	      protected boolean successful()
+	      {
+	        return !getRepository().getQueryManager().isRunning(((CDOQueryResultIteratorImpl<?>)result).getQueryID());
+	      }
+	    }.assertNoTimeOut();
+
+	    session.close();
+  }
+
+  public void testObjyStoreQueryCancel_SessionClose() throws Exception
+  {
+    //skipUnlessConfig(MEM);
+
+    CDOTransaction transaction = initialize(500);
+    CDOQuery query = transaction.createQuery("TEST", "QUERYSTRING");
+    query.setParameter("sleep", 1000L);
+    final CloseableIterator<Object> result = query.getResultAsync(Object.class);
+    transaction.getSession().close();
+
+    new PollingTimeOuter()
+    {
+      @Override
+      protected boolean successful()
+      {
+        return !getRepository().getQueryManager().isRunning(((CDOQueryResultIteratorImpl<?>)result).getQueryID());
+      }
+    }.assertNoTimeOut();
+  }
+
+  public void testObjyStoreQueryAsync_UnsupportedLanguage() throws Exception
+  {
+    CDOTransaction transaction = initialize(100);
+    CDOQuery query = transaction.createQuery("TESTss", "QUERYSTRING");
+
+    try
+    {
+      CloseableIterator<Object> result = query.getResultAsync(Object.class);
+      result.hasNext();
+      fail("Should throw an exception");
+    }
+    catch (Exception expected)
+    {
+    }
+  }
+
+  public void testObjyStoreQuerySync_UnsupportedLanguage() throws Exception
+  {
+    CDOTransaction transaction = initialize(100);
+    CDOQuery query = transaction.createQuery("TESTss", "QUERYSTRING");
+
+    try
+    {
+      query.getResult(Object.class);
+      fail("Should throw an exception");
+    }
+    catch (Exception expected)
+    {
+    }
+  }
+
+  private CDOTransaction initialize(int number)
+  {
+    CDOSession session = openSession();
+    CDOTransaction transaction = session.openTransaction();
+    CDOResource resource1 = transaction.createResource("/test1");
+
+    for (int i = 0; i < number; i++)
+    {
+      Category category1 = getModel1Factory().createCategory();
+      resource1.getContents().add(category1);
+    }
+
+    transaction.commit();
+    return transaction;
+  }
+}