package org.apidesign.component;

import java.util.Iterator;

import org.junit.BeforeClass;

import org.junit.Test;

import org.netbeans.junit.MockServices;

import static org.junit.Assert.*;

// BEGIN: lookup.mockservices.serviceloader

public class InjectionViaServiceLoaderTest {

    public InjectionViaServiceLoaderTest() {

    }

    @BeforeClass

    public static void setUpClass() throws Exception {

        // this is a support for "dynamic" registration of any class

        // into the global pool of Lookup.getDefault() and java.util.ServiceLoader

        // either in @BeforeClass or @Before register 

        // classes of the mock instances that you want your code 

        // to find

        MockServices.setServices(ImplOne.class, ImplTwo.class);

    }

    @Test

    public void singleSlot() {

        InjectionSlot result = InjectionViaServiceLoader.singleSlot();

        assertNotNull("Some result found", result);

        assertEquals("The first is ImplOne", ImplOne.class, result.getClass());

    }

    @Test

    public void multiSlot() {

        Iterator<? extends InjectionSlot> it = InjectionViaServiceLoader.multiSlot().iterator();

        assertTrue("Has at least one", it.hasNext());

        assertEquals("The first is ImplOne", ImplOne.class, it.next().getClass());

        assertTrue("Has two", it.hasNext());

        assertEquals("The second is ImplTwo", ImplTwo.class, it.next().getClass());

        assertFalse("No other instance registered", it.hasNext());

    }

    public static final class ImplOne extends InjectionSlot {

    }

    public static final class ImplTwo extends InjectionSlot {

    }

}

// END: lookup.mockservices.serviceloader