/persistence/play-java-jdbc/src/test/java/play/db/NamedDatabaseTest.java

/*
 * Copyright (C) Lightbend Inc. <https://www.lightbend.com>
 */

package play.db;

import java.util.Map;
import javax.inject.Inject;

import com.google.common.collect.ImmutableMap;
import com.google.inject.Guice;
import com.google.inject.Injector;

import org.junit.Rule;
import org.junit.rules.ExpectedException;
import org.junit.Test;

import play.ApplicationLoader.Context;
import play.Environment;
import play.inject.guice.GuiceApplicationBuilder;
import play.inject.guice.GuiceApplicationLoader;

import static org.hamcrest.CoreMatchers.*;
import static org.hamcrest.MatcherAssert.assertThat;

public class NamedDatabaseTest {

  @Rule public ExpectedException exception = ExpectedException.none();

  @Test
  public void bindDatabasesByName() {
    Map<String, Object> config =
        ImmutableMap.of(
            "db.default.driver", "org.h2.Driver",
            "db.default.url", "jdbc:h2:mem:default",
            "db.other.driver", "org.h2.Driver",
            "db.other.url", "jdbc:h2:mem:other");
    Injector injector = createInjector(config);
    assertThat(
        injector.getInstance(DefaultComponent.class).db.getUrl(), equalTo("jdbc:h2:mem:default"));
    assertThat(
        injector.getInstance(NamedDefaultComponent.class).db.getUrl(),
        equalTo("jdbc:h2:mem:default"));
    assertThat(
        injector.getInstance(NamedOtherComponent.class).db.getUrl(), equalTo("jdbc:h2:mem:other"));
  }

  @Test
  public void notBindDefaultDatabaseWithoutConfiguration() {
    Map<String, Object> config =
        ImmutableMap.of(
            "db.other.driver", "org.h2.Driver",
            "db.other.url", "jdbc:h2:mem:other");
    Injector injector = createInjector(config);
    assertThat(
        injector.getInstance(NamedOtherComponent.class).db.getUrl(), equalTo("jdbc:h2:mem:other"));
    exception.expect(com.google.inject.ConfigurationException.class);
    injector.getInstance(DefaultComponent.class);
  }

  @Test
  public void notBindNamedDefaultDatabaseWithoutConfiguration() {
    Map<String, Object> config =
        ImmutableMap.of(
            "db.other.driver", "org.h2.Driver",
            "db.other.url", "jdbc:h2:mem:other");
    Injector injector = createInjector(config);
    assertThat(
        injector.getInstance(NamedOtherComponent.class).db.getUrl(), equalTo("jdbc:h2:mem:other"));
    exception.expect(com.google.inject.ConfigurationException.class);
    injector.getInstance(NamedDefaultComponent.class);
  }

  @Test
  public void allowDefaultDatabaseNameToBeConfigured() {
    Map<String, Object> config =
        ImmutableMap.of(
            "play.db.default", "other",
            "db.other.driver", "org.h2.Driver",
            "db.other.url", "jdbc:h2:mem:other");
    Injector injector = createInjector(config);
    assertThat(
        injector.getInstance(DefaultComponent.class).db.getUrl(), equalTo("jdbc:h2:mem:other"));
    assertThat(
        injector.getInstance(NamedOtherComponent.class).db.getUrl(), equalTo("jdbc:h2:mem:other"));
    exception.expect(com.google.inject.ConfigurationException.class);
    injector.getInstance(NamedDefaultComponent.class);
  }

  @Test
  public void allowDbConfigKeyToBeConfigured() {
    Map<String, Object> config =
        ImmutableMap.of(
            "play.db.config", "databases",
            "databases.default.driver", "org.h2.Driver",
            "databases.default.url", "jdbc:h2:mem:default");
    Injector injector = createInjector(config);
    assertThat(
        injector.getInstance(DefaultComponent.class).db.getUrl(), equalTo("jdbc:h2:mem:default"));
    assertThat(
        injector.getInstance(NamedDefaultComponent.class).db.getUrl(),
        equalTo("jdbc:h2:mem:default"));
  }

  private Injector createInjector(Map<String, Object> config) {
    GuiceApplicationBuilder builder =
        new GuiceApplicationLoader().builder(new Context(Environment.simple(), config));
    return Guice.createInjector(builder.applicationModule());
  }

  public static class DefaultComponent {
    @Inject Database db;
  }

  public static class NamedDefaultComponent {
    @Inject
    @NamedDatabase("default")
    Database db;
  }

  public static class NamedOtherComponent {
    @Inject
    @NamedDatabase("other")
    Database db;
  }
}