Added missing org.simantics.db.{tests,testing} plug-ins.

[simantics/platform.git] / tests / org.simantics.db.tests / src / org / simantics / db / tests / api / readGraph / getPrincipalTypes / GetPrincipalTypesTest1.java

diff --git a/tests/org.simantics.db.tests/src/org/simantics/db/tests/api/readGraph/getPrincipalTypes/GetPrincipalTypesTest1.java b/tests/org.simantics.db.tests/src/org/simantics/db/tests/api/readGraph/getPrincipalTypes/GetPrincipalTypesTest1.java
new file mode 100644 (file)
index 0000000..21952cb
--- /dev/null
+++ b/tests/org.simantics.db.tests/src/org/simantics/db/tests/api/readGraph/getPrincipalTypes/GetPrincipalTypesTest1.java
@@ -0,0 +1,53 @@
+/*******************************************************************************
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management
+ * in Industry THTH ry.
+ * 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:
+ *     VTT Technical Research Centre of Finland - initial API and implementation
+ *******************************************************************************/
+package org.simantics.db.tests.api.readGraph.getPrincipalTypes;
+
+import java.util.Collection;
+
+import org.simantics.db.ReadGraph;
+import org.simantics.db.Resource;
+import org.simantics.db.WriteGraph;
+import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.testing.base.WriteReadTest;
+
+public class GetPrincipalTypesTest1 extends WriteReadTest {
+
+    private Resource type1;
+    private Resource type2;
+    private Resource instance1;
+    
+    @Override
+    protected void write(WriteGraph graph) throws DatabaseException {
+        type1 = graph.newResource();
+        graph.claim(type1, L0.Inherits, L0.Entity);
+
+        type2 = graph.newResource();
+        graph.claim(type2, L0.Inherits, L0.Entity);
+  
+        instance1 = graph.newResource();
+        graph.claim(instance1, L0.InstanceOf,type1);
+        graph.claim(instance1, L0.InstanceOf,type2);
+ 
+    }
+    
+    @Override
+    protected void read(ReadGraph graph) throws DatabaseException {
+       
+        Collection<Resource> result = graph.getPrincipalTypes(instance1);
+        assertNotNull("Null not allowed", result);
+        int resultsSize = result.size();
+        assertTrue("Null result.", (result != null));
+        assertTrue("Wrong types", ((result.contains(type1) && (result.contains(type2)))));
+        assertTrue("Wrong number of results: " + resultsSize,
+                       (resultsSize == 2));
+    }
+}