package com.lhkbob.entreri;

import org.junit.Assert;
import org.junit.Test;

import com.lhkbob.entreri.component.IntComponent;

public class IteratorTest {
    @Test
    public void testDisabledComponents() {
        EntitySystem system = new EntitySystem();
        IntComponent cd = system.createDataInstance(TypeId.get(IntComponent.class));
        
        Entity e1 = system.addEntity();
        e1.add(TypeId.get(IntComponent.class)).setEnabled(true);
        Entity e2 = system.addEntity();
        e2.add(TypeId.get(IntComponent.class)).setEnabled(false);
        
        ComponentIterator it = new ComponentIterator(system);
        it.addRequired(cd);
        it.reset();
        
        int count = 0;
        while(it.next()) {
            count++;
            Assert.assertSame(e1, cd.getEntity());
        }
        Assert.assertEquals(1, count);
    }
    
    @Test
    public void testIgnoreEnabledComponents() {
        EntitySystem system = new EntitySystem();
        IntComponent cd = system.createDataInstance(TypeId.get(IntComponent.class));
        
        Entity e1 = system.addEntity();
        e1.add(TypeId.get(IntComponent.class)).setEnabled(true);
        Entity e2 = system.addEntity();
        e2.add(TypeId.get(IntComponent.class)).setEnabled(false);
        
        ComponentIterator it = new ComponentIterator(system);
        it.addRequired(cd)
          .setIgnoreEnabled(true)
          .reset();

        boolean hasE1 = false;
        boolean hasE2 = false;
        while(it.next()) {
            if (e1 == cd.getEntity()) {
                Assert.assertFalse(hasE1);
                hasE1 = true;
            } else if (e2 == cd.getEntity()) {
                Assert.assertFalse(hasE2);
                hasE2 = true;
            } else {
                Assert.fail();
            }
        }
        Assert.assertTrue(hasE1);
        Assert.assertTrue(hasE2);
    }
}

@@ -1,8 +1,1449 @@
+/*%0A * Entreri, an entity-component framework in Java%0A *%0A * Copyright (c) 2012, Michael Ludwig%0A * All rights reserved.%0A *%0A * Redistribution and use in source and binary forms, with or without modification,%0A * are permitted provided that the following conditions are met:%0A *%0A *     Redistributions of source code must retain the above copyright notice,%0A *         this list of conditions and the following disclaimer.%0A *     Redistributions in binary form must reproduce the above copyright notice,%0A *         this list of conditions and the following disclaimer in the%0A *         documentation and/or other materials provided with the distribution.%0A *%0A * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS %22AS IS%22 AND%0A * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED%0A * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE%0A * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR%0A * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES%0A * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;%0A * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON%0A * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT%0A * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS%0A * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.%0A */%0A
 package 

Add missing header file