planetiler/core/src/test/java/com/onthegomap/flatmap/collections/LongLongMapTest.java

package com.onthegomap.flatmap.collections;

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

import java.nio.file.Path;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir;

public abstract class LongLongMapTest {

  protected LongLongMap map;

  @Test
  public void missingValue() {
    assertEquals(Long.MIN_VALUE, map.get(0));
  }

  @Test
  public void insertLookup() {
    map.put(1, 1);
    assertEquals(Long.MIN_VALUE, map.get(0));
    assertEquals(1, map.get(1));
    assertEquals(Long.MIN_VALUE, map.get(2));
  }

  @Test
  public void insertWithGaps() {
    map.put(1, 2);
    map.put(50, 3);
    map.put(500, 4);
    map.put(505, 5);
    assertEquals(Long.MIN_VALUE, map.get(0));
    assertEquals(2, map.get(1));
    assertEquals(Long.MIN_VALUE, map.get(2));
    assertEquals(Long.MIN_VALUE, map.get(49));
    assertEquals(3, map.get(50));
    assertEquals(Long.MIN_VALUE, map.get(51));
    assertEquals(Long.MIN_VALUE, map.get(300));
    assertEquals(Long.MIN_VALUE, map.get(499));
    assertEquals(4, map.get(500));
    assertEquals(Long.MIN_VALUE, map.get(501));
    assertEquals(5, map.get(505));
    assertEquals(Long.MIN_VALUE, map.get(506));
    assertEquals(Long.MIN_VALUE, map.get(1_000));
  }

  @Test
  public void insertMultiLookup() {
    map.put(1, 3);
    map.put(2, 4);
    map.put(1_000_000_000, Long.MAX_VALUE);
    assertEquals(Long.MIN_VALUE, map.get(0));
    assertEquals(Long.MIN_VALUE, map.get(3));
    assertArrayEquals(new long[]{3, 4, Long.MAX_VALUE, Long.MIN_VALUE},
      map.multiGet(new long[]{1, 2, 1_000_000_000, 3}));
  }

  @Test
  public void bigMultiInsert() {
    long[] key = new long[50000];
    long[] expected = new long[50000];
    for (int i = 0; i < 50000; i++) {
      map.put(i * 4, i + 1);
      key[i] = i * 4;
      expected[i] = i + 1;
    }

    long[] result = map.multiGet(key);

    assertArrayEquals(expected, result);
  }

  public static class SortedTableFileTest extends LongLongMapTest {

    @BeforeEach
    public void setup(@TempDir Path dir) {
      this.map = LongLongMap.newFileBackedSortedTable(dir.resolve("test-node-db-sorted"));
    }
  }

  public static class SortedTableMemoryTest extends LongLongMapTest {

    @BeforeEach
    public void setup() {
      this.map = LongLongMap.newInMemorySortedTable();
    }
  }

  public static class SparseArrayTest extends LongLongMapTest {

    @BeforeEach
    public void setup(@TempDir Path dir) {
      this.map = LongLongMap.newFileBackedSparseArray(dir.resolve("test-sparse-array"), 80, 100);
    }
  }

  public static class SparseArrayMemoryTest extends LongLongMapTest {

    @BeforeEach
    public void setup() {
      this.map = LongLongMap.newInMemorySparseArray(80, 100);
    }
  }

  public static class ArrayTest extends LongLongMapTest {

    @BeforeEach
    public void setup() {
      this.map = LongLongMap.newArrayBacked();
    }
  }

  public static class SqliteTest extends LongLongMapTest {

    @BeforeEach
    public void setup(@TempDir Path dir) {
      this.map = LongLongMap.newSqlite(dir.resolve("sqlite-long-long-map-test"));
    }
  }

  public static class SpareArray2Memory extends LongLongMapTest {

    @BeforeEach
    public void setup() {
      this.map = new LongLongMap.SparseArray2Memory();
    }
  }

  public static class SparseArray2 extends LongLongMapTest {

    @BeforeEach
    public void setup(@TempDir Path dir) {
      this.map = new LongLongMap.SparseArray2(dir.resolve("temp-sparse-array-2"));
    }
  }
}