package com.atlassian.johnson.event;
   
   import com.atlassian.johnson.Johnson;
   import org.junit.After;
   import org.junit.Before;
   import org.junit.Test;
   
   import java.util.function.Supplier;
   
  import static com.atlassian.johnson.event.EventLevel.ERROR;
  import static com.atlassian.johnson.event.EventLevel.FATAL;
  import static com.atlassian.johnson.event.EventLevel.WARNING;
  import static org.hamcrest.Matchers.is;
  import static org.hamcrest.Matchers.notNullValue;
  import static org.junit.Assert.assertThat;
  
  public class EventLevelsTest {
  
      @Before
      @After
      public void clearJohnsonConfig() {
          Johnson.terminate();
      }
  
      @Test(expected = IllegalStateException.class)
      public void shouldNotBeAbleToGetFatalLevelIfJohnsonNotInitialised() {
          // Invoke
          EventLevels.fatal();
      }
  
      @Test
      public void shouldBeAbleToGetErrorLevelIfJohnsonInitialised() {
          assertLevel(EventLevels::error, ERROR);
      }
  
      @Test
      public void shouldBeAbleToGetFatalLevelIfJohnsonInitialised() {
          assertLevel(EventLevels::fatal, FATAL);
      }
  
      @Test
      public void shouldBeAbleToGetWarningLevelIfJohnsonInitialised() {
          assertLevel(EventLevels::warning, WARNING);
      }
  
      private static void assertLevel(final Supplier<EventLevel> eventLevelSupplier, final String expectedLevelName) {
          // Set up
          Johnson.initialize("test-johnson-config.xml");
  
          // Invoke
          final EventLevel error = eventLevelSupplier.get();
  
          // Check
          assertThat(error, notNullValue());
          assertThat(error.getLevel(), is(expectedLevelName));
      }
  }