/**
 * Copyright (C) 2013 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.engine.exec;

import static org.testng.Assert.assertNull;
import static org.testng.Assert.assertSame;

import java.util.Arrays;
import java.util.Collections;

import org.testng.annotations.Test;

import com.opengamma.engine.ComputationTargetSpecification;
import com.opengamma.engine.calcnode.CalculationJobResultItem;
import com.opengamma.engine.depgraph.DependencyNode;
import com.opengamma.engine.depgraph.impl.DependencyNodeFunctionImpl;
import com.opengamma.engine.depgraph.impl.DependencyNodeImpl;
import com.opengamma.engine.function.EmptyFunctionParameters;
import com.opengamma.engine.value.ValueProperties;
import com.opengamma.engine.value.ValuePropertyNames;
import com.opengamma.engine.value.ValueSpecification;
import com.opengamma.engine.view.AggregatedExecutionLog;
import com.opengamma.engine.view.ExecutionLog;
import com.opengamma.util.test.TestGroup;

/**
 * Tests the {@link DependencyNodeJobExecutionResultCache} class
 */
@Test(groups = TestGroup.UNIT)
public class DependencyNodeJobExecutionResultCacheTest {

  private ValueSpecification createValueSpec(final int id) {
    return new ValueSpecification("V", ComputationTargetSpecification.NULL, ValueProperties.with(ValuePropertyNames.FUNCTION, "Test").with("Id", Integer.toString(id)).get());
  }

  private DependencyNodeJobExecutionResult createExecutionResult() {
    return new DependencyNodeJobExecutionResult("Node", new CalculationJobResultItem(Collections.<ValueSpecification>emptySet(), Collections.<ValueSpecification>emptySet(),
        ExecutionLog.EMPTY), AggregatedExecutionLog.EMPTY);
  }

  public void testPutAndGet() {
    final DependencyNodeJobExecutionResultCache cache = new DependencyNodeJobExecutionResultCache();
    assertNull(cache.get(createValueSpec(1)));
    final DependencyNodeJobExecutionResult result = createExecutionResult();
    cache.put(createValueSpec(1), result);
    assertSame(cache.get(createValueSpec(1)), result);
  }

  private DependencyNode node(final ValueSpecification... outputs) {
    return new DependencyNodeImpl(DependencyNodeFunctionImpl.of("Mock", EmptyFunctionParameters.INSTANCE), ComputationTargetSpecification.NULL, Arrays.asList(outputs),
        Collections.<ValueSpecification, DependencyNode>emptyMap());
  }

  public void testGet() {
    final DependencyNodeJobExecutionResultCache cache = new DependencyNodeJobExecutionResultCache();
    assertNull(cache.get(node()));
    final DependencyNodeJobExecutionResult result = createExecutionResult();
    cache.put(createValueSpec(1), result);
    assertNull(cache.get(node(createValueSpec(2))));
    assertSame(cache.get(node(createValueSpec(1), createValueSpec(2))), result);
  }

}