LongMappedBufferTest.java

import org.junit.Assert;
import org.junit.Test;
import org.junit.BeforeClass;

import bloomtime.LongMappedBuffer;

import java.io.File;
import java.util.BitSet;
import java.util.Random;
import java.util.TreeSet;
import jelectrum.TimeRecord;

public class LongMappedBufferTest
{
  private static LongMappedBuffer map;

  @BeforeClass
  public static void loadMap()
    throws Exception
  {
    new File("/var/hdd/clash/test").mkdirs();

    map = new LongMappedBuffer(new File("/var/hdd/clash/test/longmappedbuffer-test"), 16L*1024L*1024L*1024L);
  }

  @Test
  public void testMapEdge()
  { 

    byte[] target = new byte[65536];

    for(long loc = LongMappedBuffer.MAP_SIZE - 65536 - 100; loc < LongMappedBuffer.MAP_SIZE + 65536 + 100; loc+=913)
    {
      
      byte[] buff = TestUtil.randomBytes(65536);
      map.putBytes(loc, buff);
      map.getBytes(loc, target);
      Assert.assertArrayEquals(buff, target);
    }
  }

  @Test
  public void testBitSet()
  {
    byte[] b = new byte[1];

    long pos = 91;
    map.putBytes(91, b);
    map.getBytes(91, b);
    Assert.assertEquals(0, b[0]);

    for(int i=0; i<8; i++)
    {
      map.setBit(91*8 + i);
    }
    map.getBytes(91, b);
    BitSet bs = BitSet.valueOf(b);
    for(int i=0; i<8; i++)
    {
      Assert.assertTrue(bs.get(i));
    }
  }

  /*@Test
  public void testBitSetBulk()
  {
    TimeRecord tr = new TimeRecord();

    TimeRecord.setSharedRecord(tr);
    
    Random rnd = new Random();

    for(int j=0; j<10; j++)
    {

      TreeSet<Long> bits = new TreeSet<Long>();

      for(int i=0; i<1000; i++)
      {
        long gb = rnd.nextInt(16);
        long pos = rnd.nextInt(1024 * 1024 * 1024) ;
        pos = pos + gb * 1024L * 1024L * 1024L;
        long bit =rnd.nextInt(8);
        pos = bit + pos * 8;
        bits.add(pos);
      }
      map.setBits(bits);
      tr.printReport(System.out);
      tr.reset();
    }
  }*/



}