core/org.eclipse.smila.importing.state.objectstore.test/code/src/org/eclipse/smila/importing/state/objectstore/test/TestObjectStoreVisitedLinksService.java

/*********************************************************************************************************************
 * Copyright (c) 2008, 2012 Attensity Europe GmbH and brox IT Solutions GmbH. All rights reserved. This program and the
 * accompanying materials are made available under the terms of the Eclipse Public License v1.0 which accompanies this
 * distribution, and is available at http://www.eclipse.org/legal/epl-v10.html
 **********************************************************************************************************************/
package org.eclipse.smila.importing.state.objectstore.test;

import java.util.Collection;

import org.eclipse.smila.importing.VisitedLinksService;
import org.eclipse.smila.test.DeclarativeServiceTestCase;

/**
 * Unit test cases for the ObjectStore based VisitedLinksService implementation.
 */
public class TestObjectStoreVisitedLinksService extends DeclarativeServiceTestCase {
  /** the service under test. */
  private VisitedLinksService _service;

  @Override
  protected void setUp() throws Exception {
    super.setUp();
    _service = getService(VisitedLinksService.class);
    _service.clearAll();
  }

  /** test checking of new objects. */
  public void testCheckNewObject() throws Exception {
    final String recordId = "testCheckNewObject";
    assertFalse(_service.isVisited("s", recordId, "j1"));    
    assertFalse(_service.checkAndMarkVisited("s", recordId, "j1", "h1"));
        
    // now visited
    assertTrue(_service.isVisited("s", recordId, "j1"));
    // ... but still not visited when "repeating same bulk"
    assertFalse(_service.checkAndMarkVisited("s", recordId, "j1", "h1"));
    // ... or for a different job run
    assertFalse(_service.isVisited("s", recordId, "j2"));
  }

  /** test visited for different bulks. */
  public void testNewBulk() throws Exception {
    final String recordId = "testAddNewObject";
    _service.markAsVisited("s", recordId, "j1", "h1");
    assertTrue(_service.isVisited("s", recordId, "j1"));
    // not visited when "repeating same bulk"
    assertFalse(_service.checkAndMarkVisited("s", recordId, "j1", "h1"));
    // ... but visited for a different bulk
    assertTrue(_service.checkAndMarkVisited("s", recordId, "j1", "h2"));
  }

  /** test updating objects. */
  public void testUpdateObject() throws Exception {
    final String recordId = "testUpdateObject";
    _service.markAsVisited("s", recordId, "j1", "h1");
    // next job run
    assertFalse(_service.isVisited("s", recordId, "j2"));
    assertFalse(_service.checkAndMarkVisited("s", recordId, "j2", "h2"));
    _service.markAsVisited("s", recordId, "j2", "h2");
    assertTrue(_service.isVisited("s", recordId, "j2"));
    assertFalse(_service.checkAndMarkVisited("s", recordId, "j2", "h2"));
  }

  /** test clearing a single source. */
  public void testClearSource() throws Exception {
    final String recordId = "testClearSource";
    _service.markAsVisited("s1", recordId, "j1", "h1");
    _service.markAsVisited("s2", recordId, "j1", "h1");
    _service.clearSource("s1");
    assertFalse(_service.isVisited("s1", recordId, "j1"));
    assertTrue(_service.isVisited("s2", recordId, "j1"));
    assertFalse(_service.checkAndMarkVisited("s1", recordId, "j1", "h1"));    
    assertTrue(_service.checkAndMarkVisited("s2", recordId, "j1", "h2"));
  }

  /** test clearing the complete delta service. */
  public void testClearAll() throws Exception {
    final String recordId = "testClearAll";
    _service.markAsVisited("s1", recordId, "j1", "h1");
    _service.markAsVisited("s2", recordId, "j1", "h1");
    _service.clearAll();
    assertFalse(_service.isVisited("s1", recordId, "j1"));
    assertFalse(_service.isVisited("s2", recordId, "j1"));
    assertFalse(_service.checkAndMarkVisited("s1", recordId, "j1", "h1"));
    assertFalse(_service.checkAndMarkVisited("s2", recordId, "j1", "h1"));
  }

  /** test getting source ids. */
  public void testGetSourceIdsInitial() throws Exception {
    final Collection<String> sourceIds = _service.getSourceIds();
    assertTrue(sourceIds.isEmpty());
  }

  /** test getting source ids. */
  public void testGetSourceIds() throws Exception {
    final String recordId = "testGetSourceIds";
    _service.markAsVisited("s1", recordId, "j1", "h1");
    _service.markAsVisited("s2", recordId, "j1", "h1");
    Collection<String> sourceIds = _service.getSourceIds();
    assertEquals(2, sourceIds.size());
    assertTrue(sourceIds.contains("s1"));
    assertTrue(sourceIds.contains("s2"));
    _service.clearSource("s1");
    sourceIds = _service.getSourceIds();
    assertEquals(1, sourceIds.size());
    assertTrue(sourceIds.contains("s2"));
    _service.clearSource("s2");
    sourceIds = _service.getSourceIds();
    assertTrue(sourceIds.isEmpty());
  }

}