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));
  }

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

  @Test
  public void bigMultiInsert() {
    long[] key = new long[5000];
    long[] expected = new long[5000];
    for (int i = 0; i < 5000; i++) {
      map.put(i, i + 1);
      key[i] = i;
      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();
    }
  }
}