/*******************************************************************************
 * 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));
    }
}