/*
 * Copyright 2012 LinkedIn, Inc
 *
 * 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 com.linkedin.parseq.trace;

import static org.testng.AssertJUnit.assertEquals;
import static org.testng.AssertJUnit.assertTrue;

import java.util.ArrayList;
import java.util.List;

import org.testng.annotations.Test;

import com.linkedin.parseq.internal.IdGenerator;


/**
 * @author Jaroslaw Odzga (jodzga@linkedin.com)
 */
public class TestTraceBuilder {

  @Test
  public void testAddRelationship() {
    final ShallowTraceBuilder trace1 =
        new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task1").setResultType(ResultType.UNFINISHED);
    final ShallowTraceBuilder trace2 =
        new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task2").setResultType(ResultType.UNFINISHED);
    final TraceBuilder builder = new TraceBuilder(1024, "test", 0L);
    builder.addRelationship(Relationship.SUCCESSOR_OF, trace1, trace2);
    Trace trace = builder.build();
    assertEquals(trace1.build(), trace.getTraceMap().get(trace1.getId()));
    assertEquals(trace2.build(), trace.getTraceMap().get(trace2.getId()));
    assertEquals(1, trace.getRelationships().size());
    assertTrue(trace.getRelationships()
        .contains(new TraceRelationship(trace1, trace2, Relationship.SUCCESSOR_OF)));
  }

  @Test
  public void testAddRelationshipTwice() {
    final ShallowTraceBuilder trace1 =
        new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task1").setResultType(ResultType.UNFINISHED);
    final ShallowTraceBuilder trace2 =
        new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task2").setResultType(ResultType.UNFINISHED);
    final TraceBuilder builder = new TraceBuilder(1024, "test", 0L);
    builder.addRelationship(Relationship.SUCCESSOR_OF, trace1, trace2);
    builder.addRelationship(Relationship.SUCCESSOR_OF, trace1, trace2);
    Trace trace = builder.build();
    assertEquals(trace1.build(), trace.getTraceMap().get(trace1.getId()));
    assertEquals(trace2.build(), trace.getTraceMap().get(trace2.getId()));
    assertEquals(1, trace.getRelationships().size());
    assertTrue(trace.getRelationships()
        .contains(new TraceRelationship(trace1, trace2, Relationship.SUCCESSOR_OF)));
  }

  @Test
  public void testRelationshipRetention() {
    final TraceBuilder builder = new TraceBuilder(4096, "test", 0L);

    List<TraceRelationship> rels = new ArrayList<TraceRelationship>();
    for (int i = 0; i < 4096; i++) {
      final ShallowTraceBuilder trace1 =
          new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task1").setResultType(ResultType.UNFINISHED);
      final ShallowTraceBuilder trace2 =
          new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task2").setResultType(ResultType.UNFINISHED);
      builder.addRelationship(Relationship.SUCCESSOR_OF, trace1, trace2);
      rels.add(new TraceRelationship(trace1, trace2, Relationship.SUCCESSOR_OF));
    }

    for (int i = 0; i < 4096 * 10; i++) {
      final ShallowTraceBuilder trace1 =
          new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task1").setResultType(ResultType.UNFINISHED);
      final ShallowTraceBuilder trace2 =
          new ShallowTraceBuilder(IdGenerator.getNextId()).setName("task2").setResultType(ResultType.UNFINISHED);
      builder.addRelationship(Relationship.SUCCESSOR_OF, trace1, trace2);
    }

    Trace trace = builder.build();
    assertEquals(rels.size(), trace.getRelationships().size());
    for (TraceRelationship rel : rels) {
      assertTrue(trace.getRelationships().contains(rel));
    }
  }

}