private ImmutableMap<Class<?>, Object> generateComponents() {
   ImmutableMap.Builder<Class<?>, Object> builder = ImmutableMap.builder();
   for (Map.Entry<Class<?>, Object> entry :
       InterestRateMockSources.generateBaseComponents().entrySet()) {
     Class<?> key = entry.getKey();
     if (key.equals(SecuritySource.class)) {
       appendSecuritySource((SecuritySource) entry.getValue());
     }
     builder.put(key, entry.getValue());
   }
   return builder.build();
 }
  public void testPricingMethodCanBeCaptured() {

    // Test that we can capture results when we have trades/securities involved
    AvailableOutputs availableOutputs = createAvailableOutputs();
    AvailableImplementations availableImplementations = createAvailableImplementations();

    ViewFactory viewFactory =
        EngineTestUtils.createViewFactory(
            InterestRateMockSources.generateBaseComponents(),
            availableOutputs,
            availableImplementations);

    // Run view
    View view = viewFactory.createView(createIrsPricerConfig(), InterestRateSwapSecurity.class);

    VersionCorrection versionCorrection =
        ThreadLocalServiceContext.getInstance()
            .get(VersionCorrectionProvider.class)
            .getConfigVersionCorrection();
    CycleArguments cycleArguments =
        CycleArguments.builder(InterestRateMockSources.createMarketDataEnvironment())
            .versionCorrection(versionCorrection)
            .captureInputs(true)
            .build();

    Results run = view.run(cycleArguments, ImmutableList.of(createFixedVsLibor3mSwap()));
    Result<Object> pvResult = run.get(0, 0).getResult();
    if (!pvResult.isSuccess()) {
      fail(pvResult.toString());
    }
    Result<Object> pv01Result = run.get(0, 1).getResult();
    if (!pv01Result.isSuccess()) {
      fail(pv01Result.toString());
    }

    // Capture results
    ViewResultsSerializer serializer = new ViewResultsSerializer(run);

    // Write inputs to stream
    ByteArrayOutputStream inputsBaos = new ByteArrayOutputStream(1_000_000);
    serializer.serializeViewInputs(inputsBaos);

    ByteArrayOutputStream outputsBaos = new ByteArrayOutputStream(1_000_000);
    serializer.serializeViewOutputs(outputsBaos);

    ViewResultsDeserializer inputsDeserializer =
        new ViewResultsDeserializer(new ByteArrayInputStream(inputsBaos.toByteArray()));
    ViewInputs viewInputs = inputsDeserializer.deserialize(ViewInputs.class);
    assertThat(viewInputs, is(notNullValue()));

    ViewResultsDeserializer outputsDeserializer =
        new ViewResultsDeserializer(new ByteArrayInputStream(outputsBaos.toByteArray()));
    ViewOutputs viewOutputs = outputsDeserializer.deserialize(ViewOutputs.class);
    assertThat(viewOutputs, is(notNullValue()));

    CapturedResultsLoader loader =
        new CapturedResultsLoader(viewInputs, availableOutputs, availableImplementations);
    loader.addExtraConfigData("EmptyCurrencyMatrix", ConfigItem.of(new SimpleCurrencyMatrix()));
    Results run2 = loader.runViewFromInputs();
    System.out.println(run2);

    Result<Object> pvResult2 = run2.get(0, 0).getResult();
    if (!pvResult2.isSuccess()) {
      fail(pvResult2.toString());
    }
    Result<Object> pv01Result2 = run2.get(0, 1).getResult();
    if (!pv01Result2.isSuccess()) {
      fail(pv01Result2.toString());
    }

    // Check results are same as original ones
    assertThat(pvResult2.getValue(), is(pvResult.getValue()));
    assertThat(pv01Result2.getValue(), is(pv01Result.getValue()));

    // TODO - this should work but seems to be prevented by a Fudge issue
    // assertThat(view2Outputs, is(viewOutputs));
  }
  public void testCurveViewCanBeSavedAndRead() {

    AvailableOutputs availableOutputs = createAvailableOutputs();
    AvailableImplementations availableImplementations = createAvailableImplementations();

    ImmutableMap<Class<?>, Object> components = InterestRateMockSources.generateBaseComponents();

    ViewFactory viewFactory =
        EngineTestUtils.createViewFactory(components, availableOutputs, availableImplementations);

    View view = viewFactory.createView(createCurveBundleConfig("TEST"));

    VersionCorrection versionCorrection =
        ThreadLocalServiceContext.getInstance()
            .get(VersionCorrectionProvider.class)
            .getConfigVersionCorrection();
    CycleArguments cycleArguments =
        CycleArguments.builder(InterestRateMockSources.createMarketDataEnvironment())
            .versionCorrection(versionCorrection)
            .captureInputs(true)
            .build();

    Results run = view.run(cycleArguments);
    Result<Object> result = run.getNonPortfolioResults().get("TEST").getResult();
    if (!result.isSuccess()) {
      fail(result.getFailureMessage());
    }

    // Capture results
    ViewResultsSerializer serializer = new ViewResultsSerializer(run);

    // Write inputs to stream
    ByteArrayOutputStream inputsBaos = new ByteArrayOutputStream(1_000_000);
    serializer.serializeViewInputs(inputsBaos);

    ByteArrayOutputStream outputsBaos = new ByteArrayOutputStream(1_000_000);
    serializer.serializeViewOutputs(outputsBaos);

    ViewResultsDeserializer inputsDeserializer =
        new ViewResultsDeserializer(new ByteArrayInputStream(inputsBaos.toByteArray()));
    ViewInputs viewInputs2 = inputsDeserializer.deserialize(ViewInputs.class);
    assertThat(viewInputs2, is(notNullValue()));

    ViewResultsDeserializer outputsDeserializer =
        new ViewResultsDeserializer(new ByteArrayInputStream(outputsBaos.toByteArray()));
    ViewOutputs viewOutputs = outputsDeserializer.deserialize(ViewOutputs.class);
    assertThat(viewOutputs, is(notNullValue()));

    CapturedResultsLoader loader =
        new CapturedResultsLoader(viewInputs2, availableOutputs, availableImplementations);
    loader.addExtraConfigData("EmptyCurrencyMatrix", ConfigItem.of(new SimpleCurrencyMatrix()));
    Results run2 = loader.runViewFromInputs();

    Result<Object> result2 = run2.getNonPortfolioResults().get("TEST").getResult();
    if (!result2.isSuccess()) {
      fail(result2.toString());
    }

    // Check results are same as original ones
    ViewOutputs view2Outputs =
        ViewOutputs.builder()
            .nonPortfolioResults(run2.getNonPortfolioResults())
            .columnNames(run2.getColumnNames())
            .rows(run2.getRows())
            .build();

    assertThat(result2.getValue(), is(result.getValue()));
    assertThat(view2Outputs, is(viewOutputs));
  }