/* * Copyright 2016-2018 The OpenZipkin Authors * * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except * in compliance with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software distributed under the License * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express * or implied. See the License for the specific language governing permissions and limitations under * the License. */ package zipkin.http; import com.twitter.app.Flag; import com.twitter.app.GlobalFlag$; import java.util.Collection; import java.util.concurrent.atomic.AtomicBoolean; import org.junit.After; import org.junit.Test; import scala.Function0; import scala.Option; import scala.runtime.AbstractFunction0; import scala.runtime.BoxedUnit; import static org.assertj.core.api.Assertions.assertThat; import static scala.collection.JavaConverters.asJavaCollection; public class hostTest { @After public void resetGlobalFlags() { for (Flag<?> globalFlag: globalFlags()) globalFlag.reset(); } @Test public void defaultValue() { Option<Flag<?>> flagOption = GlobalFlag$.MODULE$.get("zipkin.http.host"); assertThat(flagOption.get().apply()).isEqualTo("localhost:9411"); } @Test public void letOverridesDefault() { final String override = "foo:9411"; final AtomicBoolean ran = new AtomicBoolean(); Function0<BoxedUnit> fn0 = new AbstractFunction0<BoxedUnit>() { @Override public BoxedUnit apply() { ran.set(true); // used to verify this block is executed. assertThat(host$.Flag.isDefined()).isTrue(); assertThat(host$.Flag.apply()).isEqualTo(override); return BoxedUnit.UNIT; } }; host$.Flag.let(override, fn0); assertThat(ran.get()).isTrue(); } @Test public void registersGlobal() { assertThat(globalFlags()) .extracting(f -> f.name()) .containsOnlyOnce("zipkin.http.host"); } Collection<Flag<?>> globalFlags() { return asJavaCollection(GlobalFlag$.MODULE$.getAll(getClass().getClassLoader())); } }