/*
 * Copyright 2018-2022 Open Networking Foundation and others.  All rights reserved.
 *
 * 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 io.atomix.storage.journal.index;

import org.junit.Test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

/**
 * Sparse journal index test.
 */
public class SparseJournalIndexTest {
  @Test
  public void testSparseJournalIndex() throws Exception {
    JournalIndex index = new SparseJournalIndex(.2);
    assertNull(index.lookup(1));
    index.index(1, 2);
    assertNull(index.lookup(1));
    index.index(2, 4);
    index.index(3, 6);
    index.index(4, 8);
    index.index(5, 10);
    assertEquals(new Position(5, 10), index.lookup(5));
    index.index(6, 12);
    index.index(7, 14);
    index.index(8, 16);
    assertEquals(new Position(5, 10), index.lookup(8));
    index.index(9, 18);
    index.index(10, 20);
    assertEquals(new Position(10, 20), index.lookup(10));
    index.truncate(8);
    assertEquals(new Position(5, 10), index.lookup(8));
    assertEquals(new Position(5, 10), index.lookup(10));
    index.truncate(4);
    assertNull(index.lookup(4));
    assertNull(index.lookup(8));

    index = new SparseJournalIndex(.2);
    assertNull(index.lookup(100));
    index.index(101, 2);
    assertNull(index.lookup(1));
    index.index(102, 4);
    index.index(103, 6);
    index.index(104, 8);
    index.index(105, 10);
    assertEquals(new Position(105, 10), index.lookup(105));
    index.index(106, 12);
    index.index(107, 14);
    index.index(108, 16);
    assertEquals(new Position(105, 10), index.lookup(108));
    index.index(109, 18);
    index.index(110, 20);
    assertEquals(new Position(110, 20), index.lookup(110));
    index.truncate(108);
    assertEquals(new Position(105, 10), index.lookup(108));
    assertEquals(new Position(105, 10), index.lookup(110));
    index.truncate(104);
    assertNull(index.lookup(104));
    assertNull(index.lookup(108));
  }
}