text: /*
   * Copyright © 2020 WireGuard LLC. All Rights Reserved.
   * SPDX-License-Identifier: Apache-2.0
   */
  
  package com.wireguard.config;
  
  import com.wireguard.config.BadConfigException.Location;
  import com.wireguard.config.BadConfigException.Reason;
  import com.wireguard.config.BadConfigException.Section;
  
  import org.junit.AfterClass;
  import org.junit.BeforeClass;
  import org.junit.Test;
  
  import java.io.IOException;
  import java.io.InputStream;
  import java.util.HashMap;
  import java.util.Map;
  
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.fail;
  
  public class BadConfigExceptionTest {
      private static final Map<String, InputStream> CONFIG_MAP = new HashMap<>();
      private static final String[] CONFIG_NAMES = {
              "invalid-key",
              "invalid-number",
              "invalid-value",
              "missing-attribute",
              "missing-section",
              "syntax-error",
              "unknown-attribute",
              "unknown-section"
      };
  
      @AfterClass
      public static void closeStreams() {
          for (final InputStream inputStream : CONFIG_MAP.values()) {
              try {
                  inputStream.close();
              } catch (final IOException ignored) {
              }
          }
      }
  
      @BeforeClass
      public static void readConfigs() {
          for (final String config : CONFIG_NAMES) {
              CONFIG_MAP.put(config, BadConfigExceptionTest.class.getClassLoader().getResourceAsStream(config + ".conf"));
          }
      }
  
      @Test
      public void throws_correctly_with_INVALID_KEY_reason() {
          try {
              Config.parse(CONFIG_MAP.get("invalid-key"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.INVALID_KEY);
              assertEquals(e.getLocation(), Location.PUBLIC_KEY);
              assertEquals(e.getSection(), Section.PEER);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException thrown during test");
          }
      }
  
      @Test
      public void throws_correctly_with_INVALID_NUMBER_reason() {
          try {
              Config.parse(CONFIG_MAP.get("invalid-number"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.INVALID_NUMBER);
              assertEquals(e.getLocation(), Location.PERSISTENT_KEEPALIVE);
              assertEquals(e.getSection(), Section.PEER);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException thrown during test");
          }
      }
  
      @Test
      public void throws_correctly_with_INVALID_VALUE_reason() {
          try {
              Config.parse(CONFIG_MAP.get("invalid-value"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.INVALID_VALUE);
              assertEquals(e.getLocation(), Location.DNS);
              assertEquals(e.getSection(), Section.INTERFACE);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException throwing during test");
          }
      }
  
      @Test
      public void throws_correctly_with_MISSING_ATTRIBUTE_reason() {
          try {
              Config.parse(CONFIG_MAP.get("missing-attribute"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.MISSING_ATTRIBUTE);
              assertEquals(e.getLocation(), Location.PUBLIC_KEY);
              assertEquals(e.getSection(), Section.PEER);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException throwing during test");
          }
      }
  
      @Test
      public void throws_correctly_with_MISSING_SECTION_reason() {
          try {
              Config.parse(CONFIG_MAP.get("missing-section"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.MISSING_SECTION);
              assertEquals(e.getLocation(), Location.TOP_LEVEL);
              assertEquals(e.getSection(), Section.CONFIG);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException throwing during test");
          }
      }
  
      @Test
      public void throws_correctly_with_SYNTAX_ERROR_reason() {
          try {
              Config.parse(CONFIG_MAP.get("syntax-error"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.SYNTAX_ERROR);
              assertEquals(e.getLocation(), Location.TOP_LEVEL);
              assertEquals(e.getSection(), Section.PEER);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException throwing during test");
          }
      }
  
      @Test
      public void throws_correctly_with_UNKNOWN_ATTRIBUTE_reason() {
          try {
              Config.parse(CONFIG_MAP.get("unknown-attribute"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.UNKNOWN_ATTRIBUTE);
              assertEquals(e.getLocation(), Location.TOP_LEVEL);
              assertEquals(e.getSection(), Section.PEER);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException throwing during test");
          }
      }
  
      @Test
      public void throws_correctly_with_UNKNOWN_SECTION_reason() {
          try {
              Config.parse(CONFIG_MAP.get("unknown-section"));
              fail("Config parsing must fail in this test");
          } catch (final BadConfigException e) {
              assertEquals(e.getReason(), Reason.UNKNOWN_SECTION);
              assertEquals(e.getLocation(), Location.TOP_LEVEL);
              assertEquals(e.getSection(), Section.CONFIG);
          } catch (final IOException e) {
              e.printStackTrace();
              fail("IOException throwing during test");
          }
      }
  }
  