SupplierInstanceBindingTest.java

/*
 * Copyright (c) 2017, 2022 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package org.glassfish.jersey.inject.hk2;

import java.util.concurrent.atomic.AtomicInteger;
import java.util.function.Supplier;

import javax.inject.Singleton;

import org.glassfish.jersey.internal.inject.InjectionManager;

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotSame;
import static org.junit.jupiter.api.Assertions.assertSame;

/**
 * Tests that {@link java.util.function.Supplier} can be registered as a instance-factory.
 *
 * @author Petr Bouda
 */
public class SupplierInstanceBindingTest {

    private static class MySupplier implements Supplier<Integer> {

        private final AtomicInteger counter = new AtomicInteger();

        @Override
        public Integer get() {
            return counter.incrementAndGet();
        }
    }

    private InjectionManager injectionManager;

    @BeforeEach
    public void setup() {
        injectionManager = BindingTestHelper.createInjectionManager();
    }

    @AfterEach
    public void teardown() {
        injectionManager.shutdown();
    }


    @Test
    public void testInstanceFactory() {
        MySupplier supplier = new MySupplier();
        supplier.get();
        supplier.get();

        BindingTestHelper.bind(injectionManager, binder ->
                binder.bindFactory(supplier).to(Integer.class));

        assertEquals((Integer) 3, injectionManager.getInstance(Integer.class));
        assertEquals((Integer) 4, injectionManager.getInstance(Integer.class));
    }

    @Test
    public void testMessages() {
        BindingTestHelper.bind(injectionManager, binder -> {
            binder.bindFactory(new SupplierGreeting()).to(Greeting.class);
            binder.bindAsContract(Conversation.class);
        });

        Conversation conversation = injectionManager.getInstance(Conversation.class);
        assertEquals(CzechGreeting.GREETING, conversation.greeting.getGreeting());
        assertEquals(CzechGreeting.GREETING, conversation.greetingSupplier.get().getGreeting());
    }

    @Test
    public void testSupplierSingletonInstancePerLookup() {
        BindingTestHelper.bind(injectionManager, binder -> {
            binder.bindFactory(new SupplierGreeting()).to(Greeting.class);
            binder.bindAsContract(Conversation.class);
        });

        Greeting greeting1 = injectionManager.getInstance(Conversation.class).greeting;
        Greeting greeting2 = injectionManager.getInstance(Conversation.class).greeting;
        Greeting greeting3 = injectionManager.getInstance(Conversation.class).greeting;

        assertNotSame(greeting1, greeting2);
        assertNotSame(greeting2, greeting3);

        Supplier<Greeting> supplier1 = injectionManager.getInstance(Conversation.class).greetingSupplier;
        Supplier<Greeting> supplier2 = injectionManager.getInstance(Conversation.class).greetingSupplier;
        Supplier<Greeting> supplier3 = injectionManager.getInstance(Conversation.class).greetingSupplier;

        assertSame(supplier1, supplier2);
        assertSame(supplier2, supplier3);
    }

    @Test
    public void testSupplierSingletonInstanceSingleton() {
        BindingTestHelper.bind(injectionManager, binder -> {
            binder.bindFactory(SupplierGreeting.class, Singleton.class).to(Greeting.class).in(Singleton.class);
            binder.bindAsContract(Conversation.class);
        });

        Greeting greeting1 = injectionManager.getInstance(Conversation.class).greeting;
        Greeting greeting2 = injectionManager.getInstance(Conversation.class).greeting;
        Greeting greeting3 = injectionManager.getInstance(Conversation.class).greeting;

        assertSame(greeting1, greeting2);
        assertSame(greeting2, greeting3);

        Supplier<Greeting> supplier1 = injectionManager.getInstance(Conversation.class).greetingSupplier;
        Supplier<Greeting> supplier2 = injectionManager.getInstance(Conversation.class).greetingSupplier;
        Supplier<Greeting> supplier3 = injectionManager.getInstance(Conversation.class).greetingSupplier;

        assertSame(supplier1, supplier2);
        assertSame(supplier2, supplier3);
    }

    @Test
    public void testSupplierBeanNamed() {
        BindingTestHelper.bind(injectionManager, binder -> {
            binder.bindFactory(new TestSuppliers.TestSupplier()).named(TestSuppliers.TEST).to(String.class);
            binder.bindFactory(new TestSuppliers.OtherTestSupplier()).named(TestSuppliers.OTHER_TEST).to(String.class);
            binder.bindAsContract(TestSuppliers.TargetSupplierBean.class);
        });

        TestSuppliers.TargetSupplierBean instance = injectionManager.getInstance(TestSuppliers.TargetSupplierBean.class);
        assertEquals(TestSuppliers.OTHER_TEST, instance.obj);
    }

    @Test
    public void testSupplierNamed() {
        BindingTestHelper.bind(injectionManager, binder -> {
            binder.bindFactory(new TestSuppliers.TestSupplier()).named(TestSuppliers.TEST).to(String.class);
            binder.bindFactory(new TestSuppliers.OtherTestSupplier()).named(TestSuppliers.OTHER_TEST).to(String.class);
            binder.bindAsContract(TestSuppliers.TargetSupplier.class);
        });

        TestSuppliers.TargetSupplier instance = injectionManager.getInstance(TestSuppliers.TargetSupplier.class);
        assertEquals(TestSuppliers.OTHER_TEST, instance.supplier.get());
    }
}