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

import static org.testng.AssertJUnit.assertEquals;

import java.util.ArrayList;

import org.testng.annotations.Test;

import com.google.common.collect.Lists;
import com.opengamma.engine.target.ComputationTargetType;
import com.opengamma.engine.value.ValueProperties;
import com.opengamma.engine.value.ValuePropertyNames;
import com.opengamma.engine.value.ValueSpecification;
import com.opengamma.engine.view.cycle.ComputationCycleQuery;
import com.opengamma.id.UniqueId;
import com.opengamma.util.test.AbstractFudgeBuilderTestCase;
import com.opengamma.util.test.TestGroup;

/**
 * Test.
 */
@Test(groups = TestGroup.UNIT)
public class ComputationCacheQueryBuilderTest extends AbstractFudgeBuilderTestCase {

  @Test
  public void testEmptyQuery() {
    ComputationCycleQuery query = new ComputationCycleQuery();
    query.setCalculationConfigurationName("DEFAULT");
    query.setValueSpecifications(new ArrayList<ValueSpecification>());
    checkCycle(query);
  }

  @Test
  public void testSingleQuery() {
    ComputationCycleQuery query = new ComputationCycleQuery();
    query.setCalculationConfigurationName("DEFAULT");
    ValueSpecification spec = ValueSpecification.of("SomeValue", ComputationTargetType.PRIMITIVE, UniqueId.of("SomeScheme", "SomeValue"), ValueProperties
        .with(ValuePropertyNames.FUNCTION, "SomeFunc").with(ValuePropertyNames.CURRENCY, "USD").get());
    query.setValueSpecifications(Lists.newArrayList(spec));
    checkCycle(query);
  }

  @Test
  public void testMultipleQuery() {
    ComputationCycleQuery query = new ComputationCycleQuery();
    query.setCalculationConfigurationName("DEFAULT");
    ValueSpecification spec = ValueSpecification.of("SomeValue", ComputationTargetType.PRIMITIVE, UniqueId.of("SomeScheme", "SomeValue"),
        ValueProperties.with(ValuePropertyNames.FUNCTION, "SomeFunc").with(ValuePropertyNames.CURRENCY, "USD").get());
    ValueSpecification spec2 = ValueSpecification.of("SomeOtherValue", ComputationTargetType.PRIMITIVE, UniqueId.of("SomeScheme", "SomeOtherValue"),
        ValueProperties.with(ValuePropertyNames.FUNCTION, "SomeOtherFunc").with(ValuePropertyNames.CURRENCY, "USD").get());
    query.setValueSpecifications(Lists.newArrayList(spec, spec2));

    checkCycle(query);
  }

  private void checkCycle(ComputationCycleQuery query) {
    ComputationCycleQuery cycled = cycleObject(ComputationCycleQuery.class, query);

    assertEquals(query.getCalculationConfigurationName(), cycled.getCalculationConfigurationName());
    assertEquals(query.getValueSpecifications(), cycled.getValueSpecifications());
  }
}