/*
 * Copyright 2020 Global Biodiversity Information Facility (GBIF)
 *
 * 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 org.gbif.api.vocabulary;

import java.util.HashSet;
import java.util.Set;

import org.junit.jupiter.api.Test;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.annotation.JsonDeserialize;
import com.fasterxml.jackson.databind.annotation.JsonSerialize;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.junit.jupiter.api.Assertions.fail;

public class CountryTest {

  @Test
  public void testFromIsoCode() {
    assertEquals(Country.ARGENTINA, Country.fromIsoCode("ar"));
    assertEquals(Country.ARGENTINA, Country.fromIsoCode("AR"));
  }

  @Test
  public void testGetIso2LetterCode() {
    Set<String> codes = new HashSet<>();
    for (Country l : Country.values()) {
      assertNotNull(l.getIso2LetterCode());
      // make sure its upper case
      assertEquals(l.getIso2LetterCode().toUpperCase(), l.getIso2LetterCode());
      // make sure its unique
      assertFalse(codes.contains(l.getIso2LetterCode()));
      codes.add(l.getIso2LetterCode());
    }
    assertEquals("DE", Country.GERMANY.getIso2LetterCode());
    assertEquals("GB", Country.UNITED_KINGDOM.getIso2LetterCode());
  }

  @Test
  public void testGetIso3LetterCode() {
    Set<String> codes = new HashSet<>();
    for (Country l : Country.values()) {
      assertNotNull(l.getIso3LetterCode());
      // make sure its upper case
      assertEquals(l.getIso3LetterCode().toUpperCase(), l.getIso3LetterCode());
      // make sure its unique
      assertFalse(codes.contains(l.getIso3LetterCode()));
      codes.add(l.getIso3LetterCode());
    }
    assertEquals("GBR", Country.UNITED_KINGDOM.getIso3LetterCode());
    assertEquals("DEU", Country.GERMANY.getIso3LetterCode());
  }

  @Test
  public void testGetIsoNumericalCode() {
    Set<Integer> codes = new HashSet<>();
    for (Country l : Country.values()) {
      assertNotNull(l.getIsoNumericalCode());
      // make sure its unique
      assertFalse(codes.contains(l.getIsoNumericalCode()));
      codes.add(l.getIsoNumericalCode());
    }
    assertEquals("GBR", Country.UNITED_KINGDOM.getIso3LetterCode());
    assertEquals("DEU", Country.GERMANY.getIso3LetterCode());
  }

  @Test
  public void testIsCustomCode() {
    for (Country l : Country.values()) {
      if (l.isOfficial() && l != Country.KOSOVO) {
        assertFalse(Country.isCustomCode(l.getIso2LetterCode()));
        assertFalse(Country.isCustomCode(l.getIso3LetterCode()));
      } else {
        assertTrue(Country.isCustomCode(l.getIso2LetterCode()));
        assertTrue(Country.isCustomCode(l.getIso3LetterCode()));
      }
    }
  }

  @Test
  public void testIsOfficial() {
    for (Country l : Country.OFFICIAL_COUNTRIES) {
      assertTrue(l.isOfficial());
    }
    int officialCountries = Country.OFFICIAL_COUNTRIES.size();
    int allCountries = Country.values().length;
    assertTrue(allCountries > officialCountries);
    assertEquals(250, officialCountries);
  }

  @Test
  public void testgetTitle() {
    for (Country l : Country.values()) {
      assertNotNull(l.getTitle());
      assertTrue(l.getTitle().length() > 2);
    }
    assertEquals("United Kingdom of Great Britain and Northern Ireland", Country.UNITED_KINGDOM.getTitle());
    assertEquals("Germany", Country.GERMANY.getTitle());
  }

  /**
   * A container of Countries using 2 properties.
   */
  private static class Container {

    public Country country;
    public Country island; // verify that the names don't matter

    public Container(Country country, Country island) {
      this.country = country;
      this.island = island;
    }

    // used by serializer below
    public Container() {
    }
  }

  /**
   * A test bean for trying the non default name serialization.
   */
  private static class CountryNameBean {

    @JsonSerialize(using = Country.TitleSerializer.class)
    @JsonDeserialize(using = Country.TitleDeserializer.class)
    public Country island; // verify that the names don't matter

    public CountryNameBean(Country island) {
      this.island = island;
    }

    // used by serializer below
    public CountryNameBean() {}
  }

  @Test
  public void testCodeUniqueness() {
    Set<String> codes = new HashSet<>();
    for (Country c : Country.values()) {
      assertFalse(codes.contains(c.getIso2LetterCode()));
      assertFalse(codes.contains(c.getIso3LetterCode()));

      codes.add(c.getIso2LetterCode());
      codes.add(c.getIso3LetterCode());

      if (c.getIsoNumericalCode() != null) {
        assertFalse(codes.contains(c.getIsoNumericalCode().toString()));
        codes.add(c.getIsoNumericalCode().toString());
      }
    }
  }

  @Test
  public void testTitleUniqueness() {
    Set<String> names = new HashSet<>();
    for (Country c : Country.values()) {
      assertFalse(names.contains(c.getTitle()));
      names.add(c.getTitle());
    }
  }

  @Test
  public void testSerDe() {
    ObjectMapper mapper = new ObjectMapper();
    Container source = new Container(Country.DENMARK, Country.FALKLAND_ISLANDS);
    try {
      String json = mapper.writeValueAsString(source);
      System.out.println(json);
      Container target = mapper.readValue(json, Container.class);
      assertEquals(source.country, target.country, "country differs");
      assertEquals(source.island, target.island, "island differs");
    } catch (Exception e) {
      fail(e.getMessage());
    }
  }

  @Test
  public void testSerDe2() {
    ObjectMapper mapper = new ObjectMapper();
    CountryNameBean source = new CountryNameBean(Country.FALKLAND_ISLANDS);
    try {
      String json = mapper.writeValueAsString(source);
      System.out.println(json);
      CountryNameBean target = mapper.readValue(json, CountryNameBean.class);
      assertEquals(source.island, target.island, "island differs");
    } catch (Exception e) {
      fail(e.getMessage());
    }
  }
}