package com.change_vision.jude.api.test.model;

import junit.framework.Test;

import com.change_vision.jude.api.inf.model.IERAttribute;
import com.change_vision.jude.api.inf.model.IEREntity;
import com.change_vision.jude.api.inf.model.IERIndex;
import com.change_vision.jude.api.inf.model.IERModel;
import com.change_vision.jude.api.inf.model.IERSchema;

public class IERIndexTest extends ITestCase {
	
	private IERSchema schema = null;
	
    public static Test suite() {
        return suite("testModel/judeAPITest/IERIndexTest.jude", IERIndexTest.class);
    }
    
    protected void setUp() {
        IERModel erModel = (IERModel)getElement(project.getOwnedElements(), "ER Model");
        schema = erModel.getSchemata()[0];
    }
    
    public void testGetParentEntity() {
        IEREntity entity = getEntity(schema.getEntities(), "Entity0");
        IERIndex[] erIndexes = entity.getERIndices();
        for (int i = 0; i < erIndexes.length; i++) {
        	IERIndex erIndex = erIndexes[i];
        	assertTrue(entity == erIndex.getParentEREntity());
        }
    }
    
    public void testGetERAttributes() {
        IEREntity entity = getEntity(schema.getEntities(), "Entity0");
        IERIndex[] erIndexes = entity.getERIndices();
        IERAttribute attribute0 = 
        	(IERAttribute)getElement(entity.getPrimaryKeys(), "Attribute0");
        IERAttribute attribute1 = 
        	(IERAttribute)getElement(entity.getPrimaryKeys(), "Attribute1");
        IERAttribute attribute2 = 
        	(IERAttribute)getElement(entity.getPrimaryKeys(), "Attribute2");
        IERAttribute attribute3 = 
        	(IERAttribute)getElement(entity.getNonPrimaryKeys(), "Attribute3");
        IERAttribute attribute4 = 
        	(IERAttribute)getElement(entity.getNonPrimaryKeys(), "Attribute4");
        IERAttribute attribute5 = 
        	(IERAttribute)getElement(entity.getNonPrimaryKeys(), "Attribute5");
        for (int i = 0; i < erIndexes.length; i++) {
        	IERIndex erIndex = erIndexes[i];
        	if ("Index1".equals(erIndex.getName())) {
        		assertTrue(erIndex.getERAttributes().length == 3);
        		assertTrue(erIndex.getERAttributes()[0] == attribute0);
        		assertTrue(erIndex.getERAttributes()[1] == attribute3);
        		assertTrue(erIndex.getERAttributes()[2] == attribute1);
        	} else if("Index2".equals(erIndex.getName())) {
        		assertTrue(erIndex.getERAttributes().length == 2);
        		assertTrue(erIndex.getERAttributes()[0] == attribute1);
        		assertTrue(erIndex.getERAttributes()[1] == attribute4);
        	} else if("Index3".equals(erIndex.getName())) {
        		assertTrue(erIndex.getERAttributes().length == 2);
        		assertTrue(erIndex.getERAttributes()[0] == attribute2);
        		assertTrue(erIndex.getERAttributes()[1] == attribute5);
        	} else if("Index4".equals(erIndex.getName())) {
        		assertTrue(erIndex.getERAttributes().length == 6);
        		assertTrue(erIndex.getERAttributes()[0] == attribute0);
        		assertTrue(erIndex.getERAttributes()[1] == attribute1);
        		assertTrue(erIndex.getERAttributes()[2] == attribute2);
        		assertTrue(erIndex.getERAttributes()[3] == attribute3);
        		assertTrue(erIndex.getERAttributes()[4] == attribute4);
        		assertTrue(erIndex.getERAttributes()[5] == attribute5);
        	}
        }
    }
    
    public void testGetERRelationships() {
        IEREntity entity = getEntity(schema.getEntities(), "Entity0");
        IERIndex[] erIndexes = entity.getERIndices();
        for (int i = 0; i < erIndexes.length; i++) {
        	IERIndex erIndex = erIndexes[i];
        	if ("Index1".equals(erIndex.getName())) {
        		assertTrue(erIndex.getERRelationships().length == 1);
        		assertTrue("Identifying Relationship1".equals(erIndex.getERRelationships()[0].getName()));
        	} else if("Index2".equals(erIndex.getName())) {
        		assertTrue(erIndex.getERRelationships().length == 0);
        	}
        }
    }
    
    public void testIsUnique() {
        IEREntity entity = getEntity(schema.getEntities(), "Entity0");
        IERIndex[] erIndexes = entity.getERIndices();
        for (int i = 0; i < erIndexes.length; i++) {
        	IERIndex erIndex = erIndexes[i];
        	if ("Index1".equals(erIndex.getName())) {
        		assertTrue(erIndex.isUnique());
        	} else if("Index2".equals(erIndex.getName())) {
        		assertTrue(erIndex.isUnique());
        	} else if("Index3".equals(erIndex.getName())) {
        		assertFalse(erIndex.isUnique());
        	} else if("Index4".equals(erIndex.getName())) {
        		assertFalse(erIndex.isUnique());
        	}
        }
    }
    
    public void testGetKind() {
        IEREntity entity = getEntity(schema.getEntities(), "Entity0");
        IERIndex[] erIndexes = entity.getERIndices();
        for (int i = 0; i < erIndexes.length; i++) {
        	IERIndex erIndex = erIndexes[i];
        	if ("Index1".equals(erIndex.getName())) {
        		assertTrue("AK1".equals(erIndex.getKind()));
        	} else if("Index2".equals(erIndex.getName())) {
        		assertTrue("AK2".equals(erIndex.getKind()));
        	} else if("Index3".equals(erIndex.getName())) {
        		assertTrue("IE1".equals(erIndex.getKind()));
        	} else if("Index4".equals(erIndex.getName())) {
        		assertTrue("IE2".equals(erIndex.getKind()));
        	}
        }
    }
    
    private IEREntity getEntity(IEREntity[] entities, String name){
        for(int i = 0; i < entities.length; i++) {
            if(name.equals(entities[i].getName())){
                return entities[i];
            }
        }
        return null;
    }
}