java/java-runtime/src/com/intellij/rt/execution/junit/segments/OutputObjectRegistry.java

/*
 * Copyright 2000-2009 JetBrains s.r.o.
 *
 * 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.intellij.rt.execution.junit.segments;

import java.util.Collection;
import java.util.Hashtable;

public abstract class OutputObjectRegistry  {
  private final Hashtable myKnownKeys = new Hashtable();
  private int myLastIndex = 0;
  private PacketProcessor myMainTransport;

  protected OutputObjectRegistry(PacketProcessor mainTransport, int lastIndex) {
    myLastIndex = lastIndex;
    myMainTransport = mainTransport;
  }

  public String referenceTo(Object test) {
    if (containsKey(test)) return getKey(test);
    return sendObject(test);
  }

  public String referenceTo(Object test, Collection packets) {
    if (containsKey(test)) return getKey(test);
    return sendObject(test, packets);
  }

  private boolean containsKey(Object test) {
    return myKnownKeys.containsKey(createObjectWrapper(test));
  }

  private String getKey(Object test) {
    return (String)myKnownKeys.get(createObjectWrapper(test));
  }

  private void putKey(Object test, String key) {
    myKnownKeys.put(createObjectWrapper(test), key);
  }

  public void forget(Object test) {
    myKnownKeys.remove(createObjectWrapper(test));
  }

  private String sendObject(Object test, Collection packets) {
    final String key = String.valueOf(myLastIndex++);
    putKey(test, key);
    final Packet packet = createPacket();
    packet.addString(PoolOfDelimiters.OBJECT_PREFIX).addReference(key);
    addStringRepresentation(test, packet);
    packet.addLong(getTestCont(test));
    packet.addString(PoolOfDelimiters.REFERENCE_END_STR);
    packets.add(packet);
    return key;
  }

  public Packet createPacket() {
    return new Packet(myMainTransport, this);
  }

  private String sendObject(Object test) {
    final String key = String.valueOf(myLastIndex++);
    putKey(test, key);
    Packet packet = createPacket().addString(PoolOfDelimiters.OBJECT_PREFIX).addReference(key);
    addStringRepresentation(test, packet);
    packet.addLong(getTestCont(test));
    packet.send();
    return key;
  }

  protected abstract int getTestCont(Object test);
  protected abstract void addStringRepresentation(Object test, Packet packet);

  protected static void addTestClass(Packet packet, String className) {
    packet.
        addLimitedString(PoolOfTestTypes.TEST_CLASS).
        addLimitedString(className);
  }

  protected void addUnknownTest(Packet packet, Object test) {
    packet.
        addLimitedString(PoolOfTestTypes.UNKNOWN).
        addLong(getTestCont(test)).
        addLimitedString(test.getClass().getName());
  }

  protected static void addAllInPackage(Packet packet, String name) {
    packet.
        addLimitedString(PoolOfTestTypes.ALL_IN_PACKAGE).
        addLimitedString(name);
  }

  protected static void addTestMethod(Packet packet, String methodName, String className) {
    packet.
        addLimitedString(PoolOfTestTypes.TEST_METHOD).
        addLimitedString(methodName).
        addLimitedString(className);
  }

  public int getKnownObject(Object description) {
    final Object o = getKey(description);
    if (o instanceof String) {
      return Integer.parseInt((String)o);
    }
    return 0;
  }

  protected Object createObjectWrapper(Object object) {
    return object;
  }
}