package io.jooby;

import com.typesafe.config.Config;

import com.typesafe.config.ConfigFactory;

import org.junit.jupiter.api.Test;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.util.Arrays;

import java.util.Collections;

import java.util.HashMap;

import java.util.Map;

import java.util.Properties;

import java.util.function.BiConsumer;

import static java.util.Arrays.asList;

import static org.junit.jupiter.api.Assertions.assertEquals;

public class EnvironmentTest {

@Test

public void defaultEnv() {

env("foo", (env, conf) -> {

assertEquals("[dev]\n"

+ "└── system properties\n"

+ " └── env variables\n"

+ " └── classpath://env/foo/application.conf\n"

+ " └── defaults", env.toString());

assertEquals("[dev]", env.getActiveNames().toString());

assertEquals(System.getProperty("user.dir"), conf.getString("user.dir"));

assertEquals("bar", conf.getString("foo"));

assertEquals(asList("a", "b", "c"), conf.getStringList("letters"));

});

env("foo", mapOf("application.env", "PROD"), (env, conf) -> {

assertEquals("[prod]\n"

+ "└── system properties\n"

+ " └── env variables\n"

+ " └── classpath://env/foo/application.prod.conf\n"

+ " └── classpath://env/foo/application.conf\n"

+ " └── defaults", env.toString());

assertEquals("[prod]", env.getActiveNames().toString());

assertEquals("bazz", conf.getString("foo"));

assertEquals(asList("a", "b", "c"), conf.getStringList("letters"));

});

env("foo", mapOf("application.env", "Test, bar"), (env, conf) -> {

assertEquals("[test, bar]\n"

+ "└── system properties\n"

+ " └── env variables\n"

+ " └── classpath://env/foo/application.test.conf\n"

+ " └── classpath://env/foo/application.bar.conf\n"

+ " └── defaults", env.toString());

assertEquals("[test, bar]", env.getActiveNames().toString());

assertEquals("test", conf.getString("foo"));

assertEquals(asList("d"), conf.getStringList("letters"));

});

env("empty", (env, conf) -> {

assertEquals("[dev]", env.getActiveNames().toString());

});

}

@Test

public void envfromfs() {

Path basedir = Paths.get(System.getProperty("user.dir"));

if (Files.exists(basedir.resolve("jooby"))) {

// maven vs IDE

basedir = basedir.resolve("jooby");

}

Path reldir = Paths.get("src/test/resources/env/foo");

basedir = basedir.resolve("src").resolve("test").resolve("resources").resolve("env")

.resolve("foo");

Environment env = Environment

.loadEnvironment(new EnvironmentOptions().setBasedir(basedir).setActiveNames("prod"));

assertEquals("bazz", env.getConfig().getString("foo"));

System.out.println(env.toString());

assertEquals("[prod]\n"

+ "└── system properties\n"

+ " └── env variables\n"

+ " └── "+ reldir.resolve("application.prod.conf")+"\n"

+ " └── "+ reldir.resolve("application.conf")+"\n"

+ " └── defaults", env.toString());

}

@Test

public void objectLookup() {

Environment env = new Environment(getClass().getClassLoader(),

ConfigFactory.parseMap(mapOf("h.pool", "1", "h.db.pool", "2")), "test");

assertEquals("1", env.getConfig().getString("h.pool"));

assertEquals("2", env.getConfig().getString("h.db.pool"));

assertEquals(mapOf("pool", "2"), env.getConfig().getConfig("h.db").root().unwrapped());

}

@Test

public void args() {

Map<String, String> args = Jooby.parseArguments("foo", " bar = ");

assertEquals("{application.env=foo, bar=}", args.toString());

assertEquals(Collections.emptyMap(), Jooby.parseArguments());

assertEquals(Collections.emptyMap(), Jooby.parseArguments((String[]) null));

}

private void env(String dir, BiConsumer<Environment, Config> consumer) {

env(dir, Collections.emptyMap(), consumer);

}

@Test

public void flattenProperties() {

Config config = ConfigFactory

.parseMap(mapOf("k", "v", "root", mapOf("map", mapOf("key1", 1, "key2", "2", "list",

Arrays.asList("a", "b")))));

Environment environment = new Environment(getClass().getClassLoader(), config);

assertEquals(mapOf("root.map.key1", "1", "root.map.key2", "2", "root.map.list", "a, b"), environment.getProperties("root"));

assertEquals(mapOf("p.map.key1", "1", "p.map.key2", "2", "p.map.list", "a, b"), environment.getProperties("root", "p"));

}

private void env(String dir, Map<String, Object> args, BiConsumer<Environment, Config> consumer) {

Properties sysprops = new Properties();

sysprops.putAll(System.getProperties());

try {

String[] names = args.getOrDefault("application.env", "dev").toString().split(",");

args.forEach((k, v) -> System.setProperty(k, v.toString()));

Environment env = Environment.loadEnvironment(new EnvironmentOptions()

.setBasedir("env/" + dir)

.setActiveNames(names)

);

consumer.accept(env, env.getConfig());

} finally {

System.setProperties(sysprops);

}

}

private Map<String, Object> mapOf(Object... values) {

Map<String, Object> hash = new HashMap<>();

for (int i = 0; i < values.length; i += 2) {

hash.put(values[i].toString(), values[i + 1]);

}

return hash;

}

}